Update app.py

app.py

@@ -10,25 +10,25 @@ from math import tau
 from PIL import Image
 
 
-def fourier_transform_drawing(input_image, frames, coefficients):
+def fourier_transform_drawing(input_image, frames, coefficients, img_size):
     # Convert PIL to OpenCV image(array)
     input_image = np.array(input_image)
     img = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)
 
     # processing
-    # Resize the image to a smaller size for faster processing
-    width = 224
-    height = 224
-    dim = (width, height)
-    resized = cv2.resize(img, dim, interpolation=cv2.INTER_AREA)
+    # resize the image to a smaller size for faster processing
+    dim = (img_size, img_size)
+    img = cv2.resize(img, dim, interpolation=cv2.INTER_AREA)
     
-    imgray = cv2.cvtColor(resized, cv2.COLOR_BGR2GRAY)
+    imgray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
     blurred = cv2.GaussianBlur(imgray, (5, 5), 0)
     (_, thresh) = cv2.threshold(blurred, 0, 255, cv2.THRESH_BINARY_INV | cv2.THRESH_OTSU)
     contours, _ = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
     
-    # take only the largest contour
-    largest_contour_idx = np.argmax([len(c) for c in contours])
+    # find the contour with the largest area
+    largest_contour_idx = np.argmax([cv2.contourArea(c) for c in contours])
+    largest_contour = contours[largest_contour_idx]
+    #largest_contour_idx = np.argmax([len(c) for c in contours])
     verts = [tuple(coord) for coord in contours[largest_contour_idx].squeeze()]
 
     xs, ys = zip(*verts)
@@ -115,7 +115,8 @@ interface = gr.Interface(
     inputs=[
         gr.Image(label="Input Image", sources=['upload'], type="pil"),
         gr.Slider(minimum=5, maximum=500, value=100, label="Number of Frames"),
-        gr.Slider(minimum=1, maximum=500, value=50, label="Number of Coefficients")
+        gr.Slider(minimum=1, maximum=500, value=50, label="Number of Coefficients"),
+        gr.Number(value=224, label="Image size", precision=0)
     ],
     outputs=gr.Video(),
     title="Fourier Transform Drawing",