Update app.py

app.py

@@ -1,90 +1,19 @@
 import gradio as gr
-from transformers import pipeline
-
-# Cargar el modelo más rápido y compatible
-chatbot = pipeline(
-    "text2text-generation",
-    model="google/flan-t5-small",
-    max_new_tokens=256,
-    temperature=0.7
-)
-
-# Función con prompt emocional
-def responder(mensaje, enfermedad):
-    prompt = f"""
-Eres FamiIA, un asistente emocional y empático. Acompañas a personas que cuidan o apoyan a alguien con {enfermedad}.
-Tu tarea es brindar:
-- apoyo emocional
-- orientación práctica
-- contención sin juicio
-- respuestas cálidas, humanas y comprensivas
-
-Mensaje del usuario:
-"{mensaje}"
-"""
-    resultado = chatbot(prompt)[0]["generated_text"]
-    return resultado.strip()
-
-# Estilo visual amigable
-custom_css = """
-body {
-    background-color: #f8f9fa;
-    font-family: 'Segoe UI', sans-serif;
-    color: #333;
-}
-.gradio-container {
-    max-width: 900px;
-    margin: auto;
-    padding: 20px;
-}
-textarea, input, select {
-    border-radius: 10px !important;
-}
-h1, h2, h3 {
-    color: #212529;
-    text-align: center;
-}
-"""
-
-# Interfaz de Gradio
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("""
-# 🩵 FamiIA — Tu Compañera Emocional
-
-### Un lugar seguro para familiares de personas con enfermedades raras.
-
-Puedes hablar con FamiIA para sentirte escuchado/a, orientado/a o simplemente acompañado/a.
-
----
-
-### 💬 Ejemplos:
-
-• Estoy agotada por cuidar a mi hijo con Ehlers-Danlos. ¿Qué hago?  
-• ¿Cómo puedo explicar esto a sus profesores?  
-• ¿Dónde puedo encontrar ayuda para mi bienestar emocional?  
-• Solo quiero escribir lo que siento hoy…
-""")
-
-    enfermedad = gr.Dropdown(
-        ["Ehlers-Danlos", "Síndrome de Rett", "Fibrosis Quística", "Otra"],
-        label="Selecciona la enfermedad",
-        value="Ehlers-Danlos"
-    )
-
-    entrada = gr.Textbox(
-        placeholder="Escribe cómo te sientes o qué necesitas...",
-        label="¿Qué quieres compartir hoy?"
-    )
-
-    salida = gr.Textbox(label="Respuesta de FamiIA")
-
-    boton = gr.Button("Hablar con FamiIA")
-
-    def conversar(mensaje, enfermedad):
-        if not mensaje.strip():
-            return "Por favor, escribe cómo te sientes o qué necesitas preguntar."
-        return responder(mensaje, enfermedad)
-
-    boton.click(fn=conversar, inputs=[entrada, enfermedad], outputs=salida)
-
-demo.launch()
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# Cargar modelo
+model_name = "RayyanAhmed9477/CPU-Compatible-Mental-Health-Model"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+def responder(mensaje):
+    inputs = tokenizer(mensaje, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=200, pad_token_id=tokenizer.eos_token_id)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+gr.Interface(fn=responder, 
+             inputs="text", 
+             outputs="text", 
+             title="Asistente de Salud Mental (IA)",
+             description="IA que acompaña emocionalmente a familiares de pacientes con enfermedades raras."
+            ).launch()