Update app.py

app.py

@@ -149,10 +149,6 @@ Your goal is to create a Recruitment Management Application HTML layout that fol
 
 @app.post("/generate_pdf")
 async def generate_pdf(request: Request):
-    """
-    Generate screenshots (screens + workflows) from generated HTML mockup
-    and compile them into a downloadable PDF.
-    """
     data = await request.json()
     html = data.get("html", "")
     if not html.strip():
@@ -165,14 +161,19 @@ async def generate_pdf(request: Request):
         f.write(html)
 
     screenshots = []
-
     try:
+        from playwright.sync_api import sync_playwright
         with sync_playwright() as p:
-            browser = p.chromium.launch(headless=True)
+            # 👇 Add this logging
+            print("✅ Launching Chromium headless browser...")
+            browser = p.chromium.launch(
+                headless=True,
+                args=["--no-sandbox", "--disable-setuid-sandbox"]
+            )
             page = browser.new_page()
             page.goto(f"file://{tmp_html_path}", wait_until="networkidle")
+            print("✅ Page loaded successfully")
 
-            # Detect clickable screens from nav/sidebar
             selectors = ["nav a", ".sidebar a", ".menu-item a", "a[href^='#']"]
             screen_links = []
             for sel in selectors:
@@ -197,17 +198,27 @@ async def generate_pdf(request: Request):
 
             browser.close()
     except Exception as e:
+        import traceback
+        print("❌ Browser capture error:", traceback.format_exc())
         return JSONResponse({"error": f"Browser capture error: {str(e)}"}, status_code=500)
 
     # Build PDF
-    c = canvas.Canvas(pdf_path, pagesize=A4)
-    width, height = A4
-    for image_path in screenshots:
-        img = ImageReader(image_path)
-        iw, ih = img.getSize()
-        scale = min(width / iw, height / ih)
-        c.drawImage(img, 0, 0, iw * scale, ih * scale)
-        c.showPage()
-    c.save()
+    try:
+        from reportlab.lib.pagesizes import A4
+        from reportlab.pdfgen import canvas
+        from reportlab.lib.utils import ImageReader
+
+        c = canvas.Canvas(pdf_path, pagesize=A4)
+        width, height = A4
+        for image_path in screenshots:
+            img = ImageReader(image_path)
+            iw, ih = img.getSize()
+            scale = min(width / iw, height / ih)
+            c.drawImage(img, 0, 0, iw * scale, ih * scale)
+            c.showPage()
+        c.save()
+    except Exception as e:
+        print("❌ PDF generation error:", e)
+        return JSONResponse({"error": f"PDF generation failed: {str(e)}"}, status_code=500)
 
     return JSONResponse({"pdf_url": f"/static/outputs/{os.path.basename(pdf_path)}"})