Update button_click_alt.py

button_click_alt.py

@@ -5,9 +5,30 @@ import tensorflow as tf
 from tensorflow.keras.preprocessing.image import img_to_array, load_img
 from process import preprocess_image
 
+def infer_text(rotated):
+    im.save("converted.png")
+    url = "https://ajax.thehive.ai/api/demo/classify?endpoint=text_recognition"
+    files = {
+        "image": ("converted.png", open("converted.png", "rb"), "image/png"),
+        "model_type": (None, "detection"),
+        "media_type": (None, "photo"),
+    }
+    headers = {"referer": "https://thehive.ai/"}
+
+    res = requests.post(url, headers=headers, files=files)
+
+    text = ""
+    for output in res.json()["response"]["output"]:
+        text += output["block_text"]
+
+    text = decode(encode(text, "latin-1", "backslashreplace"), "unicode-escape")
+
+    return text
+
+
 def find_order_id(uploaded_file, input_file, model):
     rotated = preprocess_image(uploaded_file)
-    text = pytesseract.image_to_string(rotated)
+    text = infer_text(rotated)
     with input_file as file:
         file_contents = file.read().decode()
         lines = file_contents.split('\n')
@@ -56,5 +77,3 @@ def find_order_id(uploaded_file, input_file, model):
                 }
 
     return result
-
-