import os
import gradio as gr
import google.generativeai as genai
from dotenv import load_dotenv
from prompts import system_prompt

load_dotenv()
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

model = genai.GenerativeModel(
    model_name="gemini-2.0-flash",
    generation_config={
        "temperature": 0.9,
        "top_p": 1,
        "max_output_tokens": 2048,
    }
)

def chat(message, history):
    try:
        messages = [
            {"role": "user", "parts": [system_prompt]},
            *[{"role": "user", "parts": [msg[0]]} for msg in history],
            {"role": "user", "parts": [message]}
        ]
        response = model.generate_content(messages)
        return response.text
    except Exception as e:
        return f"Error: {e}"

demo = gr.ChatInterface(
    fn=chat,
    examples=[
        "¿Qué incluye el curso CopyXpert?",
        "¿Cuál es el precio del curso?",
        "¿Cómo puedo inscribirme?",
    ],
    title="🤖Chucho Bot - CopyXpert Sales Assistant",
    description="Hi! I'm Chucho Bot, your personal assistant for the CopyXpert course. How can I help you today?"
)

demo.launch()