Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
 import gradio as gr
 from groq import Groq
+from reportlab.lib.pagesizes import A4
+from reportlab.pdfgen import canvas
 
 
 # =========================
@@ -24,11 +26,9 @@ def generate_roadmap(domain, level, time):
 
     prompt = f"""
 Create a detailed learning roadmap.
-
 Domain: {domain}
 Skill Level: {level}
 Time Available: {time}
-
 Include:
 - Weekly plan
 - Resources
@@ -56,34 +56,66 @@ Include:
         return f"❌ Error: {e}"
 
 
+# =========================
+# Generate PDF
+# =========================
+def create_pdf(text):
+
+    file_path = "/mnt/data/ai_roadmap.pdf"
+
+    c = canvas.Canvas(file_path, pagesize=A4)
+    width, height = A4
+
+    x = 40
+    y = height - 40
+
+    text_obj = c.beginText(x, y)
+    text_obj.setFont("Helvetica", 11)
+
+    for line in text.split("\n"):
+        if y < 40:
+            c.showPage()
+            text_obj = c.beginText(x, height - 40)
+            text_obj.setFont("Helvetica", 11)
+
+        text_obj.textLine(line)
+        y -= 14
+
+    c.drawText(text_obj)
+
+    # Footer
+    c.setFont("Helvetica-Bold", 10)
+    c.drawCentredString(
+        width / 2,
+        20,
+        "Made by Kubra 🌻"
+    )
+
+    c.save()
+
+    return file_path
+
+
 # =========================
 # Custom CSS
 # =========================
 custom_css = """
-
 html, body {
     margin: 0;
     padding: 0;
     font-family: Arial, sans-serif;
 }
 
-
 /* Light Mode */
 @media (prefers-color-scheme: light) {
-
     html,
     body,
-    .gradio-container,
-    .app,
-    .wrap,
-    .container {
+    .gradio-container {
         background: linear-gradient(to right, #dff5e3, #b7e4c7) !important;
         color: #1a1a1a !important;
     }
 
-    input,
-    textarea,
-    select {
+    input, textarea, select {
         background: #f6fff8 !important;
         color: #1a1a1a !important;
         border: 1px solid #9dd9b1 !important;
@@ -102,25 +134,18 @@ html, body {
     }
 }
 
-
 /* Dark Mode */
 @media (prefers-color-scheme: dark) {
-
     html,
     body,
-    .gradio-container,
-    .app,
-    .wrap,
-    .container {
+    .gradio-container {
         background: linear-gradient(to right, #667eea, #764ba2) !important;
-        color: #ffffff !important;
+        color: white !important;
     }
 
-    input,
-    textarea,
-    select {
+    input, textarea, select {
         background: #1e1e2f !important;
-        color: #ffffff !important;
+        color: white !important;
         border: 1px solid #444 !important;
     }
 
@@ -135,7 +160,6 @@ html, body {
         background: #6a70e0 !important;
     }
 }
-
 """
 
 
@@ -164,12 +188,34 @@ with gr.Blocks(css=custom_css) as app:
 
     output = gr.Markdown()
 
+    pdf_btn = gr.Button("📥 Download as PDF")
+
+    pdf_file = gr.File(label="Download PDF")
+
+
+    # Generate roadmap
     generate_btn.click(
         generate_roadmap,
         inputs=[domain, level, time],
         outputs=output
     )
 
+    # Generate PDF
+    pdf_btn.click(
+        create_pdf,
+        inputs=output,
+        outputs=pdf_file
+    )
+
+
+    # Footer
+    gr.Markdown(
+        """
+        ---
+        🌻 **Made by Kubra** 🌻
+        """
+    )
+
 
 # =========================
 # Launch