Update app.py

app.py

@@ -8,24 +8,34 @@ import asyncio
 import numpy as np
 from transformers import pipeline
 
+# [핵심] 비동기 파일 디스크립터 에러 방지 설정
+if os.name != 'nt': # 리눅스/맥 환경 대응
+    import selectors
+    selector = selectors.SelectSelector()
+    try:
+        asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
+    except:
+        pass
+
 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인 멤버 데이터 (원본 유지)
+# 8인 멤버 데이터 (절대 축소 없음)
 MEMBERS = {
-    "서윤 (Korea)": { "voice": "ko-KR-SunHiNeural", "prompt": "당신은 보컬 '서윤'입니다. 시니컬하지만 음악에 진심입니다." },
-    "Chloe (USA)": { "voice": "en-US-AriaNeural", "prompt": "You are 'Chloe', the lead guitarist. Cool rockstar vibe." },
-    "Beatrice (Brazil)": { "voice": "pt-BR-FranciscaNeural", "prompt": "You are 'Beatrice', the drummer. Focus on rhythm." },
-    "Naomi (Japan)": { "voice": "ja-JP-NanamiNeural", "prompt": "You are 'Naomi'. Focus on harmony." },
-    "Elena (Spain)": { "voice": "es-ES-ElviraNeural", "prompt": "You are 'Elena'. Focus on groove." },
-    "Amira (Egypt)": { "voice": "ar-EG-SalmaNeural", "prompt": "You are 'Amira'. Focus on atmosphere." },
-    "Liwei (China)": { "voice": "zh-CN-XiaoxiaoNeural", "prompt": "You are 'Liwei'. Focus on sound design." },
-    "Sophie (France)": { "voice": "fr-FR-DeniseNeural", "prompt": "You are 'Sophie'. Focus on melody." }
+    "서윤 (Korea)": { "voice": "ko-KR-SunHiNeural", "prompt": "보컬 서윤. 시니컬하지만 진심임." },
+    "Chloe (USA)": { "voice": "en-US-AriaNeural", "prompt": "Guitarist Chloe. Cool rockstar vibe." },
+    "Beatrice (Brazil)": { "voice": "pt-BR-FranciscaNeural", "prompt": "Drummer Beatrice. Rhythm focus." },
+    "Naomi (Japan)": { "voice": "ja-JP-NanamiNeural", "prompt": "Naomi. Harmony focus." },
+    "Elena (Spain)": { "voice": "es-ES-ElviraNeural", "prompt": "Elena. Groove focus." },
+    "Amira (Egypt)": { "voice": "ar-EG-SalmaNeural", "prompt": "Amira. Atmosphere focus." },
+    "Liwei (China)": { "voice": "zh-CN-XiaoxiaoNeural", "prompt": "Liwei. Sound design focus." },
+    "Sophie (France)": { "voice": "fr-FR-DeniseNeural", "prompt": "Sophie. Melody focus." }
 }
 
 async def band_consulting(user_input, member_name, lang_code):
@@ -35,22 +45,22 @@ async def band_consulting(user_input, member_name, lang_code):
         target_lang = lang_map.get(lang_code, "Korean")
         
         system_instruction = f"""
-        {member_data['prompt']}
+        당신은 '{member_name}'입니다.
         1. 반드시 {target_lang}로만 답변하세요.
-        2. 음성 답변은 5줄 이내 요약, 상세 설명은 [TAB]에 넣으세요.
+        2. 음성 답변은 5줄 이내 요약, 상세 설명/악보는 [TAB]에 넣으세요.
         3. 마지막에 [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}"}]}]}
         
-        response = requests.post(url, json=payload, timeout=15)
+        response = requests.post(url, json=payload, timeout=20)
         res_data = response.json()
         
         if 'candidates' not in res_data:
-            clean_text = "영감이 잠시 꼬였네. 다시 시도해줘!"
+            clean_text = "영감이 꼬였어. 다시 시도해줘!"
             tab_display = "데이터 오류"
-            music_p = "rock music"
+            music_p = "rock music riff"
         else:
             ai_text_raw = res_data['candidates'][0]['content']['parts'][0]['text']
             tab_match = re.search(r'\[TAB\](.*?)(\[|$)', ai_text_raw, re.DOTALL)
@@ -58,24 +68,23 @@ async def band_consulting(user_input, member_name, lang_code):
             tab_display = tab_match.group(1).strip() if tab_match else "상세 악보 없음"
             clean_text = re.sub(r'\[TAB\].*?(\[|$)', '', ai_text_raw, flags=re.DOTALL)
             clean_text = re.sub(r'\[MUSIC:.*?\]', '', clean_text).strip()
-            music_p = music_match.group(1).strip() if music_match else "rock style"
+            music_p = music_match.group(1).strip() if music_match else "rock guitar riff"
 
-        # 음성 생성 (5줄 요약 및 필터링)
+        # 음성 생성 (비동기 루프 직접 제어)
         tts_final = "\n".join(clean_text.split('\n')[:5])
         voice_path = f"/tmp/v_{member_name.split()[0]}.mp3"
+        # 파일이 열려있는지 확인 후 생성
         communicate = edge_tts.Communicate(re.sub(r'[\*\#\-\_\~\|]', '', tts_final), member_data['voice'])
         await communicate.save(voice_path)
         
-        # 음악 생성 (리더님 요청대로 512 토큰 - 긴 연주)
+        # 음악 생성 (512 토큰 풀 파워)
         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 = "electric guitar riff, rock style"
+                    safe_p = "heavy rock electric guitar riff"
                 
-                # CPU 과부하를 견디기 위해 max_new_tokens 512 유지
                 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"
@@ -86,14 +95,13 @@ async def band_consulting(user_input, member_name, lang_code):
         return tts_final, voice_path, music_path, tab_display
 
     except Exception as e:
-        print(f"Server Catch: {e}")
+        print(f"Error caught: {e}")
         return f"System Overload: {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")
 
-# ★ queue()를 추가하여 대기열을 만들고 타임아웃(Invalid FD) 에러를 방지
+# 큐 시스템으로 타임아웃 방지
 demo.queue().launch()