Update app.py

app.py

@@ -492,5 +492,30 @@ return (
     fixed_code,
     report
 )
+from reportlab.platypus import SimpleDocTemplate, Paragraph
+from reportlab.lib.styles import getSampleStyleSheet
+
+def create_pdf(report_text):
+    pdf_path = "security_report.pdf"
+
+    doc = SimpleDocTemplate(pdf_path)
+
+    styles = getSampleStyleSheet()
+
+    elements = [
+        Paragraph(report_text.replace("\n", "<br/>"), styles["BodyText"])
+    ]
+
+    doc.build(elements)
+
+    return pdf_path
+download_btn = gr.Button("Generate PDF Report")
+pdf_file = gr.File(label="Download Report")
+download_btn.click(
+    create_pdf,
+    inputs=report_output,
+    outputs=pdf_file
+)
+
 if __name__ == "__main__":
     demo.launch()