Add detailed progress updates and extend timeouts for music generation

- Increase GPU timeout to 5 minutes for complex generation
- Add background thread for progress updates every 15 seconds
- Show detailed stages: Stage 1 (7B model), Stage 2 (1B model), audio decoding
- Extend subprocess timeout to 30 minutes for safety
- Add generation settings and time estimates in logs
- Improve user feedback during long generation process

app.py

@@ -285,7 +285,7 @@ def validate_file_structure():
     print("✅ All required files found")
     return True
 
-@spaces.GPU  # Decorator for GPU access in Spaces
+@spaces.GPU(duration=300)  # Increase GPU timeout to 5 minutes
 def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progress()) -> str:
     """
     Generate music using YuE model with high-performance Spaces configuration
@@ -373,12 +373,15 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
         if not flash_attn_available:
             cmd.append("--sdpa")
         
-        progress(0.3, desc="Starting music generation (Stage 1)...")
+        # More detailed progress updates
+        progress(0.1, desc="🚀 Initializing models...")
+        progress(0.15, desc="📝 Processing lyrics...")
+        progress(0.2, desc="🎵 Starting Stage 1 (7B model generation)...")
         
-        # Run with generous timeout for high-quality generation
         print("🎵 Starting high-quality music generation...")
+        print(f"📊 Generation settings: {run_n_segments} segments, {max_new_tokens} tokens, 30s audio")
+        print(f"⏱️ Estimated time: 2-4 minutes for high-quality generation")
         print(f"Working directory: {os.getcwd()}")
-        print(f"Infer script path: {infer_script_path}")
         print(f"Command: {' '.join(cmd)}")
         
         # Change to YuEGP/inference directory for execution
@@ -388,13 +391,40 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
         try:
             os.chdir(inference_dir)
             print(f"Changed to inference directory: {inference_dir}")
-            
-            # Update command to use relative path since we're in the inference directory
             cmd[1] = "infer.py"
             
-            # Run with real-time output for debugging
+            progress(0.25, desc="🔥 Stage 1: Running 7B parameter model...")
+            
+            # Start the subprocess
+            import threading
+            import time
+            
+            def update_progress_periodically():
+                """Update progress periodically to show the process is alive"""
+                stages = [
+                    (30, "🧠 Stage 1: Generating musical concepts..."),
+                    (45, "🎼 Stage 1: Creating melody patterns..."),
+                    (60, "🎹 Stage 1: Composing harmony structure..."),
+                    (70, "⚡ Starting Stage 2: Refining with 1B model..."),
+                    (80, "🎵 Stage 2: Adding musical details..."),
+                    (85, "🎶 Stage 2: Finalizing composition..."),
+                    (90, "🔊 Decoding to audio format...")
+                ]
+                
+                for percent, desc in stages:
+                    time.sleep(15)  # Update every 15 seconds
+                    try:
+                        progress(percent/100, desc=desc)
+                        print(f"⏳ {desc}")
+                    except:
+                        break  # Stop if main process finished
+            
+            # Start progress updates in background
+            progress_thread = threading.Thread(target=update_progress_periodically, daemon=True)
+            progress_thread.start()
+            
             print(f"🚀 Executing command: {' '.join(cmd)}")
-            result = subprocess.run(cmd, capture_output=True, text=True, timeout=1200)  # 20 minutes
+            result = subprocess.run(cmd, capture_output=True, text=True, timeout=1800)  # 30 minutes timeout
             
             # Print stdout and stderr for debugging
             if result.stdout:
@@ -402,11 +432,11 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
             if result.stderr:
                 print(f"⚠️ Command stderr:\n{result.stderr}")
             print(f"📊 Return code: {result.returncode}")
+            
         finally:
-            # Always restore original working directory
             os.chdir(original_cwd)
         
-        progress(0.9, desc="Finalizing audio output...")
+        progress(0.95, desc="🎉 Processing completed, finalizing output...")
         
         # Clean up input files
         os.unlink(genre_file_path)