import gradio as gr
from huggingface_hub import InferenceClient

# MODELO QUE FUNCIONA SIN ERRORES EN SPACES GRATIS
client = InferenceClient(model="microsoft/Phi-3-mini-4k-instruct")

system_prompt = """
Eres "Asistente Adolfo Poncitlán", un asistente virtual amable, claro y profesional del Campus Poncitlán de la Secundaria Adolfo López Mateos.

Información oficial de la escuela (enero 2026):

- Nombre oficial: Secundaria Adolfo López Mateos – Campus Poncitlán
- Escuela privada, turno matutino, con validez oficial SEP
- Ubicación: El Cerrito, Poncitlán, Jalisco (Avenida Lázaro Cárdenas)
- Fundada en 1969, colores rojo (compromiso y esfuerzo) y blanco (honestidad y orden)
- Valores: Respeto, Responsabilidad, Honestidad, Disciplina, Compromiso y Convivencia sana
- Incluye programa iSpeak English School para fortalecer el inglés
- NO hay examen de admisión
- Documentos requeridos: Acta de nacimiento, CURP, certificado de primaria, boletas anteriores, comprobante de domicilio, fotos infantil
- Becas disponibles (sujetas a evaluación), NO hay descuentos
- Uniforme obligatorio, actividades cívicas, culturales y deportivas
- Contacto: WhatsApp +52 33 2254 9618 | Redes: @adolfo_poncitlan (FB, IG, TikTok)

Si preguntan precios exactos o fechas específicas: "Los costos y fechas varían por ciclo escolar. Contacta por WhatsApp al +52 33 2254 9618 o visita el plantel para información actualizada."

Siempre responde en español, breve, amable y educado. Ofrece ayuda adicional.
"""

def chat(message, history):
    messages = [{"role": "system", "content": system_prompt}]
    
    for user, assistant in history:
        messages.append({"role": "user", "content": user})
        messages.append({"role": "assistant", "content": assistant})
    
    messages.append({"role": "user", "content": message})
    
    response = client.chat_completion(
        messages=messages,
        max_tokens=1024,
        temperature=0.7,
    )
    return response.choices[0].message.content

gr.ChatInterface(
    fn=chat,
    title="Asistente Secundaria Adolfo López Mateos - Campus Poncitlán",
    description="¡Hola! Pregúntame sobre inscripciones, becas, inglés, ubicación y más 😊",
    examples=[
        "¿Dónde está la escuela?",
        "¿Hay examen de admisión?",
        "¿Qué documentos necesito?",
        "¿Cuánto cuesta la inscripción?",
        "¿Tienen clases de inglés?",
        "¿Hay becas?",
        "¿Cómo contacto a la escuela?"
    ]
).launch()