Update app.py

app.py

@@ -3,14 +3,16 @@ from groq import Groq
 import os
 from fpdf import FPDF
 
-# -------------------------------
+# =========================
 # INIT GROQ
-# -------------------------------
+# =========================
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# -------------------------------
+SYSTEM_PROMPT = "You are a helpful AI mentor who helps students learn skills and careers."
+
+# =========================
 # ROADMAP GENERATOR
-# -------------------------------
+# =========================
 def generate(domain, level, hours, months):
     prompt = f"""
 Create a structured learning roadmap.
@@ -20,14 +22,17 @@ Level: {level}
 Hours per day: {hours}
 Months: {months}
 
-Make a weekly roadmap with projects and milestones.
+Make weekly roadmap with projects and milestones.
 """
 
     try:
         res = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
-            messages=[{"role": "user", "content": prompt}],
-            temperature=0.7
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": prompt},
+            ],
+            temperature=0.7,
         )
         return res.choices[0].message.content
 
@@ -35,9 +40,9 @@ Make a weekly roadmap with projects and milestones.
         return f"Error: {str(e)}"
 
 
-# -------------------------------
+# =========================
 # PDF DOWNLOAD
-# -------------------------------
+# =========================
 def download_pdf(text):
     if not text or text.startswith("Error"):
         text = "No roadmap generated."
@@ -46,9 +51,9 @@ def download_pdf(text):
     pdf.add_page()
     pdf.set_font("Arial", size=12)
 
-    clean_text = text.encode("latin-1", "ignore").decode("latin-1")
+    clean = text.encode("latin-1", "ignore").decode("latin-1")
 
-    for line in clean_text.split("\n"):
+    for line in clean.split("\n"):
         pdf.multi_cell(0, 8, line)
 
     path = "roadmap.pdf"
@@ -56,77 +61,91 @@ def download_pdf(text):
     return path
 
 
-# -------------------------------
-# CHATBOT (FIXED)
-# -------------------------------
+# =========================
+# CHATBOT (GRADIO 6 SAFE)
+# =========================
 def chat(message, history):
     if history is None:
         history = []
 
-    # Convert Gradio history → Groq format
-    messages = []
-
-    for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": bot_msg})
+    # Convert history into Groq format
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for item in history:
+        messages.append(item)
 
     messages.append({"role": "user", "content": message})
 
+    # Add user msg to UI history
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": ""})
+
     try:
-        res = client.chat.completions.create(
+        stream = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=messages,
-            temperature=0.7
+            stream=True,
+            temperature=0.7,
         )
-        reply = res.choices[0].message.content
 
-    except Exception as e:
-        reply = f"Error: {str(e)}"
+        reply = ""
+        for chunk in stream:
+            if chunk.choices[0].delta.content:
+                token = chunk.choices[0].delta.content
+                reply += token
+                history[-1]["content"] = reply
+                yield "", history
 
-    history.append((message, reply))
-    return "", history
+    except Exception as e:
+        history[-1]["content"] = f"Error: {str(e)}"
+        yield "", history
 
 
-# -------------------------------
+# =========================
 # UI
-# -------------------------------
-with gr.Blocks(theme=gr.themes.Soft()) as app:
-
-    gr.Markdown(
-        """
-        # 🚀 AI Roadmap Startup  
-        Generate learning roadmaps + chat with AI mentor  
-        """
-    )
-
-    # ------------------- TAB 1
-    with gr.Tab("📚 Roadmap Generator"):
-        domain = gr.Textbox(label="Domain", placeholder="e.g. Data Science")
-        level = gr.Dropdown(
-            ["Beginner", "Intermediate", "Advanced"],
-            value="Beginner",
-            label="Level"
-        )
-        hours = gr.Slider(1, 10, value=2, label="Hours per day")
-        months = gr.Slider(1, 12, value=3, label="Months")
+# =========================
+with gr.Blocks(title="AI Roadmap Startup", theme=gr.themes.Soft()) as app:
+
+    gr.Markdown("# 🚀 AI Roadmap Generator + Mentor Chat")
+
+    with gr.Tabs():
+
+        # =====================
+        # ROADMAP TAB
+        # =====================
+        with gr.Tab("Roadmap Generator"):
+
+            domain = gr.Textbox(label="Domain", placeholder="Data Science")
+            level = gr.Dropdown(
+                ["Beginner", "Intermediate", "Advanced"],
+                value="Beginner",
+                label="Level",
+            )
+            hours = gr.Slider(1, 10, value=2, label="Hours/day")
+            months = gr.Slider(1, 12, value=3, label="Months")
 
-        btn = gr.Button("Generate Roadmap")
-        output = gr.Textbox(lines=18, label="Your Roadmap")
+            gen_btn = gr.Button("Generate Roadmap")
+            roadmap_output = gr.Textbox(lines=18)
 
-        pdf_btn = gr.Button("Download PDF")
-        pdf_file = gr.File()
+            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,
+            )
 
-    # ------------------- TAB 2
-    with gr.Tab("🤖 AI Mentor Chat"):
-        chatbot = gr.Chatbot(height=450)
-        msg = gr.Textbox remembered=False, placeholder="Ask anything..."
+            pdf_btn.click(download_pdf, roadmap_output, pdf_file)
 
-        msg.submit(chat, [msg, chatbot], [msg, chatbot])
+        # =====================
+        # CHAT TAB
+        # =====================
+        with gr.Tab("AI Mentor Chat"):
+            chatbot = gr.Chatbot(type="messages", height=500)
+            msg = gr.Textbox(placeholder="Ask anything...")
 
+            msg.submit(chat, [msg, chatbot], [msg, chatbot])
 
-# -------------------------------
+# =========================
 if __name__ == "__main__":
     app.launch()