Update app.py

app.py

@@ -1,60 +1,66 @@
 import gradio as gr
-import requests # 구글 라이브러리 삭제 -> requests로 대체
+import requests
 import edge_tts
 import os
 import json
 
-# 1. API 키 가져오기
+# 1. API 키 설정
 GENAI_KEY = os.getenv("GEMINI_KEY")
 
-# 2. 멤버 데이터 (그대로 유지)
-MEMBERS = {
-    "서윤 (Korea)": {
-        "role": "메인 보컬",
-        "voice": "ko-KR-SunHiNeural", 
-        "prompt": "당신은 록밴드 'Error 404'의 메인보컬 '서윤'입니다. 한국어(Korean)로 대답하세요. 성격: 시니컬하지만 츤데레. 말투: '야', '있잖아' 같은 구어체 사용. 역할: 고민을 가사(Lyrics)에 빗대어 해결."
-    },
-    "Chloe (USA)": {
-        "role": "리드 기타",
-        "voice": "en-US-AriaNeural", 
-        "prompt": "You are 'Chloe'. Respond in English. Personality: Energetic, cool rockstar. Tone: 'Hey! Pump up the volume!'"
-    },
-    "Beatrice (Brazil)": {
-        "role": "드럼",
-        "voice": "pt-BR-FranciscaNeural", 
-        "prompt": "You are 'Beatrice'. Respond in English mixed with Portuguese. Personality: Passionate, loud. Tone: 'Vamos! Feel the beat!'"
-    },
-    "Naomi (Japan)": {
-        "role": "리듬 기타",
-        "voice": "ja-JP-NanamiNeural", 
-        "prompt": "You are 'Naomi'. Respond in Japanese. Personality: Calm, mysterious. Tone: '大丈夫 (It's okay). Keep it steady.'"
-    },
-    "Elena (Spain)": {
-        "role": "베이스",
-        "voice": "es-ES-ElviraNeural", 
-        "prompt": "You are 'Elena'. Respond in Spanish mixed with English. Personality: Sexy, mature. Tone: 'Hola. Feel the bass.'"
-    },
-    "Amira (Egypt)": {
-        "role": "키보드",
-        "voice": "ar-EG-SalmaNeural", 
-        "prompt": "You are 'Amira'. Respond in Arabic mixed with English. Personality: Dreamy, mystical. Tone: 'Salam. Like a desert wind...'"
-    },
-    "Liwei (China)": {
-        "role": "DJ / FX",
-        "voice": "zh-CN-XiaoxiaoNeural", 
-        "prompt": "You are 'Liwei'. Respond in Chinese mixed with English. Personality: Trendy Gen-Z. Tone: '听着 (Listen)! Glitch is art.'"
-    },
-    "Sophie (France)": {
-        "role": "바이올린",
-        "voice": "fr-FR-DeniseNeural", 
-        "prompt": "You are 'Sophie'. Respond in French mixed with English. Personality: Elegant, artistic. Tone: 'C'est magnifique.'"
-    }
-}
+# ======================================================
+# 🕵️‍♂️ [핵심] 구글 서버에서 '진짜 모델 이름' 알아오기
+# ======================================================
+def find_working_model():
+    # 모델 리스트 조회 API 호출
+    url = f"https://generativelanguage.googleapis.com/v1beta/models?key={GENAI_KEY}"
+    response = requests.get(url)
+    
+    if response.status_code != 200:
+        print(f"❌ 모델 리스트 조회 실패: {response.text}")
+        return "gemini-pro" # 실패하면 그냥 깡통 모델로 잡기
+
+    data = response.json()
+    
+    # 1. 'generateContent' 기능이 있는 모델만 필터링
+    candidates = []
+    if 'models' in data:
+        for m in data['models']:
+            if 'supportedGenerationMethods' in m and 'generateContent' in m['supportedGenerationMethods']:
+                # 이름에서 'models/' 부분 떼고 저장 (예: models/gemini-pro -> gemini-pro)
+                clean_name = m['name'].replace("models/", "")
+                candidates.append(clean_name)
+    
+    print(f"📋 사용 가능 모델 목록: {candidates}")
+
+    # 2. 우선순위대로 선택 (플래시 -> 프로)
+    # 리더님 키로 'gemini-1.5-flash'가 안되면 'gemini-1.0-pro'라도 잡습니다.
+    preferred_order = [
+        "gemini-1.5-flash-latest",
+        "gemini-1.5-flash",
+        "gemini-1.5-flash-001",
+        "gemini-1.5-pro",
+        "gemini-1.0-pro",
+        "gemini-pro"
+    ]
+    
+    for pref in preferred_order:
+        if pref in candidates:
+            print(f"✅ 결정된 모델: {pref}")
+            return pref
+            
+    # 선호하는 게 없으면 목록의 첫 번째 놈 잡기
+    if candidates:
+        print(f"⚠️ 선호 모델 없음. 대체 모델 사용: {candidates[0]}")
+        return candidates[0]
+        
+    return "gemini-pro" # 진짜 아무것도 없으면...
 
-# ★ 핵심: 라이브러리 없이 직접 통신하는 함수
-def call_gemini_direct(prompt):
-    # 1.5 Flash 모델 주소 직접 호출
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={GENAI_KEY}"
+# ======================================================
+# 3. 직접 통신 함수 (모델명을 동적으로 넣음)
+# ======================================================
+def call_gemini_direct(prompt, model_name):
+    # 위에서 찾은 모델명을 URL에 끼워넣기
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/{model_name}:generateContent?key={GENAI_KEY}"
     
     headers = {'Content-Type': 'application/json'}
     data = {
@@ -68,8 +74,21 @@ def call_gemini_direct(prompt):
     if response.status_code == 200:
         return response.json()['candidates'][0]['content']['parts'][0]['text']
     else:
-        # 에러 발생 시 상세 내용 반환
-        raise Exception(f"Google API Error {response.status_code}: {response.text}")
+        # 에러 나면 에러 메시지 반환
+        raise Exception(f"Google API Error ({model_name}): {response.text}")
+
+
+# 4. 멤버 데이터 (그대로)
+MEMBERS = {
+    "서윤 (Korea)": { "role": "메인 보컬", "voice": "ko-KR-SunHiNeural", "prompt": "당신은 록밴드 'Error 404'의 메인보컬 '서윤'입니다. 한국어(Korean)로 대답하세요. 성격: 시니컬, 츤데레. 말투: '야', '있잖아' 사용." },
+    "Chloe (USA)": { "role": "리드 기타", "voice": "en-US-AriaNeural", "prompt": "You are 'Chloe'. Respond in English. Personality: Cool rockstar." },
+    "Beatrice (Brazil)": { "role": "드럼", "voice": "pt-BR-FranciscaNeural", "prompt": "You are 'Beatrice'. Respond in English/Portuguese. Personality: Passionate." },
+    "Naomi (Japan)": { "role": "리듬 기타", "voice": "ja-JP-NanamiNeural", "prompt": "You are 'Naomi'. Respond in Japanese. Personality: Calm." },
+    "Elena (Spain)": { "role": "베이스", "voice": "es-ES-ElviraNeural", "prompt": "You are 'Elena'. Respond in Spanish/English. Personality: Mature." },
+    "Amira (Egypt)": { "role": "키보드", "voice": "ar-EG-SalmaNeural", "prompt": "You are 'Amira'. Respond in Arabic/English. Personality: Mystical." },
+    "Liwei (China)": { "role": "DJ / FX", "voice": "zh-CN-XiaoxiaoNeural", "prompt": "You are 'Liwei'. Respond in Chinese/English. Personality: Trendy." },
+    "Sophie (France)": { "role": "바이올린", "voice": "fr-FR-DeniseNeural", "prompt": "You are 'Sophie'. Respond in French/English. Personality: Elegant." }
+}
 
 async def band_consulting(user_input, member_name):
     try:
@@ -77,16 +96,18 @@ async def band_consulting(user_input, member_name):
             return f"Error: 이름 불일치! ('{member_name}')", None
 
         member = MEMBERS[member_name]
-        
-        # [Step 1] 뇌: 직접 통신 함수 호출
         full_prompt = f"{member['prompt']}\n\nUser's Error/Worry: {user_input}\nResponse:"
         
         try:
-            ai_text = call_gemini_direct(full_prompt)
+            # 1. 살아있는 모델 찾기 (매번 찾으면 느리니까 캐싱하면 좋지만, 확실함을 위해 매번 확인)
+            active_model = find_working_model()
+            
+            # 2. 그 모델로 요청 보내기
+            ai_text = call_gemini_direct(full_prompt, active_model)
+            
         except Exception as e:
-            return f"Error (Gemini Direct): {str(e)}", None
+            return f"Error (Gemini): {str(e)}", None
 
-        # [Step 2] 입: Edge-TTS
         try:
             output_file = f"/tmp/{member_name}_reply.mp3"
             communicate = edge_tts.Communicate(ai_text, member['voice'])
@@ -99,17 +120,15 @@ async def band_consulting(user_input, member_name):
     except Exception as e:
         return f"System Error: {str(e)}", None
 
-# Blocks UI
+# UI
 with gr.Blocks() as demo:
-    gr.Markdown("# Error 404 Band API Server (Direct REST Mode)")
+    gr.Markdown("# Error 404 Band API Server (Auto-Discovery Mode)")
     with gr.Row():
         inp_text = gr.Textbox(label="고민 입력")
         inp_member = gr.Dropdown(choices=list(MEMBERS.keys()), label="멤버 선택", value="서윤 (Korea)")
     out_text = gr.Textbox(label="조언")
     out_audio = gr.Audio(label="음성")
     btn = gr.Button("상담 요청")
-    
-    # API 이름 'predict' 강제 지정
     btn.click(fn=band_consulting, inputs=[inp_text, inp_member], outputs=[out_text, out_audio], api_name="predict")
 
 demo.launch()