Update app.py

app.py

@@ -123,6 +123,21 @@ def display_image(img):
 
 
 # --------------------------------------------
+
+def get_dynamic_kernel(img_height):
+    # Set the kernel size to be 1% of the image height
+    kernel_height = int(0.01 * img_height)
+    
+    # Ensure the kernel height is odd
+    if kernel_height % 2 == 0:
+        kernel_height += 1
+    
+    # Set minimum and maximum limits
+    kernel_height = max(3, kernel_height)  # Minimum limit
+    kernel_height = min(11, kernel_height)  # Maximum limit
+    
+    return np.ones((kernel_height, 1), np.uint8)
+    
 def process_image(selected_image):
     # Convert PIL image to OpenCV format
     opencv_image = np.array(selected_image)
@@ -143,7 +158,9 @@ def process_image(selected_image):
     # Apply OTSU's thresholding
     _, thresh = cv2.threshold(gray_image, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
 
-    kernel = np.ones((5, 1), np.uint8)
+    
+    kernel = get_dynamic_kernel(thresh.shape[0])
+    # kernel = np.ones((5, 1), np.uint8)
     thresh = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel, iterations=2)
 
     # # Apply median blur