2

app.py

@@ -5,11 +5,20 @@ import streamlit as st
 from PIL import Image
 import pytesseract
 
+def enhance(img):
+    gaussian_thresh = cv2.adaptiveThreshold(
+    img, 
+    maxValue=255, 
+    adaptiveMethod=cv2.ADAPTIVE_THRESH_GAUSSIAN_C, 
+    thresholdType=cv2.THRESH_BINARY, 
+    blockSize=11, 
+    C=2)
+    return gaussian_thresh
 
 act =  [(150,240), (610,260)]
 
 
-def align_images(ref_gray, input_gray):
+def align_images(ref_gray, input_gray, enh= True):
     """
     Aligns the input image to the reference image using homography.
     
@@ -23,7 +32,9 @@ def align_images(ref_gray, input_gray):
     # # Convert images to grayscale
     # ref_gray = cv2.cvtColor(reference_image, cv2.COLOR_BGR2GRAY)
     # input_gray = cv2.cvtColor(input_image, cv2.COLOR_BGR2GRAY)
-
+    if enh:
+        ref_gray = enhance(ref_gray)
+        input_gray = enhance(input_gray)
     
     # Detect ORB keypoints and descriptors
     orb = cv2.ORB_create(nfeatures=500)