1.use different instrument components for different genre 2.modify estimated time

app.py

@@ -190,8 +190,29 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, theme: str, 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"{genre} electronic {mood} {theme} female airy vocal bright")
+            # Add specific instruments based on genre for better itrack generation
+            if genre == "pop":
+                instruments = "piano electric guitar bass drums synthesizer"
+            elif genre == "rock":
+                instruments = "electric guitar bass drums guitar"
+            elif genre == "jazz":
+                instruments = "piano bass drums saxophone"
+            elif genre == "classical":
+                instruments = "piano violin orchestra"
+            elif genre == "electronic":
+                instruments = "synthesizer drum machine bass synth"
+            elif genre == "folk":
+                instruments = "acoustic guitar piano"
+            elif genre == "r&b":
+                instruments = "piano bass drums electric guitar"
+            elif genre == "country":
+                instruments = "acoustic guitar bass drums"
+            elif genre == "hip-hop":
+                instruments = "drum machine bass synthesizer"
+            else:
+                instruments = "piano electric guitar bass drums"  # default
+
+            genre_file.write(f"{genre} {instruments} {mood} {theme} female airy vocal bright")
             genre_file_path = genre_file.name
         
         # Convert lyrics format for YuE compatibility
@@ -275,7 +296,7 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, theme: str, progre
         
         print("🎵 Starting high-quality music generation...")
         print(f"📊 Generation settings: {run_n_segments} segments, {max_new_tokens} tokens, 30s audio")
-        print(f"⏱️ Estimated time: 8-9 minutes for high-quality generation")
+        print(f"⏱️ Estimated time: 10-13 minutes for high-quality generation")
         print(f"Working directory: {os.getcwd()}")
         print(f"Command: {' '.join(cmd)}")
 
@@ -506,7 +527,7 @@ def build_interface():
         gr.Markdown("""
         <div class="performance-notice">
         🚀 <strong>High-Performance Mode:</strong> Running on Spaces GPU with optimized settings for best quality.
-        Generation time: ~8-9 minutes for professional-grade music with vocals and instruments.
+        Generation time: ~10-13 minutes for professional-grade music with vocals and instruments.
         </div>
         """)