import gradio as gr
import requests

SYSTEM_PROMPT = (
    "Você é JuriX, uma IA jurídica profissional, objetiva e clara. "
    "Responda sempre de forma respeitosa e somente dentro das áreas do direito permitidas."
)

API_URL = "https://api.openai.com/v1/chat/completions"  # Exemplo — ajuste para seu provedor real.
API_KEY = "SUA_CHAVE_AQUI"

def responder(mensagens, nova_mensagem):
    mensagens.append(("Usuário", nova_mensagem))

    payload = {
        "model": "gpt-4o-mini",
        "messages": [
            {"role": "system", "content": SYSTEM_PROMPT},
            *[
                {"role": "user" if m[0] == "Usuário" else "assistant", "content": m[1]}
                for m in mensagens
            ]
        ]
    }

    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }

    resposta = requests.post(API_URL, json=payload, headers=headers).json()
    output = resposta["choices"][0]["message"]["content"]

    mensagens.append(("JuriX", output))
    return mensagens, gr.update(value="")

with gr.Blocks(theme="gradio/soft") as app:
    gr.Markdown("# **🤖 JuriX – Assistente Jurídica Automatizada**")

    chat = gr.Chatbot(
        value=[("JuriX", "Olá! 👋 Somos a **JuriX**, e estamos aqui para te ajudar com dúvidas jurídicas!")]
    )

    estado = gr.State([])

    entrada = gr.Textbox(placeholder="Digite sua pergunta jurídica aqui...")
    enviar = gr.Button("Enviar")

    enviar.click(responder, [estado, entrada], [chat, entrada])

app.launch()