Update app.py

app.py

@@ -94,24 +94,24 @@ def generate_animation(frames, coefs, 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)
-    theta = np.linspace(0, tau, num=frames)
+    thetas = np.linspace(0, tau, num=frames)
     draw_x, draw_y = [], []
 
-    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))
+    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 anim, frame_times
+    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)
-    anim, frame_times = generate_animation(frames, coefs, img_size, desired_range, 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
     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, frame_times, 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, thetas), temp_file.name)
 
 def setup_gradio_interface():
     interface = gr.Interface(