1. modify generated files search logic 2.modify time to 12 mins for H200

app.py

@@ -287,7 +287,7 @@ def validate_file_structure():
     print("✅ All required files found")
     return True
 
-@spaces.GPU(duration=1200)  # Increase GPU timeout to 20 minutes for other less powerful GPUs
+@spaces.GPU(duration=720)
 def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progress()) -> str:
     """
     Generate music using YuE model with high-performance Spaces configuration
@@ -506,14 +506,25 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
         os.unlink(lyrics_file_path)
         
         if result.returncode == 0:
-            # Find generated audio file
+            # Find generated audio file - prioritize mixed audio from vocoder/mix directory
             import glob
+
+            # First look for the final mixed audio in vocoder/mix
+            mixed_files = glob.glob(os.path.join(output_dir, "vocoder/mix/*_mixed.mp3"))
+            if mixed_files:
+                progress(1.0, desc="Music generation complete!")
+                print(f"✅ Found mixed audio file: {mixed_files[0]}")
+                return mixed_files[0]
+
+            # Fallback to any MP3 file
             audio_files = glob.glob(os.path.join(output_dir, "**/*.mp3"), recursive=True)
-            
             if audio_files:
                 progress(1.0, desc="Music generation complete!")
+                print(f"✅ Found audio file: {audio_files[0]}")
                 return audio_files[0]  # Return path to generated audio
             else:
+                print(f"❌ No audio files found in {output_dir}")
+                print(f"Directory contents: {os.listdir(output_dir) if os.path.exists(output_dir) else 'Directory not found'}")
                 return "Music generation completed but no audio file found."
         else:
             error_msg = f"Return code: {result.returncode}\n"