Update app.py

app.py

@@ -1,53 +1,27 @@
 import gradio as gr
 from transformers import pipeline
-
-# Cargar un modelo de lenguaje
-from transformers import pipeline
 from datetime import datetime
 
-# Crear el chatbot
+# Crear el chatbot usando el modelo de transformers
 chatbot = pipeline("text-generation", model="gpt2")
 
+# Función para obtener la respuesta del bot
 def get_response(user_input):
-    # Respuestas personalizadas
+    # Respuestas personalizadas según ciertas palabras clave
     if "hola" in user_input.lower():
         return "¡Hola, soy Jarbot! ¿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, espero que nos volvamos a ver."
+        return "¡Hasta luego! Que tengas un buen día."
     else:
-        # Respuesta por defecto usando el modelo
+        # Respuesta por defecto usando el modelo GPT-2
         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)
-    return respuesta[0]["generated_text"]
-
-# Interfaz Gradio
-with gr.Blocks() as interfaz:
-    gr.Markdown("# Asistente Virtual")
-    chat = gr.Chatbot()
-    entrada_usuario = gr.Textbox(label="Escribe tu mensaje:")
-    boton_enviar = gr.Button("Enviar")
-    
-    def enviar_mensaje(user_input, history):
-        history = history or []
-        respuesta = responder_entrada(user_input)
-        history.append((user_input, respuesta))
-        return history, ""
-
-    boton_enviar.click(enviar_mensaje, [entrada_usuario, chat], [chat, entrada_usuario])
+# Crear la interfaz con Gradio
+iface = gr.Interface(fn=get_response, inputs="text", outputs="text", 
+                     title="Asistente Virtual", description="¡Escribe algo para interactuar con el asistente!")
 
-interfaz.launch()
+# Lanzar la interfaz
+iface.launch()