Update app.py

app.py

@@ -84,25 +84,42 @@ STRUCTURE_TAGS = [
     "[Interlude]", "[Hook]", "[Outro]", "[Inst]", "[Solo]"
 ]
 
-# HeartMuLa 스타일 태그 가이드
+# HeartMuLa 스타일 태그 가이드 (데모 페이지 기반)
 HEARTMULA_TAG_GUIDE = """
-## 🎼 HeartMuLa Tag Format Guide
+## 🎼 HeartMuLa 입력 형식
 
-### STRUCTURE TAGS (가사 내 사용):
-[Intro], [Verse], [Prechorus], [Chorus], [Bridge], [Interlude], [Hook], [Outro], [Inst], [Solo]
+### ⚠️ 중요: HeartMuLa는 2가지 입력만 받습니다!
+1. **가사 (Lyrics)** - 구조 태그 포함
+2. **스타일 태그 (Tags)** - 콤마 구분, 공백 없음
 
-### STYLE TAGS FORMAT (콤마 구분, 공백 없음):
+❌ 보컬 타입(듀엣, 그룹 등)은 지원하지 않습니다!
+
+### 📝 구조 태그 (가사 내 사용):
+`[Intro]`, `[Verse]`, `[Prechorus]`, `[Chorus]`, `[Bridge]`, `[Outro]`
+
+### 🏷️ 스타일 태그 형식 (콤마 구분, 공백 없음):
 ```
 piano,happy,wedding,synthesizer,romantic
-guitar,sad,ballad,strings,emotional
-drums,energetic,rock,electric_guitar,powerful
+energetic,drums,driving,rock,electric guitar
+soft,Sad,Ballad,Longing
 ```
 
-### RECOMMENDED TAG CATEGORIES:
-**Instruments:** piano,guitar,drums,bass,synthesizer,strings,violin,cello,trumpet,saxophone,flute,harp,organ
-**Mood:** happy,sad,romantic,energetic,calm,melancholic,uplifting,dark,dreamy,nostalgic,powerful,peaceful
-**Genre:** pop,rock,jazz,classical,electronic,folk,blues,r&b,hip_hop,disco,ballad,cinematic
-**Tempo:** fast,slow,moderate,upbeat,relaxed
+### 🎹 태그 카테고리 (실제 데모 기반):
+
+**악기 (Instruments):**
+Piano, Keyboard, synthesizer, drum machine, electric guitar, acoustic guitar, drums, Strings, bass
+
+**장르 (Genre):**
+pop, R&B, rock, Ballad, electronic, acoustic
+
+**분위기/감정 (Mood/Emotion):**
+warm, reflection, energetic, soft, Sad, Regret, Longing, meditation, faith, peaceful, Romantic, emotional, hope, hopeful, powerful, epic, driving, happy, uplifting, dreamy
+
+**상황 (Situation):**
+Cafe, wedding, Walking, self-discovery
+
+### 🌍 지원 언어:
+🇺🇸 English | 🇰🇷 Korean | 🇨🇳 Chinese | 🇯🇵 Japanese | 🇪🇸 Spanish
 """
 
 # HeartMuLa 권장 가사 구조
@@ -190,20 +207,20 @@ Every single day"""
 
 EXAMPLE_TAGS = "piano,happy,uplifting,pop"
 
-# 예제 프롬프트 (태그 세트)
+# 예제 프롬프트 (태그 세트) - HeartMuLa 데모 기반
 EXAMPLE_TAG_PRESETS = {
-    "🎹 Piano Ballad": "piano,ballad,emotional,romantic,slow",
-    "🎸 Rock Energetic": "guitar,drums,rock,energetic,powerful,fast",
-    "🎷 Jazz Smooth": "piano,saxophone,jazz,smooth,relaxed,romantic",
-    "🎵 K-Pop Dance": "synthesizer,drums,bass,kpop,energetic,upbeat,pop",
-    "🌙 Lo-Fi Chill": "piano,guitar,lofi,calm,dreamy,relaxed",
-    "🎻 Orchestral Epic": "strings,orchestra,cinematic,epic,powerful,dramatic",
-    "🪩 Disco Funk": "bass,drums,synthesizer,disco,funky,upbeat,dance",
-    "🌿 Folk Acoustic": "guitar,folk,acoustic,calm,peaceful,warm",
-    "💫 Electronic Dance": "synthesizer,drums,electronic,energetic,dance,upbeat",
-    "😢 Sad Melancholic": "piano,strings,sad,melancholic,slow,emotional",
-    "🎉 Happy Uplifting": "piano,happy,uplifting,pop,bright,joyful",
-    "🌅 Wedding Romantic": "piano,happy,wedding,synthesizer,romantic"
+    "☕ Cafe Warm Pop": "warm,reflection,pop,Cafe",
+    "🎹 R&B Soft Regret": "R&B,Keyboard,Regret,drum machine,electric guitar,synthesizer,soft",
+    "⚡ Electronic Energetic": "energetic,electronic,synthesizer,drum machine,self-discovery",
+    "😢 Sad Ballad Longing": "soft,Sad,Ballad,Longing",
+    "🧘 Meditation Peaceful": "meditation,faith,acoustic,peaceful",
+    "💒 Wedding Romantic": "wedding,Piano,Strings,acoustic guitar,pop,Romantic",
+    "💔 Emotional Piano": "Sad,Regret,electric guitar,drums,Piano,emotional",
+    "🚶 Hopeful Acoustic": "Walking,acoustic guitar,Strings,hope,hopeful",
+    "🎸 Epic Electronic": "synthesizer,drum machine,powerful,epic,Keyboard,electronic",
+    "🔥 Driving Rock": "energetic,drums,driving,rock,electric guitar",
+    "🎵 Happy Uplifting": "piano,happy,uplifting,pop",
+    "🌙 Soft Dreamy": "soft,dreamy,piano,strings,peaceful"
 }
 
 
@@ -287,21 +304,35 @@ CRITICAL OUTPUT RULES:
 OUTPUT ONLY CLEAN LYRICS WITH OPTIMAL TAG STRUCTURE IN THE ORIGINAL LANGUAGE."""
 }
 
-# 태그 생성 에이전트
-TAG_AGENT = f"""You are a style tag generator for HeartMuLa.
+# 태그 생성 에이전트 - HeartMuLa 데모 태그 기반
+TAG_AGENT = f"""You are a style tag generator for HeartMuLa music generation.
 
-{HEARTMULA_TAG_GUIDE}
+HeartMuLa uses comma-separated tags WITHOUT spaces.
+
+VALID TAG CATEGORIES (from official demo):
+
+**Instruments:**
+Piano, Keyboard, synthesizer, drum machine, electric guitar, acoustic guitar, drums, Strings, bass
+
+**Genre:**
+pop, R&B, rock, Ballad, electronic, acoustic
 
-Your task: Generate OPTIMIZED comma-separated tags (NO SPACES between tags).
+**Mood/Emotion:**
+warm, reflection, energetic, soft, Sad, Regret, Longing, meditation, faith, peaceful, Romantic, emotional, hope, hopeful, powerful, epic, driving, happy, uplifting, dreamy
+
+**Situation:**
+Cafe, wedding, Walking, self-discovery
 
 RULES:
-1. Tags must be comma-separated WITHOUT spaces: tag1,tag2,tag3
-2. Use lowercase with underscores for multi-word: electric_guitar, hip_hop
-3. Include 5-8 tags covering: instrument, mood, genre, tempo
-4. Be specific: "piano" not "keyboard", "guitar" not "strings"
+1. Output 4-6 tags, comma-separated, NO SPACES between tags
+2. Use ONLY tags from the categories above
+3. Match the requested genre and mood
 
-OUTPUT FORMAT (example):
-piano,synthesizer,happy,pop,upbeat,romantic
+OUTPUT FORMAT (examples):
+piano,happy,uplifting,pop
+warm,reflection,pop,Cafe
+energetic,drums,driving,rock,electric guitar
+soft,Sad,Ballad,Longing
 
 Output ONLY the comma-separated tags, nothing else."""
 
@@ -400,7 +431,7 @@ def clean_tags(tags: str) -> str:
 @spaces.GPU(duration=120)
 def generate_lyrics_soma(
     api_key: str, theme: str, genre: str, mood: str, 
-    language: str, vocal_type: str, additional: str, 
+    language: str, additional: str, 
     progress=gr.Progress()
 ):
     """SOMA 가사 생성"""
@@ -414,12 +445,13 @@ def generate_lyrics_soma(
 - Genre: {genre}
 - Mood: {mood}
 - Language: {language} ← CRITICAL: You MUST write ALL lyrics in {language} language!
-- Vocal Type: {vocal_type}
 {f'- Additional: {additional}' if additional else ''}
 
 ⚠️ LANGUAGE REQUIREMENT: Write the ENTIRE lyrics in {language}!
-{"- 반드시 한국어로 가사를 작성하세요! 영어 단어는 최소화하고 한글로 작성!" if language == "Korean" else ""}
-{"- 한국어와 영어를 자연스럽게 섞어서 작성하세요." if language == "Korean + English" else ""}
+{"- 반드시 한국어로 가사를 작성하세요! 한글로만 작성!" if language == "Korean" else ""}
+{"- 请用中文写所有歌词！必须全部用中文！" if language == "Chinese" else ""}
+{"- 日本語で歌詞を書いてください！全て日本語で！" if language == "Japanese" else ""}
+{"- ¡Escribe toda la letra en español! ¡Solo en español!" if language == "Spanish" else ""}
 
 USE HeartMuLa STRUCTURE TAGS:
 [Intro], [Verse], [Prechorus], [Chorus], [Bridge], [Interlude], [Hook], [Outro]
@@ -455,8 +487,14 @@ REMEMBER: ALL lyrics must be in {language}!"""
             return f"❌ {emotional}", draft, structured, "", ""
         
         progress(0.8, desc="✨ 최종 편집...")
+        lang_instruction = {
+            "Korean": "한국어로만 출력하세요!",
+            "Chinese": "用中文输出！",
+            "Japanese": "日本語で出力してください！",
+            "Spanish": "¡Escribe en español!"
+        }.get(language, "")
         final = call_groq(api_key, LYRICS_AGENTS["final_editor"],
-                         f"Output ONLY clean lyrics in {language}. Use [Prechorus] not [Pre Chorus]! {'한국어로만 출력하세요!' if language == 'Korean' else ''}", emotional)
+                         f"Output ONLY clean lyrics in {language}. Use [Prechorus] not [Pre Chorus]! {lang_instruction}", emotional)
         if final.startswith("Error:"):
             return f"❌ {final}", draft, structured, emotional, ""
         
@@ -470,7 +508,7 @@ REMEMBER: ALL lyrics must be in {language}!"""
 
 
 @spaces.GPU(duration=60)
-def quick_lyrics(api_key: str, theme: str, genre: str, mood: str, language: str, vocal_type: str, additional: str):
+def quick_lyrics(api_key: str, theme: str, genre: str, mood: str, language: str, additional: str):
     """빠른 가사 생성"""
     if not api_key or not api_key.strip():
         return "❌ API Key 필요"
@@ -482,12 +520,13 @@ def quick_lyrics(api_key: str, theme: str, genre: str, mood: str, language: str,
 - Genre: {genre}
 - Mood: {mood}
 - Language: {language} ← CRITICAL: Write ALL lyrics in {language}!
-- Vocal: {vocal_type}
 {f'- Special: {additional}' if additional else ''}
 
 ⚠️ LANGUAGE: You MUST write the ENTIRE lyrics in {language}!
-{"반드시 한국어로 작성하세요! 한글 가사만 작성!" if language == "Korean" else ""}
-{"한국어와 영어를 자연스럽게 섞어서 작성하세요." if language == "Korean + English" else ""}
+{"반드시 한국어로 작성하세요! 한글 가사만!" if language == "Korean" else ""}
+{"请用中文写歌词！" if language == "Chinese" else ""}
+{"日本語で書いてください！" if language == "Japanese" else ""}
+{"¡Escribe en español!" if language == "Spanish" else ""}
 
 USE STRUCTURE (8-10 sections):
 [Intro] → [Verse] → [Prechorus] → [Chorus] → [Verse] → [Prechorus] → [Chorus] → [Bridge] → [Chorus] → [Outro]
@@ -510,18 +549,24 @@ Output ONLY clean lyrics.""", prompt)
 
 
 @spaces.GPU(duration=60)
-def generate_tags_ai(api_key: str, genre: str, mood: str, instruments: str):
-    """AI 태그 생성"""
+def generate_tags_ai(api_key: str, genre: str, mood: str, current_tags: str):
+    """AI 태그 생성 - HeartMuLa 형식"""
     if not api_key:
         return "piano,happy,pop"
     
-    prompt = f"""Generate HeartMuLa style tags:
+    prompt = f"""Generate HeartMuLa style tags based on:
 - Genre: {genre}
 - Mood: {mood}
-- Instruments hint: {instruments}
+- Current hint: {current_tags}
+
+Use ONLY these tag categories (from HeartMuLa demo):
+- Instruments: Piano, Keyboard, synthesizer, drum machine, electric guitar, acoustic guitar, drums, Strings, bass
+- Genre: pop, R&B, rock, Ballad, electronic, acoustic
+- Mood: warm, reflection, energetic, soft, Sad, Regret, Longing, meditation, faith, peaceful, Romantic, emotional, hope, hopeful, powerful, epic, driving, happy, uplifting, dreamy
+- Situation: Cafe, wedding, Walking, self-discovery
 
-Output 5-8 comma-separated tags WITHOUT spaces.
-Example: piano,synthesizer,happy,pop,upbeat"""
+Output 4-6 comma-separated tags WITHOUT spaces between tags.
+Example: piano,happy,uplifting,pop,Romantic"""
 
     try:
         result = call_groq(api_key, TAG_AGENT, prompt)
@@ -770,19 +815,15 @@ with gr.Blocks(css=css, title="🎵 SOMA Music Studio - HeartMuLa") as demo:
             
             with gr.Row():
                 lyrics_language = gr.Dropdown(
-                    label="🌍 언어",
-                    choices=["English", "Korean", "Korean + English", "Japanese"],
-                    value="English"
-                )
-                lyrics_vocal_type = gr.Dropdown(
-                    label="🎤 보컬",
-                    choices=["Solo Female", "Solo Male", "Duet", "Group/Choir"],
-                    value="Solo Female"
+                    label="🌍 언어 (Language)",
+                    choices=["English", "Korean", "Chinese", "Japanese", "Spanish"],
+                    value="English",
+                    info="HeartMuLa 공식 지원 5개 언어"
                 )
             
             lyrics_additional = gr.Textbox(
                 label="✨ 추가 지시 (선택)",
-                placeholder="특별 요청...",
+                placeholder="특별 요청: 특정 감정, 스타일 등...",
                 lines=1
             )
             
@@ -866,13 +907,38 @@ with gr.Blocks(css=css, title="🎵 SOMA Music Studio - HeartMuLa") as demo:
             ### 💡 Tips
             - **구조 태그:** [Intro], [Verse], [Prechorus], [Chorus], [Bridge], [Outro]
             - **스타일 태그:** piano,happy,uplifting,pop (콤마 구분, 공백 없음!)
+            - **지원 언어:** 🇺🇸 EN | 🇰🇷 KO | 🇨🇳 ZH | 🇯🇵 JA | 🇪🇸 ES
+            - ⚠️ **보컬 타입(듀엣, 그룹 등)은 지원하지 않습니다!**
             - **Temperature:** 높으면 창의적, 낮으면 일관성
-            - **길이:** 긴 곡은 시간이 더 걸립니다
             """)
     
     # 가이드
     with gr.Accordion("📖 HeartMuLa 가이드", open=False):
         gr.Markdown(f"""
+## ⚠️ HeartMuLa 입력 형식 (중요!)
+
+HeartMuLa는 **2가지 입력만** 받습니다:
+1. **가사 (Lyrics)** - 구조 태그 포함
+2. **스타일 태그 (Tags)** - 콤마 구분
+
+❌ **지원하지 않는 것:**
+- 보컬 타입 선택 (듀엣, 그룹, 코러스 등)
+- 레퍼런스 오디오 (아직 미지원)
+- BPM/템포 직접 지정
+
+---
+
+## 🌍 지원 언어 (5개)
+| 언어 | 코드 | 상태 |
+|------|------|------|
+| 🇺🇸 English | EN | ✅ |
+| 🇰🇷 Korean | KO | ✅ |
+| 🇨🇳 Chinese | ZH | ✅ |
+| 🇯🇵 Japanese | JA | ✅ |
+| 🇪🇸 Spanish | ES | ✅ |
+
+---
+
 {HEARTMULA_TAG_GUIDE}
 
 ---
@@ -885,7 +951,7 @@ with gr.Blocks(css=css, title="🎵 SOMA Music Studio - HeartMuLa") as demo:
 **Paper:** [arXiv](https://arxiv.org/abs/2601.10547) | 
 **Code:** [GitHub](https://github.com/HeartMuLa/heartlib)
 
-*Licensed under Apache 2.0*
+*Licensed under CC BY-NC 4.0 (Non-Commercial)*
         """)
     
     # ========== Event Handlers ==========
@@ -907,14 +973,14 @@ with gr.Blocks(css=css, title="🎵 SOMA Music Studio - HeartMuLa") as demo:
     # 빠른 가사 생성
     quick_btn.click(
         fn=quick_lyrics,
-        inputs=[groq_key, theme_input, lyrics_genre, lyrics_mood, lyrics_language, lyrics_vocal_type, lyrics_additional],
+        inputs=[groq_key, theme_input, lyrics_genre, lyrics_mood, lyrics_language, lyrics_additional],
         outputs=[final_lyrics]
     )
     
     # SOMA 가사 생성
     soma_lyrics_btn.click(
         fn=generate_lyrics_soma,
-        inputs=[groq_key, theme_input, lyrics_genre, lyrics_mood, lyrics_language, lyrics_vocal_type, lyrics_additional],
+        inputs=[groq_key, theme_input, lyrics_genre, lyrics_mood, lyrics_language, lyrics_additional],
         outputs=[lyrics_status, step1_out, step2_out, step3_out, step4_out]
     ).then(
         fn=lambda x: x,