Update app.py

app.py

@@ -9,7 +9,6 @@ import os
 import sys
 import pickle
 import tempfile
-import shutil
 from huggingface_hub import hf_hub_download
 
 # Add text2midi model to path
@@ -125,64 +124,52 @@ def midi_to_wav(midi_path: str, wav_path: str, sample_rate: int = 44100) -> bool
     return os.path.exists(wav_path)
 
 def generate_music(prompt: str):
-    """Generate music from text prompt. Returns audio file path or tuple (sample_rate, audio_data)."""
+    """Generate music from text prompt. Returns audio file path."""
     if not prompt or not prompt.strip():
         return None
 
-    midi_path = None
-    wav_path = None
-    
     try:
-        # Create persistent output directory
-        output_dir = "output"
-        os.makedirs(output_dir, exist_ok=True)
-        
-        # Create files with proper names in persistent directory
-        import time
-        timestamp = int(time.time() * 1000)
-        midi_path = os.path.join(output_dir, f"music_{timestamp}.mid")
-        wav_path = os.path.join(output_dir, f"music_{timestamp}.wav")
-
-        # Generate MIDI
-        if MODEL_LOADED:
-            generate_midi_with_model(prompt, midi_path, max_len=512, temperature=0.9)
-        else:
-            from midiutil import MIDIFile
-            midi = MIDIFile(1)
-            midi.addTempo(0, 0, 120)
-            notes = [60, 62, 64, 65, 67, 69, 71, 72]
-            for i, note in enumerate(notes[:min(len(prompt.split()), 8)]):
-                midi.addNote(0, 0, note, i, 1, 100)
-            with open(midi_path, "wb") as f:
-                midi.writeFile(f)
-
-        # Convert to WAV
-        if SOUNDFONT_PATH and midi_to_wav(midi_path, wav_path):
-            # Verify the file exists and has content
-            if os.path.exists(wav_path) and os.path.getsize(wav_path) > 0:
-                print(f"Generated audio file: {wav_path} ({os.path.getsize(wav_path)} bytes)")
+        # Create temporary files
+        midi_file = tempfile.NamedTemporaryFile(suffix='.mid', delete=False)
+        midi_path = midi_file.name
+        midi_file.close()
+
+        wav_file = tempfile.NamedTemporaryFile(suffix='.wav', delete=False)
+        wav_path = wav_file.name
+        wav_file.close()
+
+        try:
+            # Generate MIDI
+            if MODEL_LOADED:
+                generate_midi_with_model(prompt, midi_path, max_len=512, temperature=0.9)
+            else:
+                from midiutil import MIDIFile
+                midi = MIDIFile(1)
+                midi.addTempo(0, 0, 120)
+                notes = [60, 62, 64, 65, 67, 69, 71, 72]
+                for i, note in enumerate(notes[:min(len(prompt.split()), 8)]):
+                    midi.addNote(0, 0, note, i, 1, 100)
+                with open(midi_path, "wb") as f:
+                    midi.writeFile(f)
+
+            # Convert to WAV
+            if SOUNDFONT_PATH and midi_to_wav(midi_path, wav_path):
                 return wav_path
             else:
-                print(f"WAV file is empty or doesn't exist")
                 return None
-        else:
-            print("Failed to convert MIDI to WAV")
-            return None
 
-    except Exception as e:
-        print(f"Error generating music: {e}")
-        import traceback
-        traceback.print_exc()
-        return None
-    finally:
-        # Clean up MIDI file but keep WAV for Gradio to serve
-        if midi_path and os.path.exists(midi_path):
+        finally:
             try:
                 os.unlink(midi_path)
             except:
                 pass
 
-# Create simple Gradio Interface
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return None
+
+# Create simple Gradio Interface (avoids schema generation bugs in gr.Blocks)
 demo = gr.Interface(
     fn=generate_music,
     inputs=gr.Textbox(
@@ -199,10 +186,8 @@ demo = gr.Interface(
         ["A slow emotional classical piece with violin"],
         ["Epic cinematic soundtrack with dark atmosphere"],
     ],
-    allow_flagging="never",
-    cache_examples=False
+    allow_flagging="never"
 )
 
 # Launch
-if __name__ == "__main__":
-    demo.launch(show_api=True)
+demo.launch(show_api=True)