refactor for video output

app.py

@@ -4,6 +4,8 @@ import matplotlib.pyplot as plt
 from matplotlib.animation import FuncAnimation
 import io
 import librosa
+import tempfile
+import os
 
 def extract_waveform_animation(audio_file, duration):
     # Load audio file
@@ -32,12 +34,12 @@ def extract_waveform_animation(audio_file, duration):
     # Create the animation
     ani = FuncAnimation(fig, update, frames=np.arange(0, num_frames, sr // 10), init_func=init, blit=True)
     
-    # Save the animation to a buffer
-    buf = io.BytesIO()
-    ani.save(buf, format='gif')
-    buf.seek(0)
+    # Save the animation to a temporary file
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmpfile:
+        ani.save(tmpfile.name, writer='ffmpeg', fps=30)
+        video_path = tmpfile.name
     
-    return buf
+    return video_path
 
 # Define the Gradio interface
 iface = gr.Interface(
@@ -46,8 +48,8 @@ iface = gr.Interface(
         gr.Audio(type="filepath"),
         gr.Number(label="Duration (seconds)", value=5)
     ],
-    outputs=gr.Image(type="file", format="gif"),
-    description="Upload an audio file to extract an animation from its waveform. Specify the duration of the animation."
+    outputs=gr.Video(),
+    description="Upload an audio file to extract a video animation from its waveform. Specify the duration of the animation."
 )
 
 # Launch the app