Create app.py

app.py

@@ -0,0 +1,122 @@
+import gradio as gr
+import google.generativeai as genai
+import edge_tts
+import asyncio
+import os
+
+# 1. 구글 제미나이(뇌) 연결
+# (Settings에서 설정한 비밀키를 가져옵니다)
+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." """
+    }
+}
+
+# 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인조 밴드의 두뇌와 목소리를 생성하는 서버입니다."
+)
+
+iface.launch()