Update app.py

app.py

@@ -1,57 +1,26 @@
 import gradio as gr
-from PIL import Image
 import easyocr
 import numpy as np
-import pytesseract
 
-# Initialize EasyOCR reader
-easyocr_reader = easyocr.Reader(['en'])
+# Initialize EasyOCR reader once
+reader = easyocr.Reader(['en'])
 
-# Explicitly set Tesseract binary path (Hugging Face Spaces default)
-pytesseract.pytesseract.tesseract_cmd = '/usr/bin/tesseract'
-
-def get_metrics(model):
-    # Simulated metrics for demo
-    if model == "EasyOCR":
-        return {"Accuracy": "95%", "Precision": "94%", "Pipeline": "Easy Integration (90%)"}
-    elif model == "Tesseract":
-        return {"Accuracy": "85%", "Precision": "83%", "Pipeline": "Manual Integration (75%)"}
-    else:
-        return {}
-
-def ocr_all(img, ground_truth):
+def ocr_easy(img):
     if img is None:
-        return "No image provided", "No image provided"
+        return "No image uploaded"
+    # Convert PIL Image to numpy array
     img_array = np.array(img)
-    
-    # EasyOCR
     try:
-        easyocr_result = easyocr_reader.readtext(img_array, detail=0, paragraph=True)
-        text_easy = "\n".join(easyocr_result)
+        results = reader.readtext(img_array, detail=0, paragraph=True)
+        return "\n".join(results)
     except Exception as e:
-        text_easy = f"EasyOCR Error: {str(e)}"
-    metrics_easy = get_metrics("EasyOCR")
-    
-    # Tesseract OCR with graceful error handling
-    try:
-        text_tess = pytesseract.image_to_string(img)
-    except Exception as e:
-        text_tess = f"Tesseract Error: {str(e)}"
-    metrics_tess = get_metrics("Tesseract")
-    
-    output_easy = f"**Output:**\n{text_easy}\n\n" + "\n".join([f"{k}: {v}" for k,v in metrics_easy.items()])
-    output_tess = f"**Output:**\n{text_tess}\n\n" + "\n".join([f"{k}: {v}" for k,v in metrics_tess.items()])
-    
-    return output_easy, output_tess
+        return f"EasyOCR Error: {str(e)}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# OCR Demo with EasyOCR and Tesseract")
-    img_in = gr.Image(type="pil", label="Upload Screenshot")
-    txt_in = gr.Textbox(lines=4, label="Paste Ground Truth Text Here")
-    btn = gr.Button("Run OCR")
-    out_easy = gr.Markdown(label="EasyOCR Output")
-    out_tess = gr.Markdown(label="Tesseract Output")
-
-    btn.click(ocr_all, inputs=[img_in, txt_in], outputs=[out_easy, out_tess])
+    gr.Markdown("# EasyOCR Demo")
+    img_input = gr.Image(type="pil", label="Upload Image")
+    output_text = gr.Textbox(label="OCR Result", lines=10)
+    button = gr.Button("Run EasyOCR")
+    button.click(ocr_easy, inputs=[img_input], outputs=[output_text])
 
 demo.launch()