Added gray-scale and negative preprocessing for input images (#3)

- Added gray-scale and negative preprocessing for input images (526878502041b8b920a56469195bd790e9c36405)


Co-authored-by: Anil Ozturk <[email protected]>

app.py

@@ -8,19 +8,32 @@ import openai
 import ast
 import os
 from deta import Deta
+import cv2
 
 
 openai.api_key = os.getenv('API_KEY')
 reader = Reader(["tr"])
 
+
 def get_parsed_address(input_img):
 
     address_full_text = get_text(input_img)
     return openai_response(address_full_text)
-    
 
-def get_text(input_img):
-    result = reader.readtext(input_img, detail=0)
+
+def preprocess_img(inp_image):
+    gray = cv2.cvtColor(inp_image, cv2.COLOR_BGR2GRAY)
+    gray_img = cv2.bitwise_not(gray)
+    return gray_img
+
+
+def get_text(image):
+    processed_image = preprocess_img(inp_image=image)
+    result = reader.readtext(processed_image,
+                             detail=0,
+                             paragraph=True
+                             )
+
     return " ".join(result)
 
 

requirements.txt

@@ -2,4 +2,6 @@ openai
 Pillow
 easyocr
 gradio
-deta
+deta
+opencv-python
+opencv-contrib-python