Update app.py

app.py

@@ -23,7 +23,7 @@ def process_image(input_image, img_size, blur_kernel_size, desired_range):
     x_range, y_range = np.max(xs) - np.min(xs), np.max(ys) - np.min(ys)
     scale_x, scale_y = desired_range / x_range, desired_range / y_range
     xs = (xs - np.mean(xs)) * scale_x
-    ys = (-ys + np.mean(ys)) * scale_y
+    ys = -(ys - np.mean(ys)) * scale_y
 
     return xs, ys
 
@@ -81,6 +81,14 @@ def animate(frame, coefs, time, fig, ax, background, circles, circle_lines, draw
     ax.draw_artist(drawing)
 
     fig.canvas.blit(ax.bbox)
+
+    # Capture the frame as an image
+    fig.canvas.draw()
+    image = np.frombuffer(fig.canvas.tostring_rgb(), dtype='uint8')
+    image = image.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+
+    yield Image.fromarray(image)
+
     
 def generate_animation(frames, coefs, img_size, desired_range, theta_points, coefficients):
     fig, ax, background, circles, circle_lines, drawing = setup_animation_env(img_size, desired_range, coefficients)
@@ -102,13 +110,15 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_
     # Saving the animation
     with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_file:
         anim.save(temp_file.name, fps=15)
-        return Image.fromarray(np.zeros((img_size, img_size), dtype=np.uint8)), temp_file.name
+
+    for frame in range(frames):
+        yield from animate(frame, coefs, time, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, theta), temp_file.name
 
 def setup_gradio_interface():
     interface = gr.Interface(
         fn=fourier_transform_drawing,
         inputs=[
-            gr.Image(label="Input Image", sources=['upload'], type="pil"),
+            gr.Image(label="Drawing Progress", sources=['upload'], type="pil"),
             gr.Slider(minimum=5, maximum=500, value=100, label="Number of Frames"),
             gr.Slider(minimum=1, maximum=500, value=50, label="Number of Coefficients"),
             gr.Number(value=224, label="Image Size (px)", precision=0),