Update app.py

app.py

@@ -66,7 +66,10 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_
     theta = np.linspace(0, tau, theta_points)
     coefs_static = [(np.linalg.norm(c), fr) for c, fr in coefs]
 
+    last_image = None 
+    
     def animate(i, coefs, time):
+        nonlocal last_image
         center = (0, 0)
         for idx, (r, fr) in enumerate(coefs_static):
             c_dynamic = coefs[idx][0] * np.exp(1j * (fr * tau * time[i]))
@@ -79,12 +82,12 @@ 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))
+        last_image = image
 
         # Yield the current image and a placeholder for the final animation
         yield (image, None)
@@ -106,7 +109,7 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_
     os.remove(temp_file.name)
 
     # Yield the final GIF in place of the last frame
-    yield (image, final_gif)
+    yield (last_image, final_gif)
 
 # Gradio interface setup
 interface = gr.Interface(