add GPU processing time for full model

app.py

@@ -285,7 +285,7 @@ def validate_file_structure():
     print("✅ All required files found")
     return True
 
-@spaces.GPU(duration=300)  # Increase GPU timeout to 5 minutes
+@spaces.GPU(duration=900)  # Increase GPU timeout to 15 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
@@ -403,32 +403,89 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
             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...")
+            def parse_output_and_update_progress(process):
+                """Parse subprocess output in real-time and update progress accordingly"""
+                stage1_messages = [
+                    "🧠 Stage 1: Generating musical concepts...",
+                    "🎼 Stage 1: Creating melody patterns...",
+                    "🎹 Stage 1: Composing harmony structure..."
+                ]
+                stage2_messages = [
+                    "⚡ Starting Stage 2: Refining with 1B model...",
+                    "🎵 Stage 2: Adding musical details...",
+                    "🎶 Stage 2: Finalizing composition..."
                 ]
                 
-                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()
+                stage1_progress = [0.3, 0.45, 0.6]
+                stage2_progress = [0.7, 0.8, 0.85]
+                
+                current_stage = 1
+                stage1_step = 0
+                stage2_step = 0
+                
+                output_lines = []
+                
+                try:
+                    while True:
+                        line = process.stdout.readline()
+                        if not line:
+                            break
+                            
+                        line = line.strip()
+                        output_lines.append(line)
+                        print(line)  # Still print for debugging
+                        
+                        # Check for stage transitions based on actual output
+                        if "Stage 2 inference..." in line:
+                            current_stage = 2
+                            stage2_step = 0
+                            progress(0.7, desc=stage2_messages[0])
+                            print(f"⏳ {stage2_messages[0]}")
+                        
+                        elif "Stage 2 DONE" in line:
+                            progress(0.9, desc="🔊 Decoding to audio format...")
+                            print("⏳ 🔊 Decoding to audio format...")
+                        
+                        # Update Stage 1 progress periodically
+                        elif current_stage == 1 and stage1_step < len(stage1_messages):
+                            # Update Stage 1 progress every 15 seconds or on specific markers
+                            if stage1_step < len(stage1_progress):
+                                progress(stage1_progress[stage1_step], desc=stage1_messages[stage1_step])
+                                print(f"⏳ {stage1_messages[stage1_step]}")
+                                stage1_step += 1
+                        
+                        # Update Stage 2 progress periodically
+                        elif current_stage == 2 and stage2_step < len(stage2_messages) - 1:
+                            stage2_step += 1
+                            if stage2_step < len(stage2_progress):
+                                progress(stage2_progress[stage2_step], desc=stage2_messages[stage2_step])
+                                print(f"⏳ {stage2_messages[stage2_step]}")
+                
+                except Exception as e:
+                    print(f"Progress parsing error: {e}")
+                
+                return '\n'.join(output_lines)
             
             print(f"🚀 Executing command: {' '.join(cmd)}")
-            result = subprocess.run(cmd, capture_output=True, text=True, timeout=1800)  # 30 minutes timeout
+            
+            # Use Popen for real-time output processing
+            process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, 
+                                     text=True, bufsize=1, universal_newlines=True)
+            
+            # Parse output in real-time
+            stdout_output = parse_output_and_update_progress(process)
+            
+            # Wait for process to complete and get return code
+            return_code = process.wait()
+            
+            # Create result object similar to subprocess.run
+            class Result:
+                def __init__(self, returncode, stdout, stderr=""):
+                    self.returncode = returncode
+                    self.stdout = stdout
+                    self.stderr = stderr
+            
+            result = Result(return_code, stdout_output)
             
             # Print stdout and stderr for debugging
             if result.stdout: