Update app.py

app.py

@@ -3,20 +3,25 @@ import json
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib import colors
-from io import BytesIO
+import tempfile
+import os
 
 PAGE_WIDTH, PAGE_HEIGHT = letter
 
 def generate_pdf(file):
     try:
-        # Open the uploaded file safely
+        # 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}"
 
-    pdf_output = BytesIO()
-    c = canvas.Canvas(pdf_output, pagesize=letter)
+    # Create a temporary file for the PDF
+    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    pdf_path = temp_pdf.name
+    temp_pdf.close()  # ReportLab will write to this path
+
+    c = canvas.Canvas(pdf_path, pagesize=letter)
     page_created = False
 
     for ad_size, ad_data in data.items():
@@ -33,7 +38,6 @@ def generate_pdf(file):
                 box_height = y2 - y1
                 font_size = max(6, min(18, box_height * 0.8))
 
-                # Draw bounding box (optional for debug)
                 c.setStrokeColor(colors.lightgrey)
                 c.rect(x1, PAGE_HEIGHT - y2, x2 - x1, box_height, fill=0)
 
@@ -49,23 +53,11 @@ 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")
         c.showPage()
 
     c.save()
-    pdf_output.seek(0)
-    return pdf_output, "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."
-)
 
-iface.launch()
+    return pdf_path, "PDF generated successfully!"