Update app.py

app.py

@@ -1,56 +1,47 @@
-# app.py
-import fitz                     # PyMuPDF
-from markdownify import markdownify as md
-import json
+import os, shutil, subprocess, tempfile, json
 import gradio as gr
 
-def convert_pdf_to_markdown(path):
-    """Extract each page as HTML, convert to Markdown."""
-    doc = fitz.open(path)
-    pages_md = []
-    for i, page in enumerate(doc, start=1):
-        html = page.get_text("html") or ""
-        # Clean conversion: collapse multiple newlines
-        page_md = md(html).strip()
-        pages_md.append({"page": i, "markdown": page_md})
-    return pages_md
-
 def process_upload(pdf_file, output_format):
-    """
-    pdf_file: tempfile-like object from Gradio
-    output_format: "markdown" or "json"
-    """
-    # Convert and collect
-    pages = convert_pdf_to_markdown(pdf_file.name)
-    
+    # Create temp output directory
+    out_dir = tempfile.mkdtemp()
+    # Run Marker CLI: outputs files marker-0000.md or marker-0000.json
+    fmt = "markdown" if output_format=="markdown" else "json"
+    cmd = [
+        "marker_single",
+        pdf_file.name,
+        "--output_format", fmt,
+        "--output_dir", out_dir,
+        "--paginate_output"  # page separators
+    ]
+    subprocess.run(cmd, check=True)
+    # Read and combine results
+    results = []
+    for fname in sorted(os.listdir(out_dir)):
+        path = os.path.join(out_dir, fname)
+        with open(path, 'r', encoding='utf-8') as f:
+            results.append(f.read())
+    # Cleanup
+    shutil.rmtree(out_dir)
     if output_format == "markdown":
-        # Join all pages
-        full_md = "\n\n---\n\n".join(p["markdown"] for p in pages)
-        return full_md
+        return "\n\n---\n\n".join(results)
     else:
-        # Return pretty JSON
-        return json.dumps({"pages": pages}, indent=2, ensure_ascii=False)
+        # If JSON, combine into list of pages
+        return json.dumps({"pages": results}, indent=2, ensure_ascii=False)
 
-# Gradio interface
 demo = gr.Interface(
     fn=process_upload,
     inputs=[
-        gr.File(label="Upload your PDF", file_types=[".pdf"]),
-        gr.Radio(choices=["markdown", "json"],
-                 value="markdown",
-                 label="Output format")
+        gr.File(label="Upload PDF", file_types=[".pdf"]),
+        gr.Radio(["markdown","json"], value="markdown", label="Output format")
     ],
-    outputs=gr.Code(label="Converted Output"),
-    title="PDF → Markdown/JSON Converter",
+    outputs=gr.Code(label="Output"),
+    title="PDF → Markdown/JSON with LaTeX Support",
     description=(
-        "Upload a PDF and get back a professionally converted Markdown "
-        "or a structured JSON with each page’s Markdown. "
-        "PDFs with images or complex tables may still need manual review."
-    ),
-    examples=[
-        # you can add example PDFs here if desired
-    ]
+        "Uploads a PDF and uses Marker to extract text, structure, and LaTeX math. "
+        "Choose Markdown to get a single .md with `$...$`/`$$...$$` math, "
+        "or JSON for a page-by-page array."
+    )
 )
 
-if __name__ == "__main__":
+if __name__=="__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)