Update app.py

app.py

@@ -1,52 +1,63 @@
 import gradio as gr
 from doctr.models import ocr_predictor
-from PIL import Image
 import numpy as np
+from PIL import Image
 
-# Load a lightweight CPU model
+# Initialize the model once
 model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)
 
-def extract_as_markdown(image):
+def extract_all_tables(image):
     if image is None:
         return "Please upload an image."
     
-    # Convert PIL to format DocTR understands
+    # Convert image for the model
     img_array = np.array(image)
     result = model([img_array])
     
-    markdown_output = "| Text | Confidence |\n| --- | --- |\n"
-    
-    # Export results and build a table structure
+    # Reconstruct lines based on vertical position to keep tables intact
+    markdown_output = ""
     json_export = result.export()
     
     for page in json_export['pages']:
         for block in page['blocks']:
             for line in block['lines']:
-                line_text = " ".join([word['value'] for word in line['words']])
-                # Formatting as a markdown row
-                markdown_output += f"| {line_text} | ✓ |\n"
+                # Join words in a line with spaces
+                words = [word['value'] for word in line['words']]
+                line_text = " | ".join(words) # Use pipe for table structure
+                markdown_output += f"| {line_text} |\n"
+            markdown_output += "\n---\n\n" # Separator for different blocks/tables
                 
     return markdown_output
 
-# Professional UI with standard Copy Button
+# UI without the 'show_copy_button' error
 with gr.Blocks() as demo:
-    gr.Markdown("## 📑 Accountancy Table Extractor (CPU Optimized)")
-    gr.Markdown("Extracts text into a Markdown table. Copy the result and paste directly into Excel or Word.")
+    gr.Markdown("## 📑 Accountancy Table Extractor (CPU)")
     
     with gr.Row():
         with gr.Column():
-            img_input = gr.Image(type="pil", label="Upload Paper")
-            btn = gr.Button("Extract Table", variant="primary")
+            img_input = gr.Image(type="pil", label="Upload Accountancy Paper")
+            btn = gr.Button("Extract All Tables", variant="primary")
         
         with gr.Column():
-            # show_copy_button=True is the most reliable way to copy on CPU
+            # Standard Textbox without the failing argument
             text_output = gr.Textbox(
-                label="Markdown Table Output", 
+                label="Extracted Table Data", 
                 lines=20, 
-                show_copy_button=True,
-                elem_id="result-box"
+                elem_id="output-text"
             )
             
-    btn.click(extract_as_markdown, inputs=img_input, outputs=text_output)
+            # Custom Copy Button
+            copy_btn = gr.Button("📋 Copy to Clipboard")
+            
+            # JavaScript to handle the copy action
+            copy_btn.click(None, None, None, js="""
+                () => {
+                    const text = document.querySelector('#output-text textarea').value;
+                    navigator.clipboard.writeText(text);
+                    alert('Copied to clipboard!');
+                }
+            """)
+
+    btn.click(extract_all_tables, inputs=img_input, outputs=text_output)
 
 demo.launch()