Spaces:

developer28
/

Youtubedownloader

Sleeping

developer28 commited on Jun 21, 2025

Commit

a6e9713

verified ·

1 Parent(s): c0f8987

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,34 +3,42 @@ from xhtml2pdf import pisa
 from io import BytesIO
 import os
 import uuid
 def generate_dummy_pdf():
     html = """
     <html>
     <body>
-    <h1>Hello, PDF!</h1>
-    <p>This is a test PDF file.</p>
     </body>
     </html>
     """
-    result = BytesIO()
-    pisa_status = pisa.CreatePDF(html, dest=result)
-    if pisa_status.err:
-        print("PDF generation failed")
-        return None
-    pdf_path = os.path.join(os.path.expanduser("~"), "Downloads", f"test_pdf_{uuid.uuid4().hex}.pdf")
-    with open(pdf_path, "wb") as f:
-        f.write(result.getvalue())
-    print("✅ PDF saved at:", pdf_path)
-    return pdf_path
 with gr.Blocks() as demo:
-    btn = gr.Button("📄 Generate PDF")
-    output = gr.File(label="📥 Download PDF")
-    btn.click(fn=generate_dummy_pdf, inputs=[], outputs=output)
 demo.launch()

 from io import BytesIO
 import os
 import uuid
+import tempfile
 def generate_dummy_pdf():
+    # ✅ Step 1: Create simple HTML
     html = """
     <html>
     <body>
+    <h1>Hello PDF</h1>
+    <p>This is a test PDF generated by xhtml2pdf.</p>
     </body>
     </html>
     """
+    # ✅ Step 2: Generate PDF into memory
+    pdf_data = BytesIO()
+    result = pisa.CreatePDF(html, dest=pdf_data)
+    # ✅ Step 3: Error handling
+    if result.err:
+        return "❌ PDF generation failed"
+    # ✅ Step 4: Write PDF to a real file
+    pdf_data.seek(0)
+    output_path = os.path.join(tempfile.gettempdir(), f"test_report_{uuid.uuid4().hex}.pdf")
+    with open(output_path, "wb") as f:
+        f.write(pdf_data.read())
+    print("✅ PDF generated at:", output_path)
+    return output_path  # ✅ Gradio File component works with actual file path
+# ✅ Gradio UI
 with gr.Blocks() as demo:
+    gr.Markdown("## 🧪 PDF Generation Test")
+    btn = gr.Button("📄 Generate PDF Report")
+    file_output = gr.File(label="📥 Download PDF")
+    btn.click(fn=generate_dummy_pdf, inputs=[], outputs=file_output)
 demo.launch()