Update app.py

app.py

@@ -82,17 +82,16 @@ def fourier_transform_drawing(input_image, frames, coefficients):
 
     output_animation = "output.mp4"
     
+    # Save the animation as an MP4 file
     anim.save(output_animation, fps=15)
     plt.close(fig)
 
     # Convert the mp4 file to a GIF for display in Gradio
-    os.system(f"ffmpeg -i {output_animation} -vf 'fps=10,scale=320:-1:flags=lanczos' -c:v gif -loop 0 output.gif")
+    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}")
 
-    # Read the GIF and convert to PIL Image
-    with open("output.gif", 'rb') as f:
-        gif_image = Image.open(io.BytesIO(f.read()))
-
-    return gif_image
+    # Instead of reading the GIF into a PIL Image, return the path to the GIF file
+    return gif_output
 
 # Gradio interface
 interface = gr.Interface(
@@ -102,7 +101,7 @@ interface = gr.Interface(
         gr.Slider(minimum=10, maximum=500, value=300, label="Number of Frames"),
         gr.Slider(minimum=10, maximum=500, value=300, label="Number of Coefficients")
     ],
-    outputs=gr.Image(),
+    outputs=gr.File(),
     title="Fourier Transform Drawing",
     description="Upload an image and generate a Fourier Transform drawing animation."
 )