Update app.py

app.py

@@ -59,12 +59,12 @@ def setup_animation_env(img_size, desired_range, coefficients):
 
     return fig, ax, background, circles, circle_lines, drawing
 
-def animate(frame, coefs, time, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, theta):
+def animate(frame, coefs, frame_times, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, theta):
     fig.canvas.restore_region(background)
 
     center = (0, 0)
     for idx, (r, fr) in enumerate(coefs_static):
-        c_dynamic = coefs[idx][0] * np.exp(1j * (fr * tau * time[frame]))
+        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)
@@ -89,29 +89,29 @@ def animate(frame, coefs, time, fig, ax, background, circles, circle_lines, draw
 
     return (pil_image, None)
     
-def generate_animation(frames, coefs, img_size, desired_range, theta_points, coefficients):
+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]
-    time = np.linspace(0, 1, num=frames)
-    theta = np.linspace(0, tau, theta_points)
+    frame_times = np.linspace(0, 1, num=frames)
+    theta = np.linspace(0, tau, num=frames)
     draw_x, draw_y = [], []
 
-    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 = 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, theta))
 
-    return anim
+    return anim, frame_times
     
-def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_kernel_size, desired_range, num_points, theta_points):
+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)
-    anim = generate_animation(frames, coefs, img_size, desired_range, theta_points, coefficients)
+    anim, frame_times = generate_animation(frames, coefs, img_size, desired_range, coefficients)
     
     # Saving the animation
     with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_file:
         anim.save(temp_file.name, fps=15)
 
     for frame in range(frames):
-        yield (animate(frame, coefs, time, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, theta), temp_file.name)
+        yield (animate(frame, coefs, frame_times, fig, ax, background, circles, circle_lines, drawing, draw_x, draw_y, coefs_static, theta), temp_file.name)
 
 def setup_gradio_interface():
     interface = gr.Interface(
@@ -124,7 +124,6 @@ def setup_gradio_interface():
             gr.Slider(minimum=3, maximum=11, step=2, value=5, label="Blur Kernel Size (odd number)"),
             gr.Number(value=400, label="Desired Range for Scaling", precision=0),
             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", gr.Video()],
         title="Fourier Transform Drawing",