changes

app.py

@@ -788,25 +788,8 @@ def run_ocr(
     # Prepare plain text by stripping HTML tags and replacing <br>
     plain_text = re.sub(r"<[^>]*>", "", predicted_html.replace("<br>", "\n"))
     # Write temporary files
-    txt_path = None
-    csv_path = None
-    try:
-        txt_fd = io.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8")
-        txt_fd.write(plain_text)
-        txt_fd.flush()
-        txt_path = txt_fd.name
-        txt_fd.close()
-    except Exception:
-        txt_path = None
-    try:
-        csv_fd = io.NamedTemporaryFile(delete=False, suffix=".csv", mode="w", encoding="utf-8")
-        df_all.to_csv(csv_fd, index=False)
-        csv_fd.flush()
-        csv_path = csv_fd.name
-        csv_fd.close()
-    except Exception:
-        csv_path = None
-    return overlay_img, predicted_html, df_all, txt_path, csv_path
+
+    return overlay_img, predicted_html, df_all
 
 
 # ----------------------------------------------------------------------
@@ -840,13 +823,13 @@ def create_gradio_interface():
         overlay_output = gr.Image(label="Detected Regions")
         predictions_output = gr.HTML(label="Predictions (HTML)")
         df_output = gr.DataFrame(label="Token Scores", interactive=False)
-        txt_file_output = gr.File(label="Download OCR Prediction (.txt)")
-        csv_file_output = gr.File(label="Download Token Scores (.csv)")
+        # txt_file_output = gr.File(label="Download OCR Prediction (.txt)")
+        # csv_file_output = gr.File(label="Download Token Scores (.csv)")
         # Editable text
         edited_text = gr.Textbox(
             label="Edit full predicted text", lines=8, interactive=True
         )
-        download_edited_btn = gr.Button("Download edited text")
+        # download_edited_btn = gr.Button("Download edited text")
 
         # Callback for OCR
         def on_run(image, xml, gray, binarize, metric):
@@ -855,7 +838,7 @@ def create_gradio_interface():
         run_btn.click(
             fn=on_run,
             inputs=[image_input, xml_input, apply_gray_checkbox, apply_bin_checkbox, metric_radio],
-            outputs=[overlay_output, predictions_output, df_output, txt_file_output, csv_file_output],
+            outputs=[overlay_output, predictions_output, df_output],
         )
 
         # Populate editable text with plain text from predictions
@@ -869,25 +852,7 @@ def create_gradio_interface():
             outputs=edited_text,
         )
 
-        # Download edited text by writing to a temporary file
-        def download_edited(txt):
-            if not txt:
-                return None
-            try:
-                fd = io.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8")
-                fd.write(txt)
-                fd.flush()
-                path = fd.name
-                fd.close()
-                return path
-            except Exception:
-                return None
 
-        download_edited_btn.click(
-            fn=download_edited,
-            inputs=edited_text,
-            outputs=txt_file_output,
-        )
     return demo