Upload app.py with huggingface_hub

app.py

@@ -9,6 +9,7 @@ import os
 import sys
 import pickle
 import tempfile
+import base64
 from huggingface_hub import hf_hub_download
 
 # Add text2midi model to path
@@ -124,21 +125,20 @@ 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."""
+    """Generate music from text prompt. Returns audio as data URI for API compatibility."""
     if not prompt or not prompt.strip():
         return None
 
     midi_path = None
+    wav_path = None
 
     try:
-        # Create temporary MIDI file
+        # Create temporary files
         midi_fd, midi_path = tempfile.mkstemp(suffix='.mid')
         os.close(midi_fd)
 
-        # Create WAV in Gradio's temp directory for proper file serving
-        gradio_temp = os.path.join(tempfile.gettempdir(), "gradio")
-        os.makedirs(gradio_temp, exist_ok=True)
-        wav_path = os.path.join(gradio_temp, f"audio_{os.urandom(8).hex()}.wav")
+        wav_fd, wav_path = tempfile.mkstemp(suffix='.wav')
+        os.close(wav_fd)
 
         # Generate MIDI (reduced max_len for faster testing)
         if MODEL_LOADED:
@@ -155,8 +155,12 @@ def generate_music(prompt: str):
 
         # Convert to WAV
         if SOUNDFONT_PATH and midi_to_wav(midi_path, wav_path):
-            # Return file path - don't delete, let Gradio serve it
-            return wav_path
+            # Read and encode as base64 data URI
+            with open(wav_path, 'rb') as f:
+                audio_bytes = f.read()
+            audio_b64 = base64.b64encode(audio_bytes).decode('utf-8')
+            # Return as data URI - works with Gradio Audio component
+            return f"data:audio/wav;base64,{audio_b64}"
         else:
             return None
 
@@ -166,12 +170,17 @@ def generate_music(prompt: str):
         return None
 
     finally:
-        # Only clean up MIDI file, keep WAV for Gradio to serve
+        # Clean up temp files
         if midi_path and os.path.exists(midi_path):
             try:
                 os.unlink(midi_path)
             except:
                 pass
+        if wav_path and os.path.exists(wav_path):
+            try:
+                os.unlink(wav_path)
+            except:
+                pass
 
 # Create simple Gradio Interface
 demo = gr.Interface(