Update app.py

app.py

@@ -80,28 +80,23 @@ def fourier_transform_drawing(input_image, frames, coefficients):
     time = np.linspace(0, drawing_time, num=frames)    
     anim = animation.FuncAnimation(fig, animate, frames=frames, interval=5, fargs=(coefs, time)) 
 
-    output_animation = "output.mp4"
-    
     # Save the animation as an MP4 file
+    output_animation = "output.mp4"
     anim.save(output_animation, fps=15)
     plt.close(fig)
 
-    # Convert the mp4 file to a GIF for display in Gradio
-    gif_output = "output.gif"
-    os.system(f"ffmpeg -i {output_animation} -vf 'fps=10,scale=320:-1:flags=lanczos' -c:v gif -loop 0 {gif_output}")
-
-    # Instead of reading the GIF into a PIL Image, return the path to the GIF file
-    return gif_output
+    # Return the path to the MP4 file
+    return output_animation
 
 # Gradio interface
 interface = gr.Interface(
     fn=fourier_transform_drawing,
     inputs=[
         gr.Image(label="Input Image", sources=['upload'], type="pil"),
-        gr.Slider(minimum=10, maximum=500, value=300, label="Number of Frames"),
-        gr.Slider(minimum=10, maximum=500, value=300, label="Number of Coefficients")
+        gr.Slider(minimum=10, maximum=500, value=100, label="Number of Frames"),
+        gr.Slider(minimum=10, maximum=500, value=100, label="Number of Coefficients")
     ],
-    outputs=gr.File(),
+    outputs=gr.Video(),
     title="Fourier Transform Drawing",
     description="Upload an image and generate a Fourier Transform drawing animation."
 )