Update app.py

app.py

@@ -49,19 +49,33 @@ def fourier_transform_drawing(input_image, frames, coefficients, img_size):
 
     t_list = np.linspace(0, tau, len(xs))
 
-    # compute the Fourier coefficients
     def f(t, t_list, xs, ys):
-        return np.interp(t, t_list, xs + 1j*ys)
-
-    def compute_cn(f, n):
-        coef = 1/tau*quad_vec(
-            lambda t: f(t, t_list, xs, ys)*np.exp(-n*t*1j), 
-            0, 
-            tau, 
-            limit=100, 
-            full_output=False)[0]
+        return np.interp(t, t_list, xs + 1j * ys)
+
+    def compute_cn(f, n, t_list, xs, ys):
+        num_points = 1000  # Adjust this based on required precision
+        t_values = np.linspace(0, tau, num_points)
+    
+        # Vectorized operation
+        integrand = f(t_values, t_list, xs, ys) * np.exp(-n * t_values * 1j)
+        
+        coef = np.trapz(integrand, t_values) / tau
         return coef
 
+    
+    # # compute the Fourier coefficients
+    # def f(t, t_list, xs, ys):
+    #     return np.interp(t, t_list, xs + 1j*ys)
+
+    # def compute_cn(f, n):
+    #     coef = 1/tau*quad_vec(
+    #         lambda t: f(t, t_list, xs, ys)*np.exp(-n*t*1j), 
+    #         0, 
+    #         tau, 
+    #         limit=100, 
+    #         full_output=False)[0]
+    #     return coef
+
     N = coefficients
     coefs = [(compute_cn(f, 0), 0)] + [(compute_cn(f, j), j) for i in range(1, N+1) for j in (i, -i)]
 
@@ -121,7 +135,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], ["Luffy.png", 250, 300]]
+    examples=[["Fourier2.jpg", 100, 100, 224], ["Luffy.png", 150, 200, 224]]
 )
 
 if __name__ == "__main__":