Update app.py

app.py

@@ -29,11 +29,11 @@ async def band_consulting(user_input, member_name, 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(lang_code, "Korean")
         
-        system_instruction = f"당신은 {member_name}입니다. 반드시 {target_lang}로만 답변하세요. 음성은 5줄 이내 요약, 상세 설명은 [TAB]에, 음악 프롬프트는 [MUSIC: 프롬프트]에 넣으세요."
+        # 프롬프트에 '간결하게' 대답하라는 명령을 더 명확히 전달
+        system_instruction = f"당신은 {member_name}입니다. {target_lang}로만 답변하세요. 음성은 5줄 이내로 매우 짧게 요약하고, 상세한 조언과 악보는 반드시 [TAB] 안에 넣으세요. 음악 프롬프트는 [MUSIC: 영어프롬프트]로 작성하세요."
         
         url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={GENAI_KEY}"
         
-        # ★ 차단 방지: 모든 안전 카테고리 BLOCK_NONE 설정
         payload = {
             "contents": [{"parts": [{"text": f"{system_instruction}\n질문: {user_input}"}]}],
             "safetySettings": [
@@ -41,47 +41,42 @@ async def band_consulting(user_input, member_name, lang_code):
                 {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
                 {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
                 {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}
-            ],
-            "generationConfig": {
-                "temperature": 0.7,
-                "maxOutputTokens": 1024
-            }
+            ]
         }
         
-        response = requests.post(url, json=payload)
-        res_data = response.json()
+        res = requests.post(url, json=payload).json()
         
-        # 'candidates'가 없을 경우를 위한 예외 처리 강화
-        if 'candidates' not in res_data or not res_data['candidates']:
-            # 필터링될 경우를 대비한 기본 답변 세트
-            fallback_text = f"이봐 리더, 지금 내 영감이 잠시 차단됐어. 다시 물어봐줄래? [TAB] 다시 시도해주세요. [MUSIC: powerful rock riff]"
-            ai_text_raw = fallback_text
+        # ★ 차단 발생 시 자동으로 '안전한' 음악 프롬프트와 답변을 강제로 생성해서 넘겨줌
+        if 'candidates' not in res or not res['candidates']:
+            clean_text = "이봐 리더, 이번 제안은 내 예술적 영감이 필터링될 정도로 강렬한걸? 락스타답게 조금만 순화해서 다시 물어봐줘!"
+            tab_display = "상세 조언이 필터링되었습니다. 질문을 조금 다르게 해보세요."
+            music_p = "fast heavy metal drum and bass, energetic rock style" # 안전한 기본 프롬프트
         else:
-            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 "상세 악보 없음"
-        
-        clean_text = re.sub(r'\[TAB\].*?(\[|$)', '', ai_text_raw, flags=re.DOTALL)
-        clean_text = re.sub(r'\[MUSIC:.*?\]', '', clean_text).strip()
-        
-        # 5줄 요약
-        clean_text = "\n".join(clean_text.split("\n")[:5])
+            ai_text_raw = res['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"
 
+        # 음성 생성
         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)
         
-        music_path = None
-        if music_match and music_synthesiser:
-            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)
+        # 음악 생성 (차단 위험 단어 필터링)
+        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)
 
         return clean_text, voice_path, music_path, tab_display
     except Exception as e:
-        return f"System Reset Needed: {str(e)}", None, None, "Error"
+        return f"System Overload: {str(e)}", None, None, "Error"
 
 with gr.Blocks() as demo:
     i1 = gr.Textbox(); i2 = gr.Textbox(); i3 = gr.Textbox()