Update app.py

app.py

@@ -2,8 +2,34 @@ import gradio as gr
 from transformers import pipeline
 
 # Cargar un modelo de lenguaje
+from transformers import pipeline
+from datetime import datetime
+
+# Crear el chatbot
 chatbot = pipeline("text-generation", model="gpt2")
 
+def get_response(user_input):
+    # Respuestas personalizadas
+    if "hola" in user_input.lower():
+        return "¡Hola! ¿Cómo puedo ayudarte hoy?"
+    elif "hora" in user_input.lower():
+        return f"Son las {datetime.now().strftime('%H:%M')}. ¿Hay algo más en lo que pueda ayudarte?"
+    elif "adiós" in user_input.lower():
+        return "¡Hasta luego! Que tengas un buen día."
+    else:
+        # Respuesta por defecto usando el modelo
+        response = chatbot(user_input, max_length=50)
+        return response[0]['generated_text']
+
+# Bucle para interactuar con el usuario
+while True:
+    user_input = input("Tú: ")
+    if user_input.lower() == "salir":
+        break
+    response = get_response(user_input)
+    print("Asistente:", response)
+
+
 # Función de respuesta
 def responder_entrada(user_input):
     respuesta = chatbot(user_input, max_length=100, num_return_sequences=1)