Update app.py

app.py

@@ -7,6 +7,9 @@ import requests
 # Crear el chatbot usando el modelo de transformers
 pipe = pipeline("text-generation", model="Isotonic/gpt-human-assistant")
 
+# Lista de malas palabras a filtrar
+malas_palabras = ["maldito", "estúpido", "imbécil", "tonto"]  # Puedes agregar más malas palabras aquí
+
 # Función para obtener el clima en Viña del Mar con Open-Meteo
 def get_weather():
     url = "https://api.open-meteo.com/v1/forecast?latitude=-33.0153&longitude=-71.5503&current_weather=true"
@@ -19,8 +22,19 @@ def get_weather():
     else:
         return "Lo siento, no pude obtener el clima en este momento."
 
+# Función para filtrar malas palabras
+def filtrar_malas_palabras(texto):
+    for palabra in malas_palabras:
+        if palabra in texto.lower():
+            return True
+    return False
+
 # Función para obtener la respuesta del bot
 def get_response(user_input):
+    # Comportamiento general
+    if filtrar_malas_palabras(user_input):
+        return "Por favor, mantén un lenguaje respetuoso. No se toleran malas palabras."
+
     # Respuestas personalizadas según ciertas palabras clave
     if "hola" in user_input.lower():
         return "¡Hola, soy Jarbot! ¿Cómo puedo ayudarte hoy?"
@@ -29,7 +43,7 @@ def get_response(user_input):
     if "actividades" in user_input.lower():
         return "¡Sii! Hoy tenemos..."
     if "especies" in user_input.lower():
-        return "Pues hay muchas! Aquí puedes encontrar una variedad de plantas como [cactarios, colecciones de la isla de Juan Fernández, algunos del bosque esclerófilo de la zona central y mucho más] y también hay animales como [ganzos, tortugas, etc]. ¿No es lindo?."
+        return "Pues hay muchas! Aquí puedes encontrar una variedad de plantas como [cactarios, colecciones de la isla de Juan Fernández, algunos del bosque esclerófilo de la zona central y mucho más] y también hay animales como [ganzos, tortugas, etc]. ¿No es lindo?"
     if "reglas" in user_input.lower():
         return "Pues, para preservar el entorno, te pido que no recojas plantas, respetes las áreas designadas y mantengas el Jardín limpio, nos encanta saber que cuidas del Jardín"
     if "picnic" in user_input.lower():
@@ -48,9 +62,8 @@ def get_response(user_input):
     elif "adiós" in user_input.lower():
         return "¡Bye amigo! Que tengas un buen día."
     else:
-        # Respuesta por defecto usando el modelo GPT-2
-        response = pipe(user_input, max_length=50)
-        return response[0]['generated_text']
+        # Si la pregunta no es entendida, pide reformularla
+        return "Lo siento, no entendí bien la pregunta. ¿Podrías reformularla?"
 
 # Crear la interfaz con Gradio
 iface = gr.Interface(fn=get_response, inputs="text", outputs="text", 
@@ -58,4 +71,3 @@ iface = gr.Interface(fn=get_response, inputs="text", outputs="text",
 
 # Lanzar la interfaz
 iface.launch()
-