Update app.py

app.py

@@ -50,10 +50,14 @@ Include:
             temperature=0.7
         )
 
-        return response.choices[0].message.content
+        result = response.choices[0].message.content
+
+        # Return for display + for PDF
+        return result, result
 
     except Exception as e:
-        return f"❌ Error: {e}"
+        err = f"❌ Error: {e}"
+        return err, err
 
 
 # =========================
@@ -61,6 +65,9 @@ Include:
 # =========================
 def create_pdf(text):
 
+    if not text:
+        return None
+
     file_path = "/mnt/data/ai_roadmap.pdf"
 
     c = canvas.Canvas(file_path, pagesize=A4)
@@ -73,23 +80,21 @@ def create_pdf(text):
     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)
+            y = height - 40
 
         text_obj.textLine(line)
         y -= 14
 
     c.drawText(text_obj)
 
-    # Footer
+    # Footer in PDF
     c.setFont("Helvetica-Bold", 10)
-    c.drawCentredString(
-        width / 2,
-        20,
-        "Made by Kubra 🌻"
-    )
+    c.drawCentredString(width / 2, 20, "Made by Kubra 🌻")
 
     c.save()
 
@@ -105,61 +110,6 @@ html, body {
     padding: 0;
     font-family: Arial, sans-serif;
 }
-
-/* Light Mode */
-@media (prefers-color-scheme: light) {
-    html,
-    body,
-    .gradio-container {
-        background: linear-gradient(to right, #dff5e3, #b7e4c7) !important;
-        color: #1a1a1a !important;
-    }
-
-    input, textarea, select {
-        background: #f6fff8 !important;
-        color: #1a1a1a !important;
-        border: 1px solid #9dd9b1 !important;
-        border-radius: 6px;
-    }
-
-    button {
-        background: #6fcf97 !important;
-        color: #103822 !important;
-        border-radius: 8px !important;
-        font-weight: 600;
-    }
-
-    button:hover {
-        background: #5bbd84 !important;
-    }
-}
-
-/* Dark Mode */
-@media (prefers-color-scheme: dark) {
-    html,
-    body,
-    .gradio-container {
-        background: linear-gradient(to right, #667eea, #764ba2) !important;
-        color: white !important;
-    }
-
-    input, textarea, select {
-        background: #1e1e2f !important;
-        color: white !important;
-        border: 1px solid #444 !important;
-    }
-
-    button {
-        background: #7c83fd !important;
-        color: white !important;
-        border-radius: 8px !important;
-        font-weight: 600;
-    }
-
-    button:hover {
-        background: #6a70e0 !important;
-    }
-}
 """
 
 
@@ -186,10 +136,10 @@ with gr.Blocks(css=custom_css) as app:
 
     generate_btn = gr.Button("✨ Generate Roadmap")
 
-    output = gr.Markdown()
+    output_md = gr.Markdown()
+    output_text = gr.Textbox(visible=False)   # Hidden for PDF
 
     pdf_btn = gr.Button("📥 Download as PDF")
-
     pdf_file = gr.File(label="Download PDF")
 
 
@@ -197,13 +147,14 @@ with gr.Blocks(css=custom_css) as app:
     generate_btn.click(
         generate_roadmap,
         inputs=[domain, level, time],
-        outputs=output
+        outputs=[output_md, output_text]
     )
 
+
     # Generate PDF
     pdf_btn.click(
         create_pdf,
-        inputs=output,
+        inputs=output_text,
         outputs=pdf_file
     )
 
@@ -212,7 +163,8 @@ with gr.Blocks(css=custom_css) as app:
     gr.Markdown(
         """
         ---
-        🌻 **Made by Kubra** 🌻
+        **Made by Kubra** 🌻  
+        
         """
     )