Update app.py

app.py

@@ -3,50 +3,49 @@ from docling.document_converter import DocumentConverter
 from docling.datamodel.pipeline_options import PdfPipelineOptions, TesseractCliOcrOptions
 import spaces
 
+# GPU decorator not really required for Docling OCR, but kept if you want
 @spaces.GPU
 def convert_document(file, output_format):
     pdf_opts = PdfPipelineOptions(
         do_ocr=True,
-        ocr_options=TesseractCliOcrOptions(lang=["eng"]),  # or ["eng", "ara"]
+        ocr_options=TesseractCliOcrOptions(lang=["eng"]),  
     )
-    # Load document and convert it using Docling
+
     converter = DocumentConverter(pipeline_options=pdf_opts)
     result = converter.convert(file.name)
 
-    # Check available attributes in DoclingDocument
-    available_attributes = dir(result.document)
-
-    # Choose the output format
+    # Choose output format safely
     if output_format == "Markdown":
         converted_text = result.document.export_to_markdown()
     elif output_format == "JSON":
-        converted_text = result.document.export_to_json()
+        # JSON needs to be dumped into a string for the Textbox
+        import json
+        converted_text = json.dumps(result.document.export_to_json(), indent=2)
     else:
-        converted_text = "Unsupported format"
+        converted_text = "⚠️ Unsupported format"
 
-    # Placeholder metadata extraction based on available attributes
-    metadata = {
-        "Available Attributes": available_attributes
-    }
+    # Metadata always JSON-friendly
+    metadata = {"Available Attributes": dir(result.document)}
 
     return converted_text, metadata
 
-# Define the Gradio interface using the new component syntax
+
 with gr.Blocks() as app:
-    gr.Markdown("# Document Converter with Docling")
-    gr.Markdown("Upload a document, choose the output format, and get the converted text with metadata.")
+    gr.Markdown("# 📄 Document Converter with Docling OCR")
+    gr.Markdown("Upload a PDF, choose the output format, and get the converted text + metadata.")
+
+    with gr.Row():
+        file_input = gr.File(label="Upload PDF", file_types=[".pdf"])
+        format_input = gr.Radio(["Markdown", "JSON"], label="Choose Output Format")
 
-    file_input = gr.File(label="Upload Document")
-    format_input = gr.Radio(["Markdown", "JSON"], label="Choose Output Format")
-    output_text = gr.Textbox(label="Converted Document")
+    output_text = gr.Textbox(label="Converted Document", lines=20)
     output_metadata = gr.JSON(label="Metadata")
 
-    # Define the process button and bind it to the function
     convert_button = gr.Button("Convert")
     convert_button.click(
-        convert_document,
+        fn=convert_document,
         inputs=[file_input, format_input],
         outputs=[output_text, output_metadata]
     )
 
-app.launch(debug=True)
+app.launch(debug=True)