Update app.py

app.py

@@ -77,22 +77,30 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_
         draw_y.append(center[1])
         drawing.set_data(draw_x[:i+1], draw_y[:i+1])
 
-        # Capture and yield the current plot as an image
+
+        # Capture the current plot as an image
         buf = io.BytesIO()
         plt.savefig(buf, format='png', bbox_inches='tight')
         buf.seek(0)
-        yield np.array(Image.open(buf))
+        image = np.array(Image.open(buf))
+
+        # Yield the current image and a placeholder for the final animation
+        yield (image, None)
 
-    # Generate and yield images
+    # Generate and yield images for each frame
     for frame in range(frames):
         yield from animate(frame, coefs, np.linspace(0, 1, num=frames))
 
-    # Generate final animation
+    # Generate final animation as GIF
     anim = animation.FuncAnimation(fig, animate, frames=frames, interval=5, fargs=(coefs, np.linspace(0, 1, num=frames)))
-    buf = io.BytesIO()
-    anim.save(buf, format='gif', fps=15)
-    buf.seek(0)
-    yield buf
+    gif_buf = io.BytesIO()
+    anim.save(gif_buf, format='gif', fps=15)
+    gif_buf.seek(0)
+    final_gif = Image.open(gif_buf)
+    final_gif = np.array(final_gif.convert('RGB'))
+
+    # Yield the final GIF in place of the last frame
+    yield (image, final_gif)
 
 # Gradio interface setup
 interface = gr.Interface(
@@ -107,7 +115,7 @@ interface = gr.Interface(
         gr.Number(value=1000, label="Number of Points for Integration", precision=0),
         gr.Slider(minimum=50, maximum=500, value=80, label="Theta Points for Animation")
     ],
-    outputs=["image", "file"],
+    outputs=["image", "image"],
     title="Fourier Transform Drawing",
     description="Upload an image and generate a Fourier Transform drawing animation.",
 )