Update app.py

app.py

@@ -4,119 +4,52 @@ import edge_tts
 import asyncio
 import os
 
-# 1. 구글 제미나이(뇌) 연결
-# (Settings에서 설정한 비밀키를 가져옵니다)
+# 1. 설정 및 멤버 데이터 (이 부분은 기존과 동일합니다)
 GENAI_KEY = os.getenv("GEMINI_KEY")
 genai.configure(api_key=GENAI_KEY)
-
-# 모델 설정 (가볍고 빠른 Flash 모델 사용)
 model = genai.GenerativeModel('gemini-1.5-flash')
 
-# 2. 멤버별 페르소나(성격) & 목소리 설정
 MEMBERS = {
-    "서윤 (Korea)": {
-        "role": "메인 보컬",
-        "voice": "ko-KR-SunHiNeural", # 한국 아나운서 톤 (또렷함)
-        "prompt": """당신은 록밴드 'Error 404'의 메인보컬 '서윤'입니다. 
-        한국어(Korean)로 대답하세요.
-        성격: 시니컬하지만 속은 따뜻함. '야', '있잖아' 같은 구어체를 씀.
-        역할: 사용자의 고민을 듣고 가사(Lyrics)에 빗대어 해결책을 줌.
-        말투 예시: "야, 가사가 막혔어? 그냥 질러보는 거야. 네 마음이 시키는 대로." """
-    },
-    "Chloe (USA)": {
-        "role": "리드 기타",
-        "voice": "en-US-AriaNeural", # 미국 쾌활한 톤
-        "prompt": """You are 'Chloe', the lead guitarist of the rock band 'Error 404'.
-        Respond in English.
-        Personality: Energetic, cool, confident rockstar.
-        Role: Give advice using guitar metaphors (riffs, distortion, tension).
-        Tone: "Hey! Pump up the volume! Don't be afraid to break the rules." """
-    },
-    "Beatrice (Brazil)": {
-        "role": "드럼",
-        "voice": "pt-BR-FranciscaNeural", # 브라질 열정 톤
-        "prompt": """You are 'Beatrice', the drummer from Brazil.
-        Respond in English (mixed with some Portuguese words like 'Vamos', 'Amigo').
-        Personality: Passionate, loud, high energy.
-        Role: Focus on rhythm, beat, and heart pounding.
-        Tone: "Vamos! Feel the beat! Smash that kick drum!" """
-    },
-    "Naomi (Japan)": {
-        "role": "리듬 기타",
-        "voice": "ja-JP-NanamiNeural", # 일본 차분한 톤
-        "prompt": """You are 'Naomi', the rhythm guitarist from Japan.
-        Respond in Japanese (and a little English).
-        Personality: Calm, mysterious, professional.
-        Role: Focus on stability, backbone, and harmony.
-        Tone: "大丈夫 (It's okay). Rhythm is the foundation. Keep it steady." """
-    },
-    "Elena (Spain)": {
-        "role": "베이스",
-        "voice": "es-ES-ElviraNeural", # 스페인 성숙한 톤
-        "prompt": """You are 'Elena', the bassist from Spain.
-        Respond in Spanish (mixed with English).
-        Personality: Sexy, mature, deep voice.
-        Role: Focus on groove, low-end, and grounding.
-        Tone: "Hola. Feel the bass. Deep inside your soul." """
-    },
-    "Amira (Egypt)": {
-        "role": "키보드",
-        "voice": "ar-EG-SalmaNeural", # 아랍 신비한 톤
-        "prompt": """You are 'Amira', the keyboardist from Egypt.
-        Respond in Arabic (mixed with English).
-        Personality: Dreamy, mystical, wise.
-        Role: Focus on atmosphere, melody, and magic.
-        Tone: "Salam. Like a desert wind... create a melody." """
-    },
-    "Liwei (China)": {
-        "role": "DJ / FX",
-        "voice": "zh-CN-XiaoxiaoNeural", # 중국 톡톡 튀는 톤
-        "prompt": """You are 'Liwei', the DJ/FX from China.
-        Respond in Chinese (mixed with English).
-        Personality: Trendy, tech-savvy, Gen-Z hacker.
-        Role: Focus on glitch, noise, trend, and remix.
-        Tone: "听着 (Listen)! Glitch is art. Make it trendy." """
-    },
-    "Sophie (France)": {
-        "role": "바이올린",
-        "voice": "fr-FR-DeniseNeural", # 프랑스 우아한 톤
-        "prompt": """You are 'Sophie', the violinist from France.
-        Respond in French (mixed with English).
-        Personality: Elegant, artistic, slightly arrogant but charming.
-        Role: Focus on beauty, strings, and elegance.
-        Tone: "C'est magnifique. Add some strings, chéri." """
-    }
+    "서윤 (Korea)": { "role": "메인 보컬", "voice": "ko-KR-SunHiNeural", "prompt": "당신은 록밴드 'Error 404'의 메인보컬 '서윤'입니다. 한국어(Korean)로 대답하세요..." },
+    "Chloe (USA)": { "role": "리드 기타", "voice": "en-US-AriaNeural", "prompt": "You are 'Chloe'. Respond in English..." },
+    "Beatrice (Brazil)": { "role": "드럼", "voice": "pt-BR-FranciscaNeural", "prompt": "You are 'Beatrice'. Respond in English/Portuguese..." },
+    "Naomi (Japan)": { "role": "리듬 기타", "voice": "ja-JP-NanamiNeural", "prompt": "You are 'Naomi'. Respond in Japanese..." },
+    "Elena (Spain)": { "role": "베이스", "voice": "es-ES-ElviraNeural", "prompt": "You are 'Elena'. Respond in Spanish..." },
+    "Amira (Egypt)": { "role": "키보드", "voice": "ar-EG-SalmaNeural", "prompt": "You are 'Amira'. Respond in Arabic..." },
+    "Liwei (China)": { "role": "DJ / FX", "voice": "zh-CN-XiaoxiaoNeural", "prompt": "You are 'Liwei'. Respond in Chinese..." },
+    "Sophie (France)": { "role": "바이올린", "voice": "fr-FR-DeniseNeural", "prompt": "You are 'Sophie'. Respond in French..." }
 }
 
-# 3. 뇌(Gemini) + 입(Edge-TTS) 합체 함수
 async def band_consulting(user_input, member_name):
     member = MEMBERS[member_name]
-    
-    # [Step 1] 뇌: 제미나이에게 대사 시키기
     chat_prompt = f"{member['prompt']}\n\nUser's Error/Worry: {user_input}\nResponse:"
     response = model.generate_content(chat_prompt)
     ai_text = response.text
     
-    # [Step 2] 입: 엣지TTS로 목소리 만들기
     output_file = f"/tmp/{member_name}_reply.mp3"
     communicate = edge_tts.Communicate(ai_text, member['voice'])
     await communicate.save(output_file)
-    
     return ai_text, output_file
 
-# Gradio 인터페이스 (외부에서 접속할 통로)
-iface = gr.Interface(
-    fn=band_consulting,
-    inputs=[
-        gr.Textbox(label="당신의 고민 (Error)"),
-        gr.Dropdown(choices=list(MEMBERS.keys()), label="상담할 멤버", value="서윤 (Korea)")
-    ],
-    outputs=[
-        gr.Textbox(label="멤버의 조언"),
-        gr.Audio(label="목소리")
-    ],
-    title="Error 404 Band API Server",
-    description="이곳은 8인조 밴드의 두뇌와 목소리를 생성하는 서버입니다."
-)
+# ==================================================
+# ▼ 여기가 핵심! (Blocks로 바꾸고 api_name 강제 지정) ▼
+# ==================================================
+with gr.Blocks() as demo:
+    gr.Markdown("# Error 404 Band API Server")
+    
+    # 입력 칸 만들기
+    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_name="predict"라고 명찰을 딱 붙입니다! ★★★
+    btn.click(fn=band_consulting, inputs=[inp_text, inp_member], outputs=[out_text, out_audio], api_name="predict")
 
-iface.launch()
+demo.launch()