Update app.py

app.py

@@ -4,22 +4,23 @@ from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib import colors
 import tempfile
-import os
 
 PAGE_WIDTH, PAGE_HEIGHT = letter
 
+# -------------------
+# PDF generation function
+# -------------------
 def generate_pdf(file):
     try:
-        # Read uploaded JSON file
         with open(file.name, "r", encoding="utf-8") as f:
             data = json.load(f)
     except Exception as e:
         return None, f"Error reading JSON: {e}"
 
-    # Create a temporary file for the PDF
+    # Create temporary PDF file
     temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
     pdf_path = temp_pdf.name
-    temp_pdf.close()  # ReportLab will write to this path
+    temp_pdf.close()
 
     c = canvas.Canvas(pdf_path, pagesize=letter)
     page_created = False
@@ -38,6 +39,7 @@ def generate_pdf(file):
                 box_height = y2 - y1
                 font_size = max(6, min(18, box_height * 0.8))
 
+                # Draw bounding box for debug
                 c.setStrokeColor(colors.lightgrey)
                 c.rect(x1, PAGE_HEIGHT - y2, x2 - x1, box_height, fill=0)
 
@@ -53,6 +55,7 @@ def generate_pdf(file):
         if page_created:
             c.showPage()
 
+    # Guarantee at least one page
     if not page_created:
         c.setFont("Helvetica", 12)
         c.drawString(50, 750, "No OCR text found")
@@ -61,3 +64,17 @@ def generate_pdf(file):
     c.save()
 
     return pdf_path, "PDF generated successfully!"
+
+# -------------------
+# Gradio Interface
+# -------------------
+iface = gr.Interface(
+    fn=generate_pdf,
+    inputs=gr.File(file_types=[".json"], label="Upload JSON"),
+    outputs=[gr.File(label="Download PDF"), gr.Textbox(label="Status")],
+    title="JSON to PDF Converter",
+    description="Upload a `results.json` file and download the generated PDF."
+)
+
+# Launch the app
+iface.launch()