Update app.py

app.py

@@ -8,12 +8,13 @@ from transformers import pipeline
 
 GENAI_KEY = os.getenv("GEMINI_KEY")
 
+# 음악 생성 모델 로드
 try:
     music_synthesiser = pipeline("text-to-audio", "facebook/musicgen-small")
-except:
+except Exception as e:
+    print(f"Music Model Load Error: {e}")
     music_synthesiser = None
 
-# 8인 멤버 보이스 매칭
 MEMBERS = {
     "서윤 (Korea)": "ko-KR-SunHiNeural",
     "Chloe (USA)": "en-US-AriaNeural",
@@ -30,49 +31,58 @@ async def band_consulting(user_input, member_name, consult_lang_code):
         lang_map = {"ko": "Korean", "en": "English", "ja": "Japanese", "pt": "Portuguese", "es": "Spanish", "ar": "Arabic", "zh": "Chinese", "fr": "French"}
         target_lang = lang_map.get(consult_lang_code, "Korean")
         
+        # ★ 리더님 요청: 설명 충분히, 악보 반드시 포함
         system_instruction = f"""
-        당신은 밴드 멤버 '{member_name}'입니다.
-        1. 반드시 '{target_lang}'로만 답변하세요. 영어면 영어, 일본어면 일본어만 사용합니다.
-        2. 음성 답변(TTS)은 5줄 이내로 핵심만 요약하세요.
-        3. 모든 상세 설명, 화성학 조언, 텍스트 악보는 [TAB] 섹션에 넣으세요.
-        4. 음악 생성용 영어 프롬프트는 마지막에 [MUSIC: 프롬프트] 형식으로 포함하세요.
+        당신은 락밴드 멤버 '{member_name}'입니다.
+        1. 반드시 '{target_lang}'로 답변하세요.
+        2. 고민에 대해 아주 상세하고 친절하게 설명하세요. (문장 수 제한 없음)
+        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}"}]}]}
-        res = requests.post(url, json=payload).json()
-        ai_text_raw = res['candidates'][0]['content']['parts'][0]['text']
         
-        # 데이터 정밀 파싱
+        response = requests.post(url, json=payload)
+        res_data = response.json()
+
+        # 'candidates' 에러 방어
+        if 'candidates' not in res_data:
+            return "API 응답 오류가 발생했습니다. 키를 확인해주세요.", None, None, "Score Error"
+            
+        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 Detailed Score"
         
-        # ★ 악보 읽기 방지: [TAB] 이후 내용을 통째로 삭제하여 목소리용 텍스트 생성
+        tab_display = tab_match.group(1).strip() if tab_match else "상세 악보를 생성하지 못했습니다."
+        
+        # TTS용 텍스트 (악보 기호 제외하고 순수 설명만)
         clean_text = re.sub(r'\[TAB\].*?(\[|$)', '', ai_text_raw, flags=re.DOTALL)
         clean_text = re.sub(r'\[MUSIC:.*?\]', '', clean_text).strip()
-        tts_lines = clean_text.split('\n')
-        tts_final = "\n".join(tts_lines[:5]) # 5줄 요약
 
-        # 음성 생성 (특수기호 제거)
-        tts_input = re.sub(r'[\*\#\-\_\~\|]', '', tts_final)
+        # 음성 생성 (설명이 길 경우 처음 몇 문장만 읽도록 조절 가능)
         voice_path = f"/tmp/v_{member_name.split()[0]}.mp3"
-        await edge_tts.Communicate(tts_input, MEMBERS.get(member_name, "ko-KR-SunHiNeural")).save(voice_path)
+        tts_text = " ".join(clean_text.split('.')[:3]) # 너무 길면 오류나므로 앞 3문장만 TTS
+        await edge_tts.Communicate(tts_text, MEMBERS.get(member_name, "ko-KR-SunHiNeural")).save(voice_path)
         
-        # 음악 생성 (512 tokens = 약 10~12초)
+        # 음악 생성
         music_path = None
         if music_match and music_synthesiser:
-            p = music_match.group(1).strip() + ", high quality, studio recording"
-            music_output = music_synthesiser(p, forward_params={"max_new_tokens": 512, "do_sample": True})
+            p = music_match.group(1).strip()
+            music_output = music_synthesiser(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)
 
-        return tts_final, voice_path, music_path, tab_display
+        return clean_text, voice_path, music_path, tab_display
+
     except Exception as e:
-        return str(e), None, None, "Error"
+        return f"오류 발생: {str(e)}", None, None, "Error"
 
 with gr.Blocks() as demo:
     i1 = gr.Textbox(); i2 = gr.Textbox(); i3 = gr.Textbox()
     o1 = gr.Textbox(); o2 = gr.Audio(); o3 = gr.Audio(); o4 = gr.Textbox()
-    btn = gr.Button("GO"); btn.click(band_consulting, [i1, i2, i3], [o1, o2, o3, o4], api_name="predict")
+    btn = gr.Button("GO"); btn.click(band_consulting, [i1, i2, i3], [o1, o2, o3, o4])
+
 demo.launch()