Update app.py

app.py

@@ -1,129 +1,90 @@
 import gradio as gr
-import os
 from groq import Groq
-from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
-from reportlab.lib.styles import getSampleStyleSheet
-
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
-memory = {"roadmap": ""}
-
-# ---------------- PDF ----------------
-def create_pdf(text):
-    file_path = "roadmap.pdf"
-
-    doc = SimpleDocTemplate(file_path, pagesize=letter)
-    styles = getSampleStyleSheet()
-    story = []
-
-    for line in text.split("\n"):
-        story.append(Paragraph(line, styles["Normal"]))
-        story.append(Spacer(1, 8))
-
-    doc.build(story)
-    return file_path
+import os
+from fpdf import FPDF
 
-# ---------------- GENERATE ----------------
-def generate(field, hours, months, level):
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
+# ---------- ROADMAP ----------
+def generate(domain, level, hours, months):
     prompt = f"""
-Create a professional learning roadmap.
+Create a structured learning roadmap.
 
-Field: {field}
+Domain: {domain}
 Level: {level}
-Hours/day: {hours}
+Hours per day: {hours}
 Months: {months}
 
-Make it structured with:
-- monthly plan
-- weekly goals
-- projects
-- tools
-- final outcome
+Make weekly roadmap with projects.
 """
 
-    res = client.chat.completions.create(
-        model="llama-3.3-70b-versatile",
-        messages=[{"role": "user", "content": prompt}]
-    )
+    try:
+        res = client.chat.completions.create(
+            model="llama3-70b-8192",
+            messages=[{"role": "user", "content": prompt}]
+        )
+        text = res.choices[0].message.content
+        return text
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-    roadmap = res.choices[0].message.content
-    memory["roadmap"] = roadmap
 
-    pdf = create_pdf(roadmap)
-    return roadmap, pdf
+# ---------- PDF ----------
+def download_pdf(text):
+    if not text:
+        text = "No roadmap generated"
 
-# ---------------- CHAT ----------------
-def chat_with_roadmap(message, history):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
 
-    if memory["roadmap"] == "":
-        reply = "Generate a roadmap first."
-    else:
-        prompt = f"""
-You are helping with this roadmap:
+    for line in text.split("\n"):
+        pdf.multi_cell(0, 8, line)
 
-{memory['roadmap']}
+    path = "roadmap.pdf"
+    pdf.output(path)
+    return path
 
-User: {message}
-"""
 
+# ---------- CHATBOT ----------
+def chat(message, history):
+    try:
         res = client.chat.completions.create(
-            model="llama-3.3-70b-versatile",
-            messages=[{"role":"user","content":prompt}]
+            model="llama3-70b-8192",
+            messages=[{"role": "user", "content": message}]
         )
-
         reply = res.choices[0].message.content
+    except Exception as e:
+        reply = str(e)
 
-    history = history + [
-        {"role": "user", "content": message},
-        {"role": "assistant", "content": reply}
-    ]
-
+    history.append((message, reply))
     return history, history
 
-# ---------------- UI ----------------
-with gr.Blocks(theme=gr.themes.Soft(), css="""
-#title {text-align:center;font-size:32px;font-weight:700}
-""") as app:
-
-    gr.Markdown("<div id='title'>🚀 Roadmap AI</div>")
-
-    with gr.Row():
-
-        # -------- SIDEBAR --------
-        with gr.Column(scale=1):
-            gr.Markdown("### ⚙️ Setup")
-
-            field = gr.Textbox(label="Field")
-            hours = gr.Textbox(label="Hours per day")
-            months = gr.Textbox(label="Months")
-            level = gr.Dropdown(["Beginner","Intermediate","Advanced"])
-
-            generate_btn = gr.Button("Generate Roadmap", size="lg")
 
-            pdf_file = gr.File(label="📥 Download PDF")
+# ---------- UI ----------
+with gr.Blocks(theme=gr.themes.Soft()) as app:
+    gr.Markdown("# 🚀 AI Roadmap Startup")
 
-        # -------- MAIN PANEL --------
-        with gr.Column(scale=2):
+    with gr.Tab("Roadmap Generator"):
+        domain = gr.Textbox(label="Domain")
+        level = gr.Dropdown(["Beginner","Intermediate","Advanced"])
+        hours = gr.Slider(1,10,value=2,label="Hours/day")
+        months = gr.Slider(1,12,value=3,label="Months")
 
-            roadmap_output = gr.Markdown()
+        btn = gr.Button("Generate")
 
-            gr.Markdown("### 💬 Chat with your roadmap")
+        output = gr.Textbox(lines=20)
+        pdf_btn = gr.Button("Download PDF")
+        pdf_file = gr.File()
 
-            chatbot = gr.Chatbot(type="messages", height=400)
-            msg = gr.Textbox(placeholder="Ask anything about your roadmap...")
-            send = gr.Button("Send")
+        btn.click(generate, [domain,level,hours,months], output)
+        pdf_btn.click(download_pdf, output, pdf_file)
 
-    generate_btn.click(
-        generate,
-        inputs=[field, hours, months, level],
-        outputs=[roadmap_output, pdf_file]
-    )
+    with gr.Tab("AI Chatbot"):
+        chatbot = gr.Chatbot()
+        msg = gr.Textbox()
+        state = gr.State([])
 
-    send.click(
-        chat_with_roadmap,
-        inputs=[msg, chatbot],
-        outputs=[chatbot, chatbot]
-    )
+        msg.submit(chat, [msg,state], [chatbot,state])
 
 app.launch()