Update app.py

app.py

@@ -1,13 +1,59 @@
 import gradio as gr
-from google import genai 
+import google.generativeai as genai
 import edge_tts
 import os
+import re
 
-# 1. API 키 설정
+# 1. 구글 API 설정
 GENAI_KEY = os.getenv("GEMINI_KEY")
-client = genai.Client(api_key=GENAI_KEY)
+genai.configure(api_key=GENAI_KEY)
 
-# 2. 멤버 데이터
+# ==========================================
+# 🕵️‍♂️ [생존자 탐색] 사용 가능한 모델 자동 찾기
+# ==========================================
+def get_alive_model():
+    print("📡 구글 서버에 생존 모델을 문의합니다...")
+    available_models = []
+    
+    try:
+        # 구글에 있는 모든 모델 리스트를 가져옵니다.
+        for m in genai.list_models():
+            if 'generateContent' in m.supported_generation_methods:
+                available_models.append(m.name)
+        
+        print(f"📋 발견된 모델들: {available_models}")
+
+        # 우선순위: 최신 플래시 -> 프로 -> 레거시 순서로 찾기
+        # (리더님이 말씀하신 2.5, 3.0이 있다면 여기서 걸립니다!)
+        priority_keywords = [
+            "gemini-2.0-flash", 
+            "gemini-1.5-flash", 
+            "gemini-1.5-pro",
+            "gemini-pro"
+        ]
+
+        for keyword in priority_keywords:
+            for model_name in available_models:
+                if keyword in model_name:
+                    print(f"✅ 생존 확인! [{model_name}] 모델을 사용합니다.")
+                    return genai.GenerativeModel(model_name)
+        
+        # 만약 우선순위에 없으면, 목록의 첫 번째 gemini 모델을 씁니다.
+        fallback = [m for m in available_models if "gemini" in m][0]
+        print(f"⚠️ 우선순위 모델 없음. 대체 모델 [{fallback}] 사용.")
+        return genai.GenerativeModel(fallback)
+
+    except Exception as e:
+        print(f"❌ 모델 탐색 실패: {str(e)}")
+        # 최후의 수단: 그냥 문자열로 박아버리기 (Legacy)
+        return genai.GenerativeModel("gemini-pro")
+
+# 서버 켜질 때 딱 한 번 실행해서 모델을 확정합니다.
+model = get_alive_model()
+
+# ==========================================
+# 2. 멤버 데이터 (변경 없음)
+# ==========================================
 MEMBERS = {
     "서윤 (Korea)": {
         "role": "메인 보컬",
@@ -17,37 +63,37 @@ MEMBERS = {
     "Chloe (USA)": {
         "role": "리드 기타",
         "voice": "en-US-AriaNeural", 
-        "prompt": "You are 'Chloe', the lead guitarist. Respond in English. Personality: Energetic, cool rockstar. Tone: 'Hey! Pump up the volume!'"
+        "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', the drummer from Brazil. Respond in English mixed with Portuguese. Personality: Passionate, loud. Tone: 'Vamos! Feel the beat!'"
+        "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', the rhythm guitarist. Respond in Japanese. Personality: Calm, mysterious. Tone: '大丈夫 (It's okay). Keep it steady.'"
+        "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', the bassist. Respond in Spanish mixed with English. Personality: Sexy, mature. Tone: 'Hola. Feel the bass.'"
+        "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', the keyboardist. Respond in Arabic mixed with English. Personality: Dreamy, mystical. Tone: 'Salam. Like a desert wind...'"
+        "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', the DJ. Respond in Chinese mixed with English. Personality: Trendy Gen-Z. Tone: '听着 (Listen)! Glitch is art.'"
+        "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', the violinist. Respond in French mixed with English. Personality: Elegant, artistic. Tone: 'C'est magnifique.'"
+        "prompt": "You are 'Sophie'. Respond in French mixed with English. Personality: Elegant, artistic. Tone: 'C'est magnifique.'"
     }
 }
 
@@ -58,20 +104,16 @@ async def band_consulting(user_input, member_name):
 
         member = MEMBERS[member_name]
         
-        # [Step 1] 뇌: 본명(gemini-1.5-flash-latest) 사용!
         chat_prompt = f"{member['prompt']}\n\nUser's Error/Worry: {user_input}\nResponse:"
         
         try:
-            # ★ 수정된 부분: 이름 뒤에 '-latest'를 붙였습니다.
-            response = client.models.generate_content(
-                model='gemini-1.5-flash-latest', 
-                contents=chat_prompt
-            )
+            # 확정된 모델로 생성 시도
+            response = model.generate_content(chat_prompt)
             ai_text = response.text
         except Exception as e:
-            return f"Error (Gemini): {str(e)}", None
+            # 모델이 있어도 Quota(한도) 에러가 날 수 있음 -> 에러 메시지 반환
+            return f"Error (Gemini): 모델 연결 실패.\n{str(e)}", None
 
-        # [Step 2] 입: Edge-TTS
         try:
             output_file = f"/tmp/{member_name}_reply.mp3"
             communicate = edge_tts.Communicate(ai_text, member['voice'])
@@ -84,9 +126,9 @@ async def band_consulting(user_input, member_name):
     except Exception as e:
         return f"System Error: {str(e)}", None
 
-# Blocks UI 설정
+# Blocks UI
 with gr.Blocks() as demo:
-    gr.Markdown("# Error 404 Band API Server")
+    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)")