Update app.py

app.py

@@ -1,32 +1,44 @@
 import gradio as gr
 import cv2
-import pytesseract
+import easyocr
 from PIL import Image
 
-def ocr(image):
-    # Step 1: Convert the image to gray scale
-    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+def get_grayscale(image):
+    return cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
 
-    # Step 2: Perform OCR on the image
-    text = pytesseract.image_to_string(gray)
+def thresholding(src):
+    return cv2.threshold(src,127,255, cv2.THRESH_TOZERO)[1]
 
-    # Step 3: Prepare the output
-    output = {
-        "step1": "The first step is to convert the image to grayscale. This often improves the accuracy of OCR.",
-        "step1_image": gray,
-        "step2": "The second step is to perform OCR on the grayscale image using pytesseract.",
-        "step2_text": text
-    }
+def ocr_with_easy(img):
+    gray_scale_image = get_grayscale(img)
+    thresholding(gray_scale_image)
+    cv2.imwrite('image.png', gray_scale_image)
+    reader = easyocr.Reader(['en'])
+    bounds = reader.readtext('image.png', paragraph="False", detail=0)
+    bounds = ''.join(bounds)
+    return bounds
 
-    return output
+def generate_ocr(img):
+    text_output = ''
+    if (img).any():
+        if img is not None:
+            text_output = ocr_with_easy(img)
+        else:
+            raise gr.Error("Please upload an image!!!!")
+    return text_output
 
-def process_output(output):
-    output["step1_image"] = Image.fromarray(output["step1_image"])
-    return output
+image = gr.Image()
+output = gr.Textbox(label="Output")
 
-iface = gr.Interface(fn=ocr,
-                     inputs=gr.Image(type="pil"),
-                     outputs=["text", "image", "text", "text"])
+demo = gr.Interface(
+    generate_ocr,
+    image,
+    output,
+    title="Optical Character Recognition",
+    css=".gradio-container {background-color: lightgray} #radio_div {background-color: #FFD8B4; font-size: 40px;}",
+    article = """<p style='text-align: center;'>Feel free to give us your thoughts on this demo and please contact us at
+                    <a href="mailto:[email protected]" target="_blank">[email protected]</a>
+                    <p style='text-align: center;'>Developed by: <a href="https://www.pragnakalp.com" target="_blank">Pragnakalp Techlabs</a></p>"""
+)
 
-iface.process_output = process_output
-iface.launch()
+demo.launch()