Update app.py

app.py

@@ -245,101 +245,122 @@ def update_slider(file_input):
 
 
 # Create Gradio interface
-with gr.Blocks(title="📖 Image/PDF OCR with LightOnOCR", theme=gr.themes.Soft()) as demo:
-    gr.Markdown(f"""
-# 📖 Image/PDF to Text Extraction with LightOnOCR
-
-**💡 How to use:**
-1. Upload an image or PDF
-2. For PDFs: select which page to extract (1-20)
-3. Adjust temperature if needed
-4. Click "Extract Text"
-
-**Note:** The Markdown rendering for tables may not always be perfect. Check the raw output for complex tables!
-
-**Model:** LightOnOCR-1B-1025 by LightOn AI  
-**Device:** {device.upper()}  
-**Attention:** {attn_implementation}
-""")
+# with gr.Blocks(title="📖 Image/PDF OCR with LightOnOCR", theme=gr.themes.Soft()) as demo:
+#     gr.Markdown(f"""
+# # 📖 Image/PDF to Text Extraction with LightOnOCR
+
+# **💡 How to use:**
+# 1. Upload an image or PDF
+# 2. For PDFs: select which page to extract (1-20)
+# 3. Adjust temperature if needed
+# 4. Click "Extract Text"
+
+# **Note:** The Markdown rendering for tables may not always be perfect. Check the raw output for complex tables!
+
+# **Model:** LightOnOCR-1B-1025 by LightOn AI  
+# **Device:** {device.upper()}  
+# **Attention:** {attn_implementation}
+# """)
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            file_input = gr.File(
-                label="🖼️ Upload Image or PDF",
-                file_types=[".pdf", ".png", ".jpg", ".jpeg"],
-                type="filepath"
-            )
-            rendered_image = gr.Image(
-                label="📄 Preview",
-                type="pil",
-                height=400,
-                interactive=False
-            )
-            num_pages = gr.Slider(
-                minimum=1,
-                maximum=20,
-                value=1,
-                step=1,
-                label="PDF: Page Number",
-                info="Select which page to extract"
-            )
-            page_info = gr.Textbox(
-                label="Processing Info",
-                value="",
-                interactive=False
-            )
-            temperature = gr.Slider(
-                minimum=0.0,
-                maximum=1.0,
-                value=0.2,
-                step=0.05,
-                label="Temperature",
-                info="0.0 = deterministic, Higher = more varied"
-            )
-            submit_btn = gr.Button("Extract Text", variant="primary")
-            clear_btn = gr.Button("Clear", variant="secondary")
+#     with gr.Row():
+#         with gr.Column(scale=1):
+#             file_input = gr.File(
+#                 label="🖼️ Upload Image or PDF",
+#                 file_types=[".pdf", ".png", ".jpg", ".jpeg"],
+#                 type="filepath"
+#             )
+#             rendered_image = gr.Image(
+#                 label="📄 Preview",
+#                 type="pil",
+#                 height=400,
+#                 interactive=False
+#             )
+#             num_pages = gr.Slider(
+#                 minimum=1,
+#                 maximum=20,
+#                 value=1,
+#                 step=1,
+#                 label="PDF: Page Number",
+#                 info="Select which page to extract"
+#             )
+#             page_info = gr.Textbox(
+#                 label="Processing Info",
+#                 value="",
+#                 interactive=False
+#             )
+#             temperature = gr.Slider(
+#                 minimum=0.0,
+#                 maximum=1.0,
+#                 value=0.2,
+#                 step=0.05,
+#                 label="Temperature",
+#                 info="0.0 = deterministic, Higher = more varied"
+#             )
+#             submit_btn = gr.Button("Extract Text", variant="primary")
+#             clear_btn = gr.Button("Clear", variant="secondary")
         
-        with gr.Column(scale=2):
-            output_text = gr.Markdown(
-                label="📄 Extracted Text (Rendered)",
-                value="*Extracted text will appear here...*"
-            )
-            medications_output = gr.Textbox(
-    label="💊 Extracted Medicines/Drugs",
-    placeholder="Medicine/drug names will appear here...",
-    lines=2,
-    max_lines=5,
-    interactive=False,
-    show_copy_button=True
-)
+#         with gr.Column(scale=2):
+#             output_text = gr.Markdown(
+#                 label="📄 Extracted Text (Rendered)",
+#                 value="*Extracted text will appear here...*"
+#             )
+#             medications_output = gr.Textbox(
+#     label="💊 Extracted Medicines/Drugs",
+#     placeholder="Medicine/drug names will appear here...",
+#     lines=2,
+#     max_lines=5,
+#     interactive=False,
+#     show_copy_button=True
+# )
     
-    with gr.Row():
-        with gr.Column():
-            raw_output = gr.Textbox(
-                label="Raw Markdown Output",
-                placeholder="Raw text will appear here...",
-                lines=20,
-                max_lines=30,
-                show_copy_button=True
-            )
+#     with gr.Row():
+#         with gr.Column():
+#             raw_output = gr.Textbox(
+#                 label="Raw Markdown Output",
+#                 placeholder="Raw text will appear here...",
+#                 lines=20,
+#                 max_lines=30,
+#                 show_copy_button=True
+#             )
     
-    # Event handlers
+#     # Event handlers
+#     submit_btn.click(
+#     fn=process_input,
+#     inputs=[file_input, temperature, num_pages, ],
+#     outputs=[output_text, medications_output, raw_output, page_info, rendered_image, num_pages]
+# )
+
+with gr.Blocks(title="💊 Medicine Extraction", theme=gr.themes.Soft()) as demo:
+    file_input = gr.File(
+        label="🖼️ Upload Image or PDF",
+        file_types=[".pdf", ".png", ".jpg", ".jpeg"],
+        type="filepath"
+    )
+    temperature = gr.Slider(
+        minimum=0.0,
+        maximum=1.0,
+        value=0.2,
+        step=0.05,
+        label="Temperature",
+        info="0.0 = deterministic, Higher = more varied"
+    )
+    medicines_output = gr.Textbox(
+        label="💊 Extracted Medicines/Drugs",
+        placeholder="Medicine/drug names will appear here...",
+        lines=2,
+        max_lines=5,
+        interactive=False,
+        show_copy_button=True
+    )
+    submit_btn = gr.Button("Extract Medicines", variant="primary")
+
     submit_btn.click(
-    fn=process_input,
-    inputs=[file_input, temperature, num_pages, ],
-    outputs=[output_text, medications_output, raw_output, page_info, rendered_image, num_pages]
-)
-    
-    file_input.change(
-        fn=update_slider,
-        inputs=[file_input],
-        outputs=[num_pages]
+        fn=process_input,  # already yields medicines as second output
+        inputs=[file_input, temperature, 1],  # fix page=1 or expose slider
+        outputs=[gr.update(), medicines_output, gr.update(), gr.update(), gr.update(), gr.update()]
     )
+
     
-    clear_btn.click(
-        fn=lambda: (None, "*Extracted text will appear here...*", "", "", None, 1),
-        outputs=[file_input, output_text, raw_output, page_info, rendered_image, num_pages]
-    )
 
 
 if __name__ == "__main__":