import os
import gradio as gr
import openai

openai.api_key = os.getenv("GROQ_API_KEY")
openai.api_base = "https://api.groq.com/openai/v1"

def ask_okitakoy(prompt):
    try:
        response = openai.chat.completions.create(
            model="mixtral-8x7b-32768",
            messages=[{"role": "user", "content": prompt}]
        )
        return response.choices[0].message.content
    except Exception as e:
        return f"Erreur : {str(e)}"

iface = gr.Interface(
    fn=ask_okitakoy,
    inputs=gr.Textbox(placeholder="Pose ta question à Okitakoy...", lines=2),
    outputs="text",
    title="Okitakoy - Mon Assistant IA (via Groq)",
    description="Assistant IA rapide, gratuit et intelligent basé sur Mixtral via Groq."
)

iface.launch()