Update app.py

app.py

@@ -1,4 +1,4 @@
-import os, re, uuid, torch, scipy.io.wavfile, edge_tts, asyncio
+import os, re, uuid, torch, scipy.io.wavfile, edge_tts, asyncio, random
 import numpy as np
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
@@ -31,23 +31,35 @@ class ModelManager:
             cls._music_pipeline = pipeline("text-to-audio", "facebook/musicgen-small", device="cpu")
         return cls._music_pipeline
 
-MEMBERS_VOICE = {
-    "서윤 (Korea)": "ko-KR-SunHiNeural", "Chloe (USA)": "en-US-AriaNeural", 
-    "Naomi (Japan)": "ja-JP-NanamiNeural", "Beatrice (Brazil)": "pt-BR-FranciscaNeural",
-    "Elena (Spain)": "es-ES-ElviraNeural", "Amira (Egypt)": "ar-EG-SalmaNeural",
-    "Liwei (China)": "zh-CN-XiaoxiaoNeural", "Sophie (France)": "fr-FR-DeniseNeural"
+# 8인 멤버 설정 (국가별 언어 및 보이스)
+MEMBERS_CONFIG = {
+    "서윤 (Korea)": {"voice": "ko-KR-SunHiNeural", "lang": "Korean"},
+    "Chloe (USA)": {"voice": "en-US-AriaNeural", "lang": "English"},
+    "Naomi (Japan)": {"voice": "ja-JP-NanamiNeural", "lang": "Japanese"},
+    "Beatrice (Brazil)": {"voice": "pt-BR-FranciscaNeural", "lang": "Portuguese"},
+    "Elena (Spain)": {"voice": "es-ES-ElviraNeural", "lang": "Spanish"},
+    "Amira (Egypt)": {"voice": "ar-EG-SalmaNeural", "lang": "Arabic"},
+    "Liwei (China)": {"voice": "zh-CN-XiaoxiaoNeural", "lang": "Chinese"},
+    "Sophie (France)": {"voice": "fr-FR-DeniseNeural", "lang": "French"}
 }
 
-async def band_consulting(user_input, member_name, lang_code, g_inst, b_inst, d_inst, chords):
+async def band_consulting(user_input, member_name, consult_lang, g_inst, b_inst, d_inst, chords):
     req_id = str(uuid.uuid4())[:8]
     voice_path = f"/tmp/v_{req_id}.mp3"
     music_path = f"/tmp/m_{req_id}.wav"
     
+    # 1. 멤버 선정 로직: 영어가 선택되면 랜덤 멤버가 담당
+    actual_member = member_name
+    if consult_lang == "English":
+        actual_member = random.choice(list(MEMBERS_CONFIG.keys()))
+    
+    target_lang = MEMBERS_CONFIG[actual_member]["lang"] if consult_lang == "Native" else consult_lang
+    
     jam_context = f"Guitar: {g_inst}, Bass: {b_inst}, Drums: {d_inst}, Chords: {chords}"
-    system_prompt = f"""당신은 락스타 {member_name}입니다. 반드시 {lang_code}로 답변하세요.
-    전문가로서 5~7문장의 깊이 있는 상담을 제공하세요.
-    [TAB] 섹션에는 상세한 악보나 코드를 적으세요.
-    [MUSIC] 섹션에는 다음 JAM 요청을 반영한 영어 프롬프트를 작성하세요: {jam_context}"""
+    system_prompt = f"""You are {actual_member}. Respond ONLY in {target_lang}.
+    Provide 5-7 lines of professional music advice. 
+    [TAB] Section: Detailed chords/tabs.
+    [MUSIC] Section: English prompt reflecting: {jam_context}"""
     
     ai_text_raw = ""
     groq_client = ModelManager.get_groq()
@@ -65,16 +77,19 @@ async def band_consulting(user_input, member_name, lang_code, g_inst, b_inst, d_
         out = qwen(f"<|im_start|>system\n{system_prompt}<|im_end|>\n<|im_start|>user\n{user_input}<|im_end|>\nassistant\n", max_new_tokens=1024)
         ai_text_raw = out[0]['generated_text'].split("assistant\n")[-1]
 
+    # 파싱
     tab_match = re.search(r'\[TAB\](.*?)(\[|$)', ai_text_raw, re.DOTALL | re.IGNORECASE)
     music_match = re.search(r'\[MUSIC:(.*?)\]', ai_text_raw, re.IGNORECASE)
     tab_display = tab_match.group(1).strip() if tab_match else "No Data"
     clean_text = re.sub(r'\[TAB\].*?(\[|$)', '', ai_text_raw, flags=re.DOTALL | re.IGNORECASE)
     clean_text = re.sub(r'\[MUSIC:.*?\]', '', clean_text, flags=re.IGNORECASE).strip()
     
-    voice_name = MEMBERS_VOICE.get(member_name, "ko-KR-SunHiNeural")
+    # TTS
+    voice_name = MEMBERS_CONFIG[actual_member]["voice"]
     communicate = edge_tts.Communicate(re.sub(r'[\*\#\-\_\~\|]', '', clean_text), voice_name)
     await communicate.save(voice_path)
 
+    # MusicGen (12초)
     music_gen = ModelManager.get_music()
     music_p = music_match.group(1).strip() if music_match else "rock"
     music_output = music_gen(music_p, forward_params={"max_new_tokens": 512})
@@ -82,13 +97,12 @@ async def band_consulting(user_input, member_name, lang_code, g_inst, b_inst, d_
     audio_int16 = (audio_data * 32767).astype(np.int16)
     scipy.io.wavfile.write(music_path, music_output["sampling_rate"], audio_int16)
 
-    return clean_text, voice_path, music_path, tab_display
+    return f"[{actual_member}] {clean_text}", voice_path, music_path, tab_display
 
 with gr.Blocks() as demo:
-    # 7개의 입력 (고민, 멤버, 상담언어, 기타, 베이스, 드럼, 코드)
-    inputs = [gr.Textbox(visible=False) for _ in range(7)]
-    outputs = [gr.Textbox(visible=False), gr.Audio(visible=False), gr.Audio(visible=False), gr.Textbox(visible=False)]
+    in_list = [gr.Textbox(visible=False) for _ in range(7)]
+    out_list = [gr.Textbox(visible=False), gr.Audio(visible=False), gr.Audio(visible=False), gr.Textbox(visible=False)]
     btn = gr.Button("API", visible=False)
-    btn.click(band_consulting, inputs, outputs, api_name="predict")
+    btn.click(band_consulting, in_list, out_list, api_name="predict")
 
 demo.queue().launch()