Update app.py

app.py

@@ -16,8 +16,8 @@ def run_marker(pdf_path, out_dir, fmt):
         pdf_path,
         "--output_format", fmt,
         "--output_dir", out_dir,
-        "--extract_images", "True",    # pass explicit boolean
-        "--paginate_output", "True"
+        "--extract_images",    # standalone flag
+        "--paginate_output"    # standalone flag
     ]
     subprocess.run(cmd, check=True)
 
@@ -38,7 +38,7 @@ def process_upload(pdf_file, output_format):
     fmt = {
         "markdown": "markdown",
         "json":     "json",
-        "docx":     "markdown"   # produce .md before converting to DOCX
+        "docx":     "markdown"   # still produce .md before converting to DOCX
     }[output_format]
 
     # 3) Run Marker CLI
@@ -70,11 +70,14 @@ def process_upload(pdf_file, output_format):
         return docx_path
 
     # 6) Non-DOCX: clean up and return Markdown or JSON string
-    shutil.rmtree(out_dir)
+    result = None
     if output_format == "markdown":
-        return "\n\n---\n\n".join(pages)
+        result = "\n\n---\n\n".join(pages)
     else:
-        return json.dumps({"pages": pages}, indent=2, ensure_ascii=False)
+        result = json.dumps({"pages": pages}, indent=2, ensure_ascii=False)
+
+    shutil.rmtree(out_dir)
+    return result
 
 # Gradio Interface
 demo = gr.Interface(
@@ -97,4 +100,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    demo.launch(server_name="0.0.0.0", server_port=7860)