Update app.py

app.py

@@ -81,24 +81,44 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size):
     fig.set_size_inches(15, 15)
 
     draw_x, draw_y = [], []
+
+    # Pre-compute static values outside the animate function
+    theta = np.linspace(0, tau, 80)
+    coefs_static = [(np.linalg.norm(c), fr) for c, fr in coefs]  # Assuming `r` remains constant
     
     def animate(i, coefs, time):
         t = time[i]
         center = (0, 0)
-        theta = np.linspace(0, tau, 80)
-        for _, (c, fr) in enumerate(coefs):
-            c = c * np.exp(1j*(fr * tau * t))
-            r = np.linalg.norm(c)
+    
+        # Loop over coefficients
+        for _, (r, fr) in enumerate(coefs_static):
+            c_dynamic = coefs[_][0] * np.exp(1j * (fr * tau * t))  # Dynamic part of 'c'
             x, y = center[0] + r * np.cos(theta), center[1] + r * np.sin(theta)
-            circle_lines[_].set_data([center[0], center[0] + np.real(c)], [center[1], center[1] + np.imag(c)])
+            circle_lines[_].set_data([center[0], center[0] + np.real(c_dynamic)], [center[1], center[1] + np.imag(c_dynamic)])
             circles[_].set_data(x, y) 
-            center = (center[0] + np.real(c), center[1] + np.imag(c))
+            center = (center[0] + np.real(c_dynamic), center[1] + np.imag(c_dynamic))
         
         draw_x.append(center[0])
         draw_y.append(center[1])
-
         drawing.set_data(draw_x[:i+1], draw_y[:i+1])
 
+    # def animate(i, coefs, time):
+    #     t = time[i]
+    #     center = (0, 0)
+    #     theta = np.linspace(0, tau, 80)
+    #     for _, (c, fr) in enumerate(coefs):
+    #         c = c * np.exp(1j*(fr * tau * t))
+    #         r = np.linalg.norm(c)
+    #         x, y = center[0] + r * np.cos(theta), center[1] + r * np.sin(theta)
+    #         circle_lines[_].set_data([center[0], center[0] + np.real(c)], [center[1], center[1] + np.imag(c)])
+    #         circles[_].set_data(x, y) 
+    #         center = (center[0] + np.real(c), center[1] + np.imag(c))
+        
+    #     draw_x.append(center[0])
+    #     draw_y.append(center[1])
+
+    #     drawing.set_data(draw_x[:i+1], draw_y[:i+1])
+
     drawing_time = 1
     time = np.linspace(0, drawing_time, num=frames)    
     anim = animation.FuncAnimation(fig, animate, frames=frames, interval=5, fargs=(coefs, time)) 
@@ -122,7 +142,7 @@ interface = gr.Interface(
     outputs=gr.Video(),
     title="Fourier Transform Drawing",
     description="Upload an image and generate a Fourier Transform drawing animation. You can find out more about the project here : https://github.com/staghado/fourier-draw",
-    examples=[["Fourier2.jpg", 100, 100, 224], ["Luffy.png", 150, 200, 224]]
+    examples=[["Fourier2.jpg", 100, 200, 224], ["Luffy.png", 100, 100, 224]]
 )
 
 if __name__ == "__main__":