Update app.py

app.py

@@ -8,11 +8,9 @@ from transformers import pipeline
 
 GENAI_KEY = os.getenv("GEMINI_KEY")
 
-# 음악 생성 모델 로드
 try:
     music_synthesiser = pipeline("text-to-audio", "facebook/musicgen-small")
-except Exception as e:
-    print(f"Music Model Load Error: {e}")
+except:
     music_synthesiser = None
 
 MEMBERS = {
@@ -26,18 +24,17 @@ MEMBERS = {
     "Sophie (France)": "fr-FR-DeniseNeural"
 }
 
-async def band_consulting(user_input, member_name, consult_lang_code):
+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(consult_lang_code, "Korean")
+        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}'입니다.
-        1. 반드시 '{target_lang}'로 답변하세요.
-        2. 고민에 대해 아주 상세하고 친절하게 설명하세요. (문장 수 제한 없음)
-        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}"
@@ -45,27 +42,23 @@ async def band_consulting(user_input, member_name, consult_lang_code):
         
         response = requests.post(url, json=payload)
         res_data = response.json()
-
-        # 'candidates' 에러 방어
+        
         if 'candidates' not in res_data:
-            return "API 응답 오류가 발생했습니다. 키를 확인해주세요.", None, None, "Score Error"
+            return "API Error: No candidates found", 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 "상세 악보 없음"
         
-        tab_display = tab_match.group(1).strip() if tab_match else "상세 악보를 생성하지 못했습니다."
-        
-        # TTS용 텍스트 (악보 기호 제외하고 순수 설명만)
+        # TTS에서 악보 도려내기
         clean_text = re.sub(r'\[TAB\].*?(\[|$)', '', ai_text_raw, flags=re.DOTALL)
         clean_text = re.sub(r'\[MUSIC:.*?\]', '', clean_text).strip()
 
-        # 음성 생성 (설명이 길 경우 처음 몇 문장만 읽도록 조절 가능)
+        # 음성 생성
         voice_path = f"/tmp/v_{member_name.split()[0]}.mp3"
-        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)
+        await edge_tts.Communicate(clean_text, MEMBERS.get(member_name, "ko-KR-SunHiNeural")).save(voice_path)
         
         # 음악 생성
         music_path = None
@@ -76,13 +69,11 @@ async def band_consulting(user_input, member_name, consult_lang_code):
             scipy.io.wavfile.write(music_path, music_output["sampling_rate"], music_output["audio"][0].T)
 
         return clean_text, voice_path, music_path, tab_display
-
     except Exception as e:
-        return f"오류 발생: {str(e)}", None, None, "Error"
+        return 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])
-
+    btn = gr.Button("GO"); btn.click(band_consulting, [i1, i2, i3], [o1, o2, o3, o4], api_name="predict")
 demo.launch()