Update app.py

app.py

@@ -1,57 +1,75 @@
 import gradio as gr
 import os
+import time
 from basic_pitch.inference import predict
 
-def convert_audio_to_midi(audio_filepath):
-    """
-    Takes an audio file path, runs Basic Pitch, and saves/returns a MIDI file.
-    """
+# Configuration
+OUTPUT_FOLDER = "midi_outputs"
+CLEANUP_THRESHOLD_SECONDS = 60  # 1 minute
+ 
+# Ensure the output directory exists
+if not os.path.exists(OUTPUT_FOLDER):
+    os.makedirs(OUTPUT_FOLDER) 
+ 
+def cleanup_old_files():
+    """Removes MIDI files older than 1 minute from the output folder."""
+    now = time.time()
+    for filename in os.listdir(OUTPUT_FOLDER):
+        file_path = os.path.join(OUTPUT_FOLDER, filename)
+        # Check if it's a file and if it's older than our threshold
+        if os.path.isfile(file_path) and filename.endswith(".mid"):
+            file_age = now - os.path.getmtime(file_path)
+            if file_age > CLEANUP_THRESHOLD_SECONDS:
+                try:
+                    os.remove(file_path)
+                    print(f"Cleaned up old file: {filename}")
+                except Exception as e:
+                    print(f"Failed to delete {filename}: {e}")
+
+def convert_audio_to_midi(audio_filepath): 
     if audio_filepath is None:
         return None
+    
+    cleanup_old_files()
         
     try:
-        # Run the basic pitch model on the audio file
-        # predict() returns model_output, midi_data, and note_events
+        # predict returns (model_output, midi_data, note_events)
         _, midi_data, _ = predict(audio_filepath)
         
-        # Define a temporary output path for the generated MIDI file
-        output_midi_path = "output.mid"
+        timestamp = int(time.time())
+        filename = f"output_{timestamp}.mid"
+        output_midi_path = os.path.join(OUTPUT_FOLDER, filename)
         
-        # Save the MIDI data to the file
         midi_data.write(output_midi_path)
         
-        return output_midi_path
-        
+        return output_midi_path # Gradio handles making this a downloadable URL
     except Exception as e:
-        print(f"Error during conversion: {e}")
+        print(f"Error: {e}")
         return None
 
 # Build the Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
-        # 🎵 Basic Pitch: Audio to MIDI Converter
-        Upload an audio file (vocals, guitar, piano, etc.) and convert it into a MIDI file using Spotify's open-source **Basic Pitch** machine learning model.
+        # 🎵 MIDI Converter
+        Upload audio to convert to MIDI. Files are saved to the `/midi_outputs` folder.
+        **Note:** Outputs are automatically deleted after 1 minute to save space.
         """
     )
     
     with gr.Row():
         with gr.Column():
-            # User uploads audio; Gradio passes the file path to our function
-            audio_input = gr.Audio(type="filepath", label="Upload Audio File (WAV, MP3, etc.)")
+            audio_input = gr.Audio(type="filepath", label="Upload Audio File")
             convert_btn = gr.Button("Convert to MIDI", variant="primary")
             
         with gr.Column():
-            # The output will be a downloadable file
-            midi_output = gr.File(label="Download Generated MIDI")
+            midi_output = gr.File(label="Download MIDI (Expires in 60s)")
 
-    # Link the button click to the conversion function
     convert_btn.click(
         fn=convert_audio_to_midi,
         inputs=audio_input,
         outputs=midi_output
     )
 
-# Launch the app
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch()