Update app.py

app.py

@@ -82,15 +82,16 @@ 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 the current plot as an image
-        buf = io.BytesIO()
-        plt.savefig(buf, format='png', bbox_inches='tight')
-        buf.seek(0)
-        image = np.array(Image.open(buf))
+        # Convert the Matplotlib figure to a PIL Image
+        canvas = fig.canvas
+        canvas.draw()
+        image = Image.frombytes('RGB', 
+                                canvas.get_width_height(),
+                                canvas.tostring_rgb())
         last_image = image
 
         # Yield the current image and a placeholder for the final animation
-        yield (image, None)
+        yield (last_image, None)
 
     # Generate and yield images for each frame
     for frame in range(frames):
@@ -101,9 +102,6 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_
         anim = animation.FuncAnimation(fig, animate, frames=frames, interval=5, fargs=(coefs, np.linspace(0, 1, num=frames)))
         anim.save(temp_file.name, fps=15)
 
-    # Remove the temporary file
-    # os.remove(temp_file.name)
-
     yield (last_image, temp_file.name)
 
 # Gradio interface setup