import gradio as gr
from mistralai import Mistral, UserMessage

import os

title = "Gaia Mistral Chat Demo"
description = "Example of simple chatbot with Gradio and Mistral AI via its API"
placeholder = "Posez moi une question sur l'agriculture"
examples = ["Comment fait on pour produire du maïs ?", "Rédige moi une lettre pour faire un stage dans une exploitation agricole", "Comment reprendre une exploitation agricole ?"]
api_key = os.environ.get("MISTRAL_API_KEY")

#client = MistralClient(api_key=api_key)
client = Mistral(api_key=api_key)
model = 'open-mixtral-8x7b'
"""
def chat_with_mistral(user_input, history):
    messages = [{"role": "user", "content": user_input}]

    chat_response = client.chat.complete(model=model, messages=messages)
    return chat_response.choices[0].message.content
"""

def chat_with_mistral(user_input, history):
    messages = [{"role": "user", "content": user_input}]
    return "This is a dummy response"

iface = gr.ChatInterface(
    fn=chat_with_mistral,
    chatbot=gr.Chatbot(height=300),
    textbox=gr.Textbox(placeholder=placeholder, container=False, scale=7),
    title=title,
    description=description,
    theme="soft",
    examples=examples,
    cache_examples=True,
)

iface.launch(share=True)