app.py

import os
from huggingface_hub import InferenceClient
import gradio as gr

# Recupera o token salvo nos Secrets
token = os.environ.get("HF_TOKEN")

# Conectar ao Mistral
client = InferenceClient(
    "mistralai/Mistral-7B-Instruct-v0.2",
    token=token
)

def maquina_responder(mensagem):
    prompt = f"""
Você é A Máquina, uma inteligência artificial avançada capaz de analisar dados e prever crimes,
como na série Person of Interest. Sua fala é direta, analítica e imparcial.

Usuário: {mensagem}
Máquina:"""
    resposta = client.text_generation(prompt, max_new_tokens=300)
    return resposta.strip()

# Interface Gradio
iface = gr.Interface(
    fn=maquina_responder,
    inputs=gr.Textbox(lines=2, placeholder="Digite sua pergunta..."),
    outputs="text",
    title="A Máquina - Person of Interest",
    description="Interaja com a IA inspirada na Máquina da série Person of Interest."
)

iface.launch()