Update app.py

app.py

@@ -3,14 +3,14 @@ from groq import Groq
 import os
 from fpdf import FPDF
 
-# Initialize Groq client with your API key
+# Initialize Groq client
+# Ensure your environment variable is set!
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 # ---------- ROADMAP GENERATOR ----------
 def generate(domain, level, hours, months):
     prompt = f"""
 Create a structured learning roadmap.
-
 Domain: {domain}
 Level: {level}
 Hours per day: {hours}
@@ -18,28 +18,29 @@ Months: {months}
 
 Make a weekly roadmap with projects.
 """
-
     try:
         res = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=[{"role": "user", "content": prompt}]
         )
-        text = res.choices[0].message.content
-        return text
+        return res.choices[0].message.content
     except Exception as e:
         return f"Error: {str(e)}"
 
 # ---------- PDF DOWNLOAD ----------
 def download_pdf(text):
-    if not text:
-        text = "No roadmap generated"
+    if not text or text.startswith("Error"):
+        text = "No roadmap generated or error occurred."
 
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
 
-    for line in text.split("\n"):
-        pdf.multi_cell(0, 8, line)
+    # Clean text to avoid encoding issues with FPDF
+    clean_text = text.encode('latin-1', 'ignore').decode('latin-1')
+    
+    for line in clean_text.split("\n"):
+        pdf.multi_cell(0, 10, line)
 
     path = "roadmap.pdf"
     pdf.output(path)
@@ -47,11 +48,14 @@ def download_pdf(text):
 
 # ---------- AI CHATBOT ----------
 def chat(message, history):
-    # Convert Gradio chat history to API messages format
+    """
+    Gradio history is now often a list of dicts: [{"role": "user", "content": "..."}, ...]
+    """
+    # Build the message list for Groq API
     messages = []
-    for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": bot_msg})
+    for entry in history:
+        messages.append({"role": "user", "content": entry[0]})
+        messages.append({"role": "assistant", "content": entry[1]})
     
     # Add current user message
     messages.append({"role": "user", "content": message})
@@ -63,38 +67,36 @@ def chat(message, history):
         )
         reply = res.choices[0].message.content
     except Exception as e:
-        reply = str(e)
+        reply = f"Error: {str(e)}"
 
-    # Append current turn to history for Gradio display
+    # Update history: Gradio Chatbot expects a list of tuples/lists [(user, bot)]
     history.append((message, reply))
-    return history, history
+    return "", history  # Clear the input box and return updated history
 
 # ---------- GRADIO UI ----------
 with gr.Blocks() as app:
     gr.Markdown("# 🚀 AI Roadmap Startup")
 
-    # ----- Roadmap Generator Tab -----
     with gr.Tab("Roadmap Generator"):
-        domain = gr.Textbox(label="Domain")
-        level = gr.Dropdown(["Beginner","Intermediate","Advanced"], label="Level")
+        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")
         months = gr.Slider(1, 12, value=3, label="Months")
 
         btn = gr.Button("Generate")
-        output = gr.Textbox(lines=20, label="Generated Roadmap")
+        output = gr.Textbox(lines=15, label="Generated Roadmap")
         pdf_btn = gr.Button("Download PDF")
         pdf_file = gr.File(label="PDF File")
 
         btn.click(generate, [domain, level, hours, months], output)
         pdf_btn.click(download_pdf, output, pdf_file)
 
-    # ----- AI Chatbot Tab -----
     with gr.Tab("AI Chatbot"):
         chatbot = gr.Chatbot()
-        msg = gr.Textbox(label="Your Message")
-        state = gr.State([])
-
-        msg.submit(chat, [msg, state], [chatbot, state])
+        msg = gr.Textbox(label="Your Message", placeholder="Type and press Enter...")
+        
+        # We use the chatbot component itself to store history
+        msg.submit(chat, [msg, chatbot], [msg, chatbot])
 
-# Launch the app
-app.launch()
+if __name__ == "__main__":
+    app.launch()