Update app.py

app.py

@@ -1,26 +1,59 @@
 import gradio as gr
 import easyocr
 import numpy as np
+from difflib import SequenceMatcher
 
-# Initialize EasyOCR reader once with English language
+# Initialize EasyOCR reader once with English
 reader = easyocr.Reader(['en'], gpu=False)
 
-def ocr_easy(img):
+def calculate_similarity(text1, text2):
+    # Calculate similarity ratio for accuracy and pipeline score
+    return SequenceMatcher(None, text1.lower().strip(), text2.lower().strip()).ratio()
+
+def ocr_easy_with_accuracy(img, correct_text):
     if img is None:
-        return "No image uploaded"
+        return "No image uploaded", "", ""
+
+    # Convert uploaded image (PIL) to numpy array
     img_array = np.array(img)
+
     try:
-        # OCR with detail=0 returns text only, paragraph=True groups lines
+        # Run EasyOCR with paragraph grouping and no detailed bounding boxes
         results = reader.readtext(img_array, detail=0, paragraph=True)
-        return "\n".join(results)
+        detected_text = "\n".join(results)
+
+        if correct_text.strip():
+            accuracy = calculate_similarity(detected_text, correct_text)
+            pipeline_score = accuracy  # Using same metric for demo
+
+            accuracy_str = f"{accuracy:.2%}"
+            pipeline_score_str = f"{pipeline_score:.2%}"
+        else:
+            accuracy_str = "No ground truth provided"
+            pipeline_score_str = "No ground truth provided"
+
+        return detected_text, accuracy_str, pipeline_score_str
+
     except Exception as e:
-        return f"EasyOCR Error: {str(e)}"
+        error_msg = f"EasyOCR Error: {str(e)}"
+        return error_msg, "", ""
 
 with gr.Blocks() as demo:
-    gr.Markdown("# EasyOCR OCR Demo")
-    img_input = gr.Image(type="pil", label="Upload Image")
+    gr.Markdown("# EasyOCR with Accuracy and Pipeline Score")
+
+    with gr.Row():
+        img_input = gr.Image(type="pil", label="Upload Image")
+        correct_text_input = gr.Textbox(label="Enter Correct Text for Accuracy Check", lines=4)
+
     output_text = gr.Textbox(label="OCR Result", lines=10)
-    run_button = gr.Button("Run OCR")
-    run_button.click(ocr_easy, inputs=img_input, outputs=output_text)
+    accuracy_output = gr.Textbox(label="Accuracy", interactive=False)
+    pipeline_output = gr.Textbox(label="Pipeline Integration Score", interactive=False)
+
+    run_button = gr.Button("Run OCR and Check Accuracy")
+    run_button.click(
+        ocr_easy_with_accuracy,
+        inputs=[img_input, correct_text_input],
+        outputs=[output_text, accuracy_output, pipeline_output]
+    )
 
 demo.launch()