Update app.py

app.py

@@ -67,7 +67,7 @@ def animate(frame, coefs, frame_times, fig, ax, background, circles, circle_line
         c_dynamic = coefs[idx][0] * np.exp(1j * (fr * tau * frame_times[frame]))
         x, y = center[0] + r * np.cos(theta[frame]), center[1] + r * np.sin(theta[frame])
         circle_lines[idx].set_data([center[0], center[0] + np.real(c_dynamic)], [center[1], center[1] + np.imag(c_dynamic)])
-        circles[idx].set_data(x, y)
+        circles[idx].set_data([x], [y])
         center = (center[0] + np.real(c_dynamic), center[1] + np.imag(c_dynamic))
 
     draw_x.append(center[0])
@@ -91,7 +91,6 @@ def animate(frame, coefs, frame_times, fig, ax, background, circles, circle_line
     
 def generate_animation(frames, coefs, img_size, desired_range, coefficients):
     fig, ax, background, circles, circle_lines, drawing = setup_animation_env(img_size, desired_range, coefficients)
-    print(coefs)
     coefs_static = [(np.linalg.norm(c), fr) for c, fr in coefs]
     frame_times = np.linspace(0, 1, num=frames)
     thetas = np.linspace(0, tau, num=frames)
@@ -100,19 +99,27 @@ def generate_animation(frames, coefs, img_size, desired_range, coefficients):
     anim = animation.FuncAnimation(fig, animate, frames=frames, interval=5, fargs=(coefs, frame_times, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, thetas))
 
     return fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, anim, frame_times, thetas, coefs_static
-    
+
 def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_kernel_size, desired_range, num_points):
     xs, ys = process_image(input_image, img_size, blur_kernel_size, desired_range)
     coefs = calculate_fourier_coefficients(xs, ys, num_points, coefficients)
-    fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, anim, frame_times, thetas, coefs_static = generate_animation(frames, coefs, img_size, desired_range, coefficients)
     
-    # Saving the animation
+    # Setup animation environment
+    fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, anim, frame_times, thetas, coefs_static = generate_animation(frames, coefs, img_size, desired_range, coefficients)
+
+    # Create a temporary file for the video
     with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_file:
-        anim.save(temp_file.name, fps=15)
+        video_path = temp_file.name
 
+    # Generate and save each frame as a PIL image, and ultimately the video
     for frame in range(frames):
-        yield (animate(frame, coefs, frame_times, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, thetas), temp_file.name)
+        pil_image, _ = animate(frame, coefs, time, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, theta)
+        yield pil_image, video_path
 
+    # Save the animation as a video
+    anim = animation.FuncAnimation(fig, animate, frames=frames, interval=5, fargs=(coefs, time, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, theta))
+    anim.save(video_path, fps=15)
+    
 def setup_gradio_interface():
     interface = gr.Interface(
         fn=fourier_transform_drawing,