Update app.py

app.py

@@ -3,23 +3,25 @@ from groq import Groq
 import os
 from fpdf import FPDF
 
-# =======================
+# =========================
 # INIT
-# =======================
+# =========================
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 SYSTEM_PROMPT = "You are a helpful AI mentor."
 
-# =======================
-# ROADMAP
-# =======================
+# =========================
+# ROADMAP GENERATOR
+# =========================
 def generate(domain, level, hours, months):
     prompt = f"""
 Create a structured learning roadmap.
 
 Domain: {domain}
 Level: {level}
-Hours/day: {hours}
+Hours per day: {hours}
 Months: {months}
+
+Make weekly roadmap with projects.
 """
 
     try:
@@ -36,16 +38,16 @@ Months: {months}
         return f"Error: {str(e)}"
 
 
-# =======================
-# PDF
-# =======================
+# =========================
+# PDF DOWNLOAD
+# =========================
 def download_pdf(text):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
 
     if not text:
-        text = "No roadmap."
+        text = "No roadmap generated."
 
     clean = text.encode("latin-1", "ignore").decode("latin-1")
 
@@ -57,19 +59,25 @@ def download_pdf(text):
     return path
 
 
-# =======================
-# CHAT (GRADIO 6 SAFE)
-# =======================
+# =========================
+# CHATBOT (METADATA SAFE)
+# =========================
 def chat(message, history):
     history = history or []
 
+    # ---- Build messages for Groq (NO metadata)
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
 
-    for h in history:
-        messages.append(h)
+    for item in history:
+        if isinstance(item, dict):
+            role = item.get("role")
+            content = item.get("content")
+            if role and content:
+                messages.append({"role": role, "content": content})
 
     messages.append({"role": "user", "content": message})
 
+    # ---- Add to UI history
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": ""})
 
@@ -93,18 +101,18 @@ def chat(message, history):
         yield "", history
 
 
-# =======================
+# =========================
 # UI
-# =======================
+# =========================
 with gr.Blocks(title="AI Roadmap Startup") as app:
 
-    gr.Markdown("# 🚀 AI Roadmap Generator + Chat")
+    gr.Markdown("# 🚀 AI Roadmap Generator + AI Mentor")
 
     with gr.Tabs():
 
         # -------- ROADMAP TAB
-        with gr.Tab("Roadmap"):
-            domain = gr.Textbox(label="Domain")
+        with gr.Tab("Roadmap Generator"):
+            domain = gr.Textbox(label="Domain", placeholder="Data Science")
             level = gr.Dropdown(
                 ["Beginner", "Intermediate", "Advanced"],
                 value="Beginner",
@@ -113,22 +121,22 @@ with gr.Blocks(title="AI Roadmap Startup") as app:
             hours = gr.Slider(1, 10, value=2, label="Hours/day")
             months = gr.Slider(1, 12, value=3, label="Months")
 
-            btn = gr.Button("Generate")
-            output = gr.Textbox(lines=18)
+            gen_btn = gr.Button("Generate Roadmap")
+            roadmap_output = gr.Textbox(lines=18)
 
             pdf_btn = gr.Button("Download PDF")
             pdf_file = gr.File()
 
-            btn.click(generate, [domain, level, hours, months], output)
-            pdf_btn.click(download_pdf, output, pdf_file)
+            gen_btn.click(generate, [domain, level, hours, months], roadmap_output)
+            pdf_btn.click(download_pdf, roadmap_output, pdf_file)
 
         # -------- CHAT TAB
-        with gr.Tab("Chat"):
+        with gr.Tab("AI Chatbot"):
             chatbot = gr.Chatbot(height=500)
             msg = gr.Textbox(placeholder="Ask anything...")
 
             msg.submit(chat, [msg, chatbot], [msg, chatbot])
 
-# =======================
+# =========================
 if __name__ == "__main__":
     app.launch(theme=gr.themes.Soft())