import os
import gradio as gr
from groq import Groq

client = Groq(api_key=os.getenv("AiRoadmapGenerator"))

def build_prompt(domain, level, days):
    return f"""
Create a detailed learning roadmap for:
Domain: {domain}
Level: {level}
Duration: {days} days

Include:
- Weekly plan
- Daily goals
- Tools
- Resources
- Projects
"""

def generate_roadmap(domain, level, days):
    prompt = build_prompt(domain, level, days)

    response = client.chat.completions.create(
        model="llama3-70b-8192",
        messages=[{"role": "user", "content": prompt}],
        temperature=0.6,
        max_tokens=2000
    )

    return response.choices[0].message.content

app = gr.Interface(
    fn=generate_roadmap,
    inputs=[
        gr.Textbox(label="Domain"),
        gr.Dropdown(["Beginner","Intermediate","Advanced"]),
        gr.Number(label="Days")
    ],
    outputs="markdown",
    title="AI Learning Roadmap Generator"
)

app.launch()