Update app.py

app.py

@@ -65,7 +65,7 @@ def animate(frame, coefs, time, fig, ax, background, circles, circle_lines, draw
     center = (0, 0)
     for idx, (r, fr) in enumerate(coefs_static):
         c_dynamic = coefs[idx][0] * np.exp(1j * (fr * tau * time[frame]))
-        x, y = center[0] + r * np.cos(theta), center[1] + r * np.sin(theta)
+        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)
         center = (center[0] + np.real(c_dynamic), center[1] + np.imag(c_dynamic))
@@ -81,14 +81,13 @@ def animate(frame, coefs, time, fig, ax, background, circles, circle_lines, draw
     ax.draw_artist(drawing)
 
     fig.canvas.blit(ax.bbox)
+    
+    # Convert canvas to PIL Image using buffer_rgba
+    #fig.canvas.draw()
+    buffer = fig.canvas.buffer_rgba()
+    pil_image = Image.frombuffer("RGBA", fig.canvas.get_width_height(), buffer, "raw", "RGBA", 0, 1)
 
-    # 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)
-
+    return (pil_image, None)
     
 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)
@@ -112,7 +111,7 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size, blur_
         anim.save(temp_file.name, fps=15)
 
     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
+        yield (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(