Update app.py

app.py

@@ -4,52 +4,28 @@ from docling.document_converter import DocumentConverter, PdfFormatOption
 from docling.datamodel.base_models import InputFormat
 import tempfile
 
-# -------------------------
-# PDF → Markdown / JSON
-# -------------------------
 def pdf_to_markdown(file):
-    if file is None:
-        return "No file uploaded."
-
     # Save uploaded file temporarily
     tmp_path = file.name
-
-    # Initialize Docling converter for PDF
+    # Convert PDF using Docling/VLM (Granite Docling)
     converter = DocumentConverter(
         format_options={
             InputFormat.PDF: PdfFormatOption()
         }
     )
-
-    # Convert PDF
     result = converter.convert(tmp_path)
     doc = result.document
-
-    # Export to Markdown
+    # Export to Markdown (or you can export to JSON via doc.model_dump())
     md = doc.export_to_markdown()
-
     return md
 
-# -------------------------
-# Gradio Interface
-# -------------------------
-output_box = gr.Textbox(
-    label="Extracted Markdown",
-    lines=30,        # Visible lines (enlarge as needed)
-    max_lines=2000,  # Max scrollable content
-    scrollable=True
-)
-
 interface = gr.Interface(
     fn=pdf_to_markdown,
     inputs=gr.File(file_types=[".pdf"]),
-    outputs=output_box,
+    outputs="text",
     title="PDF → Markdown/JSON with Granite Docling",
     description="Upload a PDF and get parsed Markdown (or JSON) using Granite Docling via Docling."
 )
 
-# -------------------------
-# Launch App
-# -------------------------
 if __name__ == "__main__":
     interface.launch()