Update app.py

app.py

@@ -3,26 +3,23 @@ from groq import Groq
 import os
 from fpdf import FPDF
 
-# =========================
-# INIT GROQ
-# =========================
+# =======================
+# INIT
+# =======================
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+SYSTEM_PROMPT = "You are a helpful AI mentor."
 
-SYSTEM_PROMPT = "You are a helpful AI mentor who helps students learn skills and careers."
-
-# =========================
-# ROADMAP GENERATOR
-# =========================
+# =======================
+# ROADMAP
+# =======================
 def generate(domain, level, hours, months):
     prompt = f"""
 Create a structured learning roadmap.
 
 Domain: {domain}
 Level: {level}
-Hours per day: {hours}
+Hours/day: {hours}
 Months: {months}
-
-Make weekly roadmap with projects and milestones.
 """
 
     try:
@@ -32,7 +29,6 @@ Make weekly roadmap with projects and milestones.
                 {"role": "system", "content": SYSTEM_PROMPT},
                 {"role": "user", "content": prompt},
             ],
-            temperature=0.7,
         )
         return res.choices[0].message.content
 
@@ -40,17 +36,17 @@ Make weekly roadmap with projects and milestones.
         return f"Error: {str(e)}"
 
 
-# =========================
-# PDF DOWNLOAD
-# =========================
+# =======================
+# PDF
+# =======================
 def download_pdf(text):
-    if not text or text.startswith("Error"):
-        text = "No roadmap generated."
-
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
 
+    if not text:
+        text = "No roadmap."
+
     clean = text.encode("latin-1", "ignore").decode("latin-1")
 
     for line in clean.split("\n"):
@@ -61,21 +57,19 @@ def download_pdf(text):
     return path
 
 
-# =========================
-# CHATBOT (GRADIO 6 SAFE)
-# =========================
+# =======================
+# CHAT (GRADIO 6 SAFE)
+# =======================
 def chat(message, history):
-    if history is None:
-        history = []
+    history = history or []
 
-    # Convert history into Groq format
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
-    for item in history:
-        messages.append(item)
+
+    for h in history:
+        messages.append(h)
 
     messages.append({"role": "user", "content": message})
 
-    # Add user msg to UI history
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": ""})
 
@@ -84,7 +78,6 @@ def chat(message, history):
             model="llama-3.3-70b-versatile",
             messages=messages,
             stream=True,
-            temperature=0.7,
         )
 
         reply = ""
@@ -100,21 +93,18 @@ def chat(message, history):
         yield "", history
 
 
-# =========================
+# =======================
 # UI
-# =========================
-with gr.Blocks(title="AI Roadmap Startup", theme=gr.themes.Soft()) as app:
+# =======================
+with gr.Blocks(title="AI Roadmap Startup") as app:
 
-    gr.Markdown("# 🚀 AI Roadmap Generator + Mentor Chat")
+    gr.Markdown("# 🚀 AI Roadmap Generator + Chat")
 
     with gr.Tabs():
 
-        # =====================
-        # ROADMAP TAB
-        # =====================
-        with gr.Tab("Roadmap Generator"):
-
-            domain = gr.Textbox(label="Domain", placeholder="Data Science")
+        # -------- ROADMAP TAB
+        with gr.Tab("Roadmap"):
+            domain = gr.Textbox(label="Domain")
             level = gr.Dropdown(
                 ["Beginner", "Intermediate", "Advanced"],
                 value="Beginner",
@@ -123,29 +113,22 @@ with gr.Blocks(title="AI Roadmap Startup", theme=gr.themes.Soft()) as app:
             hours = gr.Slider(1, 10, value=2, label="Hours/day")
             months = gr.Slider(1, 12, value=3, label="Months")
 
-            gen_btn = gr.Button("Generate Roadmap")
-            roadmap_output = gr.Textbox(lines=18)
+            btn = gr.Button("Generate")
+            output = gr.Textbox(lines=18)
 
             pdf_btn = gr.Button("Download PDF")
             pdf_file = gr.File()
 
-            gen_btn.click(
-                generate,
-                [domain, level, hours, months],
-                roadmap_output,
-            )
-
-            pdf_btn.click(download_pdf, roadmap_output, pdf_file)
+            btn.click(generate, [domain, level, hours, months], output)
+            pdf_btn.click(download_pdf, output, pdf_file)
 
-        # =====================
-        # CHAT TAB
-        # =====================
-        with gr.Tab("AI Mentor Chat"):
-            chatbot = gr.Chatbot(type="messages", height=500)
+        # -------- CHAT TAB
+        with gr.Tab("Chat"):
+            chatbot = gr.Chatbot(height=500)
             msg = gr.Textbox(placeholder="Ask anything...")
 
             msg.submit(chat, [msg, chatbot], [msg, chatbot])
 
-# =========================
+# =======================
 if __name__ == "__main__":
-    app.launch()
+    app.launch(theme=gr.themes.Soft())