Update app.py

app.py

@@ -4,14 +4,12 @@ from moviepy.editor import AudioFileClip, ImageClip, CompositeVideoClip
 import tempfile
 import os
 
-def create_waveform_video(image, audio):
-    # Save uploaded files temporarily
+def create_waveform_video(image, audio_path):
+    # Save uploaded image temporarily
     image_path = tempfile.mktemp(suffix=".png")
-    audio_path = tempfile.mktemp(suffix=".mp3")
     video_path = tempfile.mktemp(suffix=".mp4")
     
     image.save(image_path)
-    audio.save(audio_path)
     
     # Load audio
     audio_clip = AudioFileClip(audio_path)
@@ -27,17 +25,12 @@ def create_waveform_video(image, audio):
     # Normalize waveform for overlay
     waveform = np.interp(waveform, (waveform.min(), waveform.max()), (-1, 1))
     
-    # Overlay waveform on image (optional customization: size and position of waveform)
-    waveform_clip = ImageClip(image_path).set_duration(audio_clip.duration)
-    waveform_clip = waveform_clip.fx(CompositeVideoClip, [[waveform, 'center', 0.8]]) 
-
     # Combine image and audio into one video
-    final_clip = CompositeVideoClip([img_clip, waveform_clip.set_audio(audio_clip)])
+    final_clip = img_clip.set_audio(audio_clip)
     final_clip.write_videofile(video_path, codec="libx264", fps=24)
     
     # Cleanup temporary files
     os.remove(image_path)
-    os.remove(audio_path)
     
     return video_path
 
@@ -45,7 +38,7 @@ iface = gr.Interface(
     fn=create_waveform_video,
     inputs=[
         gr.Image(type="pil", label="Upload Image"),
-        gr.Audio(type="file", label="Upload Audio")
+        gr.Audio(type="filepath", label="Upload Audio")
     ],
     outputs=gr.Video(label="Generated Video"),
     title="Image + Audio to Video with Waveform Overlay"