modify lyric and genre according to YuE's README

.gitignore

@@ -114,3 +114,6 @@ backup/
 
 # test files
 tests/
+
+# log
+logs.txt

app.py

@@ -175,7 +175,7 @@ def validate_file_structure():
     return True
 
 @spaces.GPU(duration=1200) # H200 on ZeroGPU is free for 25mins, for compatibility on A10G large and L40s
-def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progress()) -> str:
+def generate_music_spaces(lyrics: str, genre: str, mood: str, theme: str, progress=gr.Progress()) -> str:
     """
     Generate music using YuE model with high-performance Spaces configuration
     """
@@ -191,7 +191,7 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
         # Create temporary files
         with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as genre_file:
             # genre_file.write(f"instrumental,{genre},{mood},male vocals")
-            genre_file.write(f"instrumental {genre} {mood} female vocals airy vocal")
+            genre_file.write(f"{genre} electronic {mood} {theme} female airy vocal bright")
             genre_file_path = genre_file.name
         
         # Convert lyrics format for YuE compatibility
@@ -201,11 +201,11 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
         # Extract only the actual lyrics content, removing AI commentary
         formatted_lyrics_for_yue = formatted_lyrics
         
-        # Convert **VERSE 1** to [VERSE], **CHORUS** to [CHORUS], etc.
-        formatted_lyrics_for_yue = re.sub(r'\*\*(VERSE\s*\d*)\*\*', r'[\1]', formatted_lyrics_for_yue)
-        formatted_lyrics_for_yue = re.sub(r'\*\*(CHORUS)\*\*', r'[\1]', formatted_lyrics_for_yue)
-        formatted_lyrics_for_yue = re.sub(r'\*\*(BRIDGE)\*\*', r'[\1]', formatted_lyrics_for_yue)
-        formatted_lyrics_for_yue = re.sub(r'\*\*(OUTRO)\*\*', r'[\1]', formatted_lyrics_for_yue)
+        # Convert **VERSE 1** to [verse], **CHORUS** to [chorus], etc. (lowercase as expected by YuE)
+        formatted_lyrics_for_yue = re.sub(r'\*\*(VERSE\s*\d*)\*\*', r'[verse]', formatted_lyrics_for_yue)
+        formatted_lyrics_for_yue = re.sub(r'\*\*(CHORUS)\*\*', r'[chorus]', formatted_lyrics_for_yue)
+        formatted_lyrics_for_yue = re.sub(r'\*\*(BRIDGE)\*\*', r'[bridge]', formatted_lyrics_for_yue)
+        formatted_lyrics_for_yue = re.sub(r'\*\*(OUTRO)\*\*', r'[outro]', formatted_lyrics_for_yue)
         
         # Remove AI commentary (lines that don't contain actual lyrics)
         lines = formatted_lyrics_for_yue.split('\n')
@@ -214,16 +214,19 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
         
         for line in lines:
             line = line.strip()
-            # Start collecting from first section marker
-            if re.match(r'\[(VERSE|CHORUS|BRIDGE|OUTRO)', line):
+            # Start collecting from first section marker (lowercase as expected by YuE)
+            if re.match(r'\[(verse|chorus|bridge|outro)', line, re.IGNORECASE):
                 in_song = True
             # Stop at AI commentary
             if in_song and line and not line.startswith('[') and any(phrase in line.lower() for phrase in ['how do you like', 'would you like', 'let me know', 'take a look']):
                 break
             if in_song:
                 clean_lines.append(line)
-        
+
+        # Join with proper section separators as required by YuE (double newlines between sections)
         formatted_lyrics_for_yue = '\n'.join(clean_lines).strip()
+        # Add double newlines between sections for YuE format
+        formatted_lyrics_for_yue = re.sub(r'\n(\[(?:verse|chorus|bridge|outro)\])', r'\n\n\1', formatted_lyrics_for_yue)
         
         print(f"🐛 DEBUG - Original lyrics length: {len(formatted_lyrics)}")
         print(f"🐛 DEBUG - Converted lyrics for YuE: '{formatted_lyrics_for_yue}'")
@@ -516,7 +519,7 @@ def build_interface():
                     gr.Markdown("### 🎛️ Music Settings")
                     with gr.Row():
                         genre = gr.Dropdown(
-                            choices=["pop", "rock", "jazz", "classical", "electronic", "folk", "r&b", "country", "hip-hop"],
+                            choices=["pop", "rock", "jazz", "classical", "folk", "r&b", "country", "hip-hop"],
                             value="pop", label="Genre"
                         )
                         mood = gr.Dropdown(
@@ -637,7 +640,7 @@ def build_interface():
         # Music generation with progress
         generate_btn.click(
             fn=generate_music_spaces,
-            inputs=[lyrics_display, genre, mood],
+            inputs=[lyrics_display, genre, mood, theme],
             outputs=[music_output],
             queue=True,
             show_progress=True