Update app.py

app.py

@@ -6,8 +6,10 @@ import numpy as np
 from scipy.integrate import quad_vec
 from math import tau
 import os
+from PIL import Image
+import io
 
-def fourier_transform_drawing(input_image, output_animation, frames, coefficients):
+def fourier_transform_drawing(input_image, frames, coefficients):
     # Convert input_image to an OpenCV image
     input_image = np.array(input_image)
     img = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)
@@ -78,22 +80,29 @@ def fourier_transform_drawing(input_image, output_animation, frames, coefficient
     time = np.linspace(0, drawing_time, num=frames)    
     anim = animation.FuncAnimation(fig, animate, frames=frames, interval=5, fargs=(coefs, time)) 
 
+    output_animation = "output.mp4"
+    
     anim.save(output_animation, fps=15)
     plt.close(fig)
 
-    return output_animation
+    # 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")
+
+    # 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
 
 # Gradio interface
 interface = gr.Interface(
     fn=fourier_transform_drawing,
     inputs=[
-        
         gr.Image(label="Input Image", sources=['upload'], type="pil"),
-        gr.Textbox(label="Output Animation Path"),
         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="file",
+    outputs=gr.Image(format="gif"),
     title="Fourier Transform Drawing",
     description="Upload an image and generate a Fourier Transform drawing animation."
 )