Update app.py

app.py

@@ -3,93 +3,130 @@ from groq import Groq
 import os
 from fpdf import FPDF
 
-# Initialize Groq client
+# -------------------------------
+# INIT GROQ
+# -------------------------------
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# ---------- ROADMAP GENERATOR ----------
+# -------------------------------
+# ROADMAP GENERATOR
+# -------------------------------
 def generate(domain, level, hours, months):
     prompt = f"""
 Create a structured learning roadmap.
+
 Domain: {domain}
 Level: {level}
 Hours per day: {hours}
 Months: {months}
 
-Make a weekly roadmap with projects.
+Make a weekly roadmap with projects and milestones.
 """
+
     try:
         res = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
-            messages=[{"role": "user", "content": prompt}]
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.7
         )
         return res.choices[0].message.content
+
     except Exception as e:
         return f"Error: {str(e)}"
 
-# ---------- PDF DOWNLOAD ----------
+
+# -------------------------------
+# PDF DOWNLOAD
+# -------------------------------
 def download_pdf(text):
     if not text or text.startswith("Error"):
-        text = "No roadmap generated or error occurred."
+        text = "No roadmap generated."
 
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
 
-    clean_text = text.encode('latin-1', 'ignore').decode('latin-1')
+    clean_text = text.encode("latin-1", "ignore").decode("latin-1")
+
     for line in clean_text.split("\n"):
-        pdf.multi_cell(0, 10, line)
+        pdf.multi_cell(0, 8, line)
 
     path = "roadmap.pdf"
     pdf.output(path)
     return path
 
-# ---------- AI CHATBOT ----------
+
+# -------------------------------
+# CHATBOT (FIXED)
+# -------------------------------
 def chat(message, history):
     if history is None:
         history = []
 
-    # Build message list for Groq API
-    messages = history.copy()
+    # 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})
+
     messages.append({"role": "user", "content": message})
 
     try:
         res = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
-            messages=messages
+            messages=messages,
+            temperature=0.7
         )
         reply = res.choices[0].message.content
+
     except Exception as e:
         reply = f"Error: {str(e)}"
 
-    # Update history
-    history.append({"role": "user", "content": message})
-    history.append({"role": "assistant", "content": reply})
+    history.append((message, reply))
+    return "", history
+
 
-    return "", history  # Clear input box, return updated history
+# -------------------------------
+# UI
+# -------------------------------
+with gr.Blocks(theme=gr.themes.Soft()) as app:
 
-# ---------- GRADIO UI ----------
-with gr.Blocks() as app:
-    gr.Markdown("# 🚀 AI Roadmap Startup")
+    gr.Markdown(
+        """
+        # 🚀 AI Roadmap Startup  
+        Generate learning roadmaps + chat with AI mentor  
+        """
+    )
 
-    with gr.Tab("Roadmap Generator"):
+    # ------------------- TAB 1
+    with gr.Tab("📚 Roadmap Generator"):
         domain = gr.Textbox(label="Domain", placeholder="e.g. Data Science")
-        level = gr.Dropdown(["Beginner","Intermediate","Advanced"], label="Level", value="Beginner")
-        hours = gr.Slider(1, 10, value=2, label="Hours/day")
+        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")
 
-        btn = gr.Button("Generate")
-        output = gr.Textbox(lines=15, label="Generated Roadmap")
+        btn = gr.Button("Generate Roadmap")
+        output = gr.Textbox(lines=18, label="Your Roadmap")
+
         pdf_btn = gr.Button("Download PDF")
-        pdf_file = gr.File(label="PDF File")
+        pdf_file = gr.File()
 
         btn.click(generate, [domain, level, hours, months], output)
         pdf_btn.click(download_pdf, output, pdf_file)
 
-    with gr.Tab("AI Chatbot"):
-        chatbot = gr.Chatbot()
-        msg = gr.Textbox(label="Your Message", placeholder="Type and press Enter...")
+    # ------------------- TAB 2
+    with gr.Tab("🤖 AI Mentor Chat"):
+        chatbot = gr.Chatbot(height=450)
+        msg = gr.Textbox remembered=False, placeholder="Ask anything..."
 
         msg.submit(chat, [msg, chatbot], [msg, chatbot])
 
+
+# -------------------------------
 if __name__ == "__main__":
     app.launch()