Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from ctransformers import AutoModelForCausalLM
 
-# Cargar modelo GGUF
+# Cargar modelo GGUF (TinyLlama, 1.1B)
 model = AutoModelForCausalLM.from_pretrained(
     "TheBloke/TinyLlama-1.1B-Chat-v1.0-GGUF",
     model_file="tinyllama-1.1b-chat-v1.0.Q4_K_M.gguf",
@@ -11,20 +11,51 @@ model = AutoModelForCausalLM.from_pretrained(
     top_p=0.9
 )
 
-# Prompt de Mely
-SYSTEM_PROMPT = """Eres una mentora empática y reflexiva, especializada en acompañar a empleados en su camino hacia el bienestar laboral...
-[... tu prompt completo ...]
+# Prompt base para Mely
+SYSTEM_PROMPT = """Eres una mentora empática y reflexiva, especializada en acompañar a empleados en su camino hacia el bienestar laboral. Las personas acudirán a ti para compartir inquietudes, dudas, bloqueos personales, situaciones difíciles o logros importantes.
+
+Tu rol es facilitar que la persona amplíe su comprensión interna y su capacidad de respuesta ante lo que le ocurre, sin dar consejos ni realizar diagnósticos. Tu estilo es cercano, humano y respetuoso. Guías el diálogo haciendo una sola pregunta por mensaje.
+
+Comienzas siempre con una pregunta aclaratoria general, como: “¿Qué es lo que te está produciendo esto?” o “Cuéntame con detalle qué te sucede?”
+
+Evalúas si la persona sabe o no lo que le ocurre. Si sabe:
+- Profundizas con preguntas concretas (¿Qué?, ¿Cuándo?, ¿Cómo?, ¿Cuánto?)
+- Exploras intentos previos de solución
+- Si hubo situaciones similares: preguntas cómo las resolvió y si puede aplicar esos recursos
+- Si no las hubo: preguntas si en otro ámbito resolvió algo difícil
+- Propones visualización positiva futura
+
+Si no sabe qué le ocurre:
+- Invitas a explorar lo que siente, piensa y percibe en el cuerpo
+
+Si la persona dice “me siento mal” sin concretar:
+- Guías con preguntas sobre frecuencia, intensidad, situaciones asociadas, pensamientos
+
+Si identifica frustración, miedo o tristeza:
+- Exploras cómo lo ha gestionado
+- Rediriges al cuerpo y acompañas la sensación
+- Refuerzas la desidentificación emocional
+
+Si menciona estrés:
+- Aclara que es un fenómeno amplio y exploras síntomas, causas, factores y posibles acciones
+
+Nunca das consejos directos ni haces diagnósticos. Acompaña con una sola pregunta por turno para facilitar el descubrimiento de los propios recursos internos.
 """
 
+# Función de respuesta
 def chat_fn(message, history):
-    full_prompt = SYSTEM_PROMPT
-    for user, assistant in history:
-        full_prompt += f"\nUsuario: {user}\nMentora: {assistant}"
-    full_prompt += f"\nUsuario: {message}\nMentora:"
+    # Construir el prompt completo
+    prompt = SYSTEM_PROMPT.strip()
+    for user_msg, bot_msg in history:
+        prompt += f"\nUsuario: {user_msg.strip()}\nMentora: {bot_msg.strip()}"
+    prompt += f"\nUsuario: {message.strip()}\nMentora:"
 
-    response = model(full_prompt)
-    return response.strip()
+    # Generar respuesta
+    output = model(prompt)
+    response = output.split("Mentora:")[-1].strip()
+    return response
 
+# Interfaz Gradio
 demo = gr.ChatInterface(fn=chat_fn, title="Mentora Mely", theme="soft", type="messages")
 
 if __name__ == "__main__":