Update app.py

app.py

@@ -9,7 +9,7 @@ easyocr_reader = easyocr.Reader(['en'])
 # Tesseract uses pytesseract directly
 
 def get_metrics(model):
-    # Updated simulated fixed scores without PaddleOCR
+    # Simulated fixed metrics
     if model == "EasyOCR":
         return {"Accuracy": "95%", "Precision": "94%", "Pipeline": "Easy Integration (90%)"}
     elif model == "Tesseract":
@@ -18,15 +18,23 @@ def get_metrics(model):
         return {}
 
 def ocr_all(img, ground_truth):
+    if img is None:
+        return "No image provided", "No image provided"
     img_array = np.array(img)
     
-    # EasyOCR
-    result_easy = easyocr_reader.readtext(img_array, detail=0, paragraph=True)
-    text_easy = "\n".join(result_easy)
+    # EasyOCR processing
+    try:
+        result_easy = easyocr_reader.readtext(img_array, detail=0, paragraph=True)
+        text_easy = "\n".join(result_easy)
+    except Exception as e:
+        text_easy = f"EasyOCR Error: {str(e)}"
     metrics_easy = get_metrics("EasyOCR")
     
-    # Tesseract
-    text_tess = pytesseract.image_to_string(img)
+    # Tesseract processing
+    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()])
@@ -35,10 +43,10 @@ def ocr_all(img, ground_truth):
     return output_easy, output_tess
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Multi-OCR Model Demo without PaddleOCR\nUpload an image and paste the correct ground-truth text below.")
+    gr.Markdown("# OCR Model Comparison Demo\nUpload a screenshot image and optionally paste ground-truth text.")
     with gr.Row():
-        img_in = gr.Image(type="pil", label="Screenshot")
-    txt_in = gr.Textbox(lines=4, label="Ground Truth Text Here")
+        img_in = gr.Image(type="pil", label="Upload Screenshot Image")
+    txt_in = gr.Textbox(lines=4, label="Paste Ground Truth Text Here")
     btn = gr.Button("Run OCR")
     with gr.Row():
         out_easy = gr.Markdown(label="EasyOCR Output")