Update tabs/image_ocr_llm.py

tabs/image_ocr_llm.py

@@ -1,14 +1,17 @@
 import gradio as gr
-from components.llm_ocr import extract_text_llm
+from components.llm_ocr_blip import extract_text_blip
 
 def image_ocr_llm_tab():
-    gr.Markdown("## 📸 Extract Handwritten Text from Image using LLaMA-Vision", elem_classes="centered-text")
+    with gr.Tab("🧾 Image OCR (LLM-based)"):
+        gr.Markdown("## 🧾 Upload Image and Extract Text", elem_classes="centered-text")
+        with gr.Row():
+            with gr.Column():
+                img_input = gr.Image(type="pil", label="Upload Image (handwritten or typed)")
+                submit_btn = gr.Button("Extract Text", elem_id="process-btn")
+                clear_btn = gr.Button("Clear")
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            input_image = gr.Image(type="filepath", label="Upload Handwritten Image")
-            submit_btn = gr.Button("Extract Text")
-        with gr.Column(scale=2):
-            extracted_output = gr.Textbox(label="Extracted Text", lines=20)
+            with gr.Column():
+                output_box = gr.Textbox(label="📄 Extracted Text", lines=12)
 
-    submit_btn.click(fn=extract_text_llm, inputs=input_image, outputs=extracted_output)
+        submit_btn.click(extract_text_blip, inputs=img_input, outputs=output_box)
+        clear_btn.click(lambda: "", inputs=[], outputs=output_box)