import os  # Biblioteca para interagir com o sistema operacional
import gradio as gr  # Importa Gradio para criar a interface web
from groq import Groq  # Importa o cliente da API Groq

# Defina sua chave da API do GROQ
GROQ_API_KEY = os.getenv('GROQ_API_KEY', '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')  # Substitua pela sua chave

# Inicializa o cliente da API Groq
client = Groq(api_key=GROQ_API_KEY)

# Função para processar a entrada do usuário e gerar uma resposta
def assistente_agricultura(user_prompt):
    if user_prompt.strip() == "15":
        return "Encerrando assistente Agro! Até mais! 🚜"

    # Faz a solicitação ao modelo
    completion = client.chat.completions.create(
        model="llama3-8b-8192",
        messages=[
            {"role": "system", "content": "Você é um assistente especializado em agricultura."},
            {"role": "user", "content": user_prompt}
        ],
        temperature=0,
        max_tokens=1024,
        top_p=1,
        stream=False
    )

    # Retorna a resposta do assistente
    return completion.choices[0].message.content

# Configuração da interface Gradio
iface = gr.Interface(
    fn=assistente_agricultura,
    inputs=gr.Textbox(lines=2, placeholder="Digite sua pergunta sobre agricultura..."),
    outputs="text",
    title="🌾 Assistente Agro IA",
    description="Digite sua pergunta sobre agricultura e receba respostas de IA especializadas! 🚜",
    live=True
)

# Executa a interface web
if __name__ == "__main__":
    iface.launch()