Update app.py

app.py

@@ -8,16 +8,18 @@ from transformers import pipeline
 
 GENAI_KEY = os.getenv("GEMINI_KEY")
 
+# 음악 모델 로드
 try:
     music_synthesiser = pipeline("text-to-audio", "facebook/musicgen-small")
 except:
     music_synthesiser = None
 
+# 8인 멤버 보이스 데이터 (절대 삭제 안 함)
 MEMBERS = {
     "서윤 (Korea)": "ko-KR-SunHiNeural",
     "Chloe (USA)": "en-US-AriaNeural",
-    "Naomi (Japan)": "ja-JP-NanamiNeural",
     "Beatrice (Brazil)": "pt-BR-FranciscaNeural",
+    "Naomi (Japan)": "ja-JP-NanamiNeural",
     "Elena (Spain)": "es-ES-ElviraNeural",
     "Amira (Egypt)": "ar-EG-SalmaNeural",
     "Liwei (China)": "zh-CN-XiaoxiaoNeural",
@@ -26,14 +28,19 @@ MEMBERS = {
 
 async def band_consulting(user_input, member_name, lang_code):
     try:
+        # ★ 언어 설정 완벽 고정
         lang_map = {"ko":"Korean","en":"English","ja":"Japanese","pt":"Portuguese","es":"Spanish","ar":"Arabic","zh":"Chinese","fr":"French"}
         target_lang = lang_map.get(lang_code, "Korean")
         
-        # 프롬프트에 '간결하게' 대답하라는 명령을 더 명확히 전달
-        system_instruction = f"당신은 {member_name}입니다. {target_lang}로만 답변하세요. 음성은 5줄 이내로 매우 짧게 요약하고, 상세한 조언과 악보는 반드시 [TAB] 안에 넣으세요. 음악 프롬프트는 [MUSIC: 영어프롬프트]로 작성하세요."
+        system_instruction = f"""
+        당신은 밴드 멤버 '{member_name}'입니다.
+        1. 반드시 '{target_lang}'로만 답변하세요. 다른 언어는 섞지 마세요.
+        2. 음성 답변(TTS)용 본문은 5줄 이내로 핵심만 요약하세요.
+        3. 모든 상세 설명, 조언, 텍스트 악보는 반드시 [TAB] 섹션에 넣으세요.
+        4. 음악 생성용 영어 프롬프트는 마지막에 [MUSIC: 프롬프트] 형식으로 포함하세요.
+        """
         
         url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={GENAI_KEY}"
-        
         payload = {
             "contents": [{"parts": [{"text": f"{system_instruction}\n질문: {user_input}"}]}],
             "safetySettings": [
@@ -44,39 +51,51 @@ async def band_consulting(user_input, member_name, lang_code):
             ]
         }
         
-        res = requests.post(url, json=payload).json()
+        response = requests.post(url, json=payload)
+        res_data = response.json()
         
-        # ★ 차단 발생 시 자동으로 '안전한' 음악 프롬프트와 답변을 강제로 생성해서 넘겨줌
-        if 'candidates' not in res or not res['candidates']:
-            clean_text = "이봐 리더, 이번 제안은 내 예술적 영감이 필터링될 정도로 강렬한걸? 락스타답게 조금만 순화해서 다시 물어봐줘!"
-            tab_display = "상세 조언이 필터링되었습니다. 질문을 조금 다르게 해보세요."
-            music_p = "fast heavy metal drum and bass, energetic rock style" # 안전한 기본 프롬프트
+        # API 차단 시 Fallback
+        if 'candidates' not in res_data or not res_data['candidates']:
+            clean_text = "이봐 리더, 내 영감이 잠시 필터링됐어. 조금 더 락스타다운 질문으로 다시 해줘!"
+            tab_display = "데이터가 차단되었습니다."
+            music_p = "heavy metal rock riff"
         else:
-            ai_text_raw = res['candidates'][0]['content']['parts'][0]['text']
-            # 데이터 추출 로직 (기존과 동일)
+            ai_text_raw = res_data['candidates'][0]['content']['parts'][0]['text']
             tab_match = re.search(r'\[TAB\](.*?)(\[|$)', ai_text_raw, re.DOTALL)
             music_match = re.search(r'\[MUSIC:(.*?)\]', ai_text_raw, re.IGNORECASE)
             tab_display = tab_match.group(1).strip() if tab_match else "No Score Available"
             clean_text = re.sub(r'\[TAB\].*?(\[|$)', '', ai_text_raw, flags=re.DOTALL)
             clean_text = re.sub(r'\[MUSIC:.*?\]', '', clean_text).strip()
-            clean_text = "\n".join(clean_text.split("\n")[:5])
-            music_p = music_match.group(1).strip().lower() if music_match else "rock guitar riff"
+            music_p = music_match.group(1).strip() if music_match else "rock guitar riff"
 
-        # 음성 생성
+        # ★ 5줄 요약 강제
+        tts_lines = clean_text.split('\n')
+        tts_final = "\n".join(tts_lines[:5])
+
+        # 음성 생성 (악보 기호 완전 제거)
+        tts_input = re.sub(r'[\*\#\-\_\~\|]', '', tts_final)
         voice_path = f"/tmp/v_{member_name.split()[0]}.mp3"
-        await edge_tts.Communicate(clean_text, MEMBERS.get(member_name, "ko-KR-SunHiNeural")).save(voice_path)
+        await edge_tts.Communicate(tts_input, MEMBERS.get(member_name, "ko-KR-SunHiNeural")).save(voice_path)
         
-        # 음악 생성 (차단 위험 단어 필터링)
-        if any(x in music_p for x in ['laugh', 'vocal', 'voice', 'human', 'sing', 'person']):
-            music_p = "intense synthesizer rock riff, high energy"
-
-        music_output = music_synthesiser(music_p, forward_params={"max_new_tokens": 512})
-        music_path = f"/tmp/m_{member_name.split()[0]}.wav"
-        scipy.io.wavfile.write(music_path, music_output["sampling_rate"], music_output["audio"][0].T)
+        # 음악 생성 (System Overload 방지)
+        music_path = None
+        if music_synthesiser:
+            try:
+                # 필터링 우회
+                safe_p = music_p.lower()
+                if any(x in safe_p for x in ['laugh', 'vocal', 'voice', 'human']):
+                    safe_p = "distorted electric guitar riff, power rock"
+                
+                music_output = music_synthesiser(safe_p, forward_params={"max_new_tokens": 512})
+                if music_output and "audio" in music_output:
+                    music_path = f"/tmp/m_{member_name.split()[0]}.wav"
+                    scipy.io.wavfile.write(music_path, music_output["sampling_rate"], music_output["audio"][0].T)
+            except:
+                music_path = None
 
-        return clean_text, voice_path, music_path, tab_display
+        return tts_final, voice_path, music_path, tab_display
     except Exception as e:
-        return f"System Overload: {str(e)}", None, None, "Error"
+        return f"System Error: {str(e)}", None, None, "Error"
 
 with gr.Blocks() as demo:
     i1 = gr.Textbox(); i2 = gr.Textbox(); i3 = gr.Textbox()