Update app.py

app.py

@@ -10,6 +10,8 @@ IMPROVEMENTS OVER v1:
 Run AFTER notebooks 1-4 to use the improved models.
 Upload this to: https://huggingface.co/spaces/ayshajavd/code-security-analyzer
 """
+from reportlab.platypus import SimpleDocTemplate, Paragraph
+from reportlab.lib.styles import getSampleStyleSheet
 import json
 import re
 import time
@@ -20,8 +22,7 @@ from transformers import (
     AutoModelForSequenceClassification,
     T5ForConditionalGeneration,
 )
-from reportlab.platypus import SimpleDocTemplate, Paragraph
-from reportlab.lib.styles import getSampleStyleSheet
+
 from huggingface_hub import hf_hub_download
 import numpy as np
 
@@ -371,6 +372,28 @@ def create_pdf(report_text):
 
     doc.build(elements)
 
+    return pdf_path
+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(
+            str(report_text).replace("\n", "<br/>"),
+            styles["BodyText"]
+        )
+    ]
+
+    doc.build(elements)
+
     return pdf_path
 EXAMPLES = [
     ["""import sqlite3\n\ndef get_user(username):\n    conn = sqlite3.connect('users.db')\n    query = f"SELECT * FROM users WHERE username = '{username}'"\n    return conn.execute(query).fetchone()\n"""],
@@ -395,8 +418,9 @@ with gr.Blocks(
 
     **v2 Improvements:** Per-class threshold optimization | Temperature-calibrated probabilities | Asymmetric Loss training | GraphCodeBERT-base (125M params) | CodeT5+ 220M CWE-aware fixer
     """)
+ with gr.Row():
 
-    with gr.Row():
+    # LEFT COLUMN
     with gr.Column(scale=1):
 
         code_input = gr.Code(
@@ -419,6 +443,7 @@ with gr.Blocks(
                 size="lg"
             )
 
+    # RIGHT COLUMN
     with gr.Column(scale=1):
 
         report_output = gr.Markdown(
@@ -437,6 +462,9 @@ with gr.Blocks(
             label="JSON Report",
             visible=False
         )
+    
+
+  
 
     gr.Examples(examples=EXAMPLES, inputs=[code_input], label="Example Code Snippets")
 
@@ -447,7 +475,9 @@ with gr.Blocks(
         fn=analyze_code, inputs=[code_input], outputs=[report_output], api_name="analyze"
                      )
     json_btn.click(
-        fn=show_json, inputs=[code_input], outputs=[json_output]
+        fn=show_json, 
+        inputs=[code_input], 
+        outputs=[json_output]
     )
     download_btn.click(
     fn=create_pdf,