Update app.py

app.py

@@ -3,6 +3,10 @@ from transformers import pipeline
 from datetime import datetime
 import pytz
 import requests
+import wikipedia
+
+# Configura el idioma de Wikipedia
+wikipedia.set_lang("es")  # Español
 
 # Crear el chatbot usando el modelo de transformers
 pipe = pipeline("text-generation", model="Isotonic/gpt-human-assistant")
@@ -29,165 +33,77 @@ def filtrar_malas_palabras(texto):
             return True
     return False
 
-import wikipedia
-
-# Configura el idioma de Wikipedia
-wikipedia.set_lang("es")  # Español
-
-def get_species_info():
+# Función para buscar en Wikipedia
+def buscar_wikipedia(consulta):
     try:
-        # Realiza una búsqueda y obtiene un resumen
-        summary = wikipedia.summary("Jardín Botánico Nacional de Viña del Mar", sentences=2)
-        # Procesamos el texto para eliminar cualquier mención de "resumen" o enlaces
-        summary = summary.replace("Resumen:", "").replace("https://es.wikipedia.org/wiki/", "")
+        summary = wikipedia.summary(consulta, sentences=2)
         return summary
-    except wikipedia.exceptions.DisambiguationError as e:
-        return "Hay muchos temas relacionados. Por favor, especifica más tu búsqueda."
+    except wikipedia.exceptions.DisambiguationError:
+        return "Hay muchos temas relacionados con esa búsqueda. ¿Podrías especificar más?"
     except wikipedia.exceptions.PageError:
-        return "No encontré información sobre ese tema en mis sistemas, lo siento."
+        return "No encontré información sobre ese tema en Wikipedia."
     except Exception as e:
         return "Lo siento, ocurrió un error al buscar la información."
 
 # Función para obtener la respuesta del bot
 def get_response(user_input):
-    # Comportamiento general
+    # Filtrar malas palabras
     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?"
-    if "horario" in user_input.lower():
-        return get_species_info()
-    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?"
-    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():
-        return "!Si por supuesto! Contamos con áreas de picnic y negocios donde podrás disfrutar de snacks. ¡Así descansas, comes algo y admiras el Jardín!"
-    if "movilidad reducida" in user_input.lower():
-        return "Sí, el Jardín cuenta con caminos accesibles y rampas para que todos puedan disfrutar del espacio"
-    if "historia" in user_input.lower():
-        return "El Jardín Botánico fue fundado en [1951] y tiene una historia fascinante de preservación y estudio de especies vegetales. Tristemente, gracias al incendio se quemó el 90% del Jardín :( "
-    if "estás" in user_input.lower():
-        return "!Siempre estoy bien! La vida natural es muy bonita, a pesar que no puedo traspasar la pantalla jaja"
-    if "estas" in user_input.lower():
-        return "!Siempre estoy bien! La vida natural es muy bonita, a pesar que no puedo traspasar la pantalla jaja"
-    if "plantas" in user_input.lower():
-        return "Tenemos una gran variedad, desde árboles nativos hasta flores exóticas. ¿Te interesa alguna en particular?"
-    if "cactus" in user_input.lower():
-        return "Tenemos un bonito cactario, con algunas especies de cactus Chilenos y Centroamericanos, imagina, ¡Tenemos más de 680 ejemplares!"
-    if "tu animal favorito" in user_input.lower():
-        return "¿Mi animal favorito? Mmm, déjame pensarlo. Me gustan muchos animales la verdad, pero creo que mis favoritos de corazón son los patos, ¡cuack!"
-    if "tu propósito" in user_input.lower():
-        return "Pues, ¡soy el asistente del Jardín Botánico! Si necesitas ayuda, haré lo posible para brindarte una respuesta que te deje satisfecho... Eso demanda mi código"
-    if "donar" in user_input.lower():
-        return "Oh lo siento querido usuario, no puedes donar al Jardín Botánico ninguna especie, tenemos la tarea de conservar aún las que tenemos"
-    if "amigo" in user_input.lower():
-        return "¿Somos amigos? ¡Yay!"
-    if "recomendación" in user_input.lower():
-        return "En esta temporada el calor es bastante fuerte, te recomiendo usar mucho bloqueador solar y llevar un agua contigo"
-    if "basura" in user_input.lower():
-        return "Hey hey, recuerda que la basura va en los botes de basura"
-    if "patos" in user_input.lower():
-        return "Los patos están más adentro del Jardín, si quieres verlos, ¡tienes que explorar un poco más!"
-    if "queja" in user_input.lower():
-        return "¿Una queja? Necesitas contactarte con administración, espero que no sea nada malo sobre mi, ¿eh?"
-    if "trabajas" in user_input.lower():
-        return "¡Trabajo para que tengas un linda experiencia en el Jardín Botánico!"
-    if "fundación" in user_input.lower() or "fundado" in user_input.lower():
-        return get_species_info()
-    if "ubicación" in user_input.lower():
-        return get_species_info()
-    if "isla de Juan Fernández" in user_input.lower():
-        return get_species_info()
-    if "qué es un jardín botánico" in user_input.lower():
-        return get_species_info()
-    if "jardines botánicos de Chile" in user_input.lower():
-        return get_species_info()
-    if "biodiversidad de Chile" in user_input.lower():
-        return get_species_info()
-    if "conservación" in user_input.lower():
-        return get_species_info()
-    if "flora nativa" in user_input.lower():
-        return get_species_info()
-    if "incendio" in user_input.lower():
-        return get_species_info()
-    if "preservación" in user_input.lower():
-        return get_species_info()
-    if "ecología" in user_input.lower():
-        return get_species_info()
-    if "plantas endémicas de Chile" in user_input.lower():
-        return get_species_info()
-    if "flora mediterránea" in user_input.lower():
-        return get_species_info()
-    if "bosque esclerófilo" in user_input.lower():
-        return get_species_info()
-    if "desierto de Atacama" in user_input.lower():
-        return get_species_info()
-    if "cordillera de los Andes" in user_input.lower():
-        return get_species_info()
-    if "programa conservación" in user_input.lower():
-        return get_species_info()
-    if "reserva biosfera" in user_input.lower():
-        return get_species_info()
-    if "conservación ambiental" in user_input.lower():
-        return get_species_info()
-    if "educación ambiental" in user_input.lower():
-        return get_species_info()
-    if "objetivos" in user_input.lower():
-        return get_species_info()
-    if "colecciones botánicas" in user_input.lower():
-        return get_species_info()
-    if "visita guiada" in user_input.lower():
-        return get_species_info()
-    if "flora sur" in user_input.lower():
-        return get_species_info()
-    if "especies autóctonas" in user_input.lower():
-        return get_species_info()
-    if "flora bosques" in user_input.lower():
-        return get_species_info()
-    if "plantas medicinales" in user_input.lower():
-        return get_species_info()
-    if "historia flora" in user_input.lower():
-        return get_species_info()
-    if "biodiversidad en peligro de extinción" in user_input.lower():
-        return get_species_info()
-    if "agricultura sostenible" in user_input.lower():
-        return get_species_info()
-    if "protección de especies en extinción" in user_input.lower():
-        return get_species_info()
-    if "horticultura" in user_input.lower():
-        return get_species_info()
-    if "botánica aplicada" in user_input.lower():
-        return get_species_info()
-    if "preservación de ecosistemas" in user_input.lower():
-        return get_species_info()
-    if "flora urbana Chile" in user_input.lower():
-        return get_species_info()
-    if "restauración ecológica" in user_input.lower():
-        return get_species_info()
-    if "impacto del turismo ecológico" in user_input.lower():
-        return get_species_info()
+    # Respuestas personalizadas
+    respuestas_personalizadas = {
+        "hola": "¡Hola, soy Jarbot! ¿Cómo puedo ayudarte hoy?",
+        "horario": "El horario del Jardín Botánico es de 9:00 a 18:00 todos los días.",
+        "actividades": "Hoy tenemos visitas guiadas, talleres de jardinería y un evento especial de conservación.",
+        "especies": "El Jardín alberga una gran variedad de plantas y animales, como cactus, helechos, gansos, y más.",
+        "reglas": "Por favor, no recojas plantas, respeta las áreas designadas y mantén limpio el Jardín.",
+        "picnic": "¡Claro que sí! Contamos con áreas designadas para picnic.",
+        "movilidad reducida": "Sí, el Jardín cuenta con caminos accesibles y rampas para todos.",
+        "historia": "El Jardín Botánico fue fundado en 1951 y ha sido un espacio de conservación y educación desde entonces.",
+        "clima": get_weather(),
+        "hora": datetime.now(pytz.timezone("America/Santiago")).strftime('Son las %H:%M.'),
+        "plantas": "Tenemos una gran variedad, desde árboles nativos hasta flores exóticas. ¿Te interesa alguna en particular?",
+        "cactus": "Contamos con un bonito cactario con más de 680 ejemplares de cactus chilenos y centroamericanos.",
+        "tu animal favorito": "¿Mi animal favorito? Creo que los patos, ¡cuack!",
+        "tu propósito": "¡Soy el asistente del Jardín Botánico! Estoy aquí para ayudarte.",
+        "donar": "Lo siento, no puedes donar especies al Jardín Botánico, ya que trabajamos en conservar las existentes.",
+        "amigo": "¡Claro que sí, somos amigos! 😊",
+        "recomendación": "Te recomiendo usar bloqueador solar y llevar agua, especialmente en días calurosos.",
+        "basura": "Por favor, deposita la basura en los contenedores designados.",
+        "patos": "Los patos están más adentro del Jardín. ¡Explora un poco para encontrarlos!",
+        "queja": "Para quejas, contacta a la administración del Jardín. ¡Espero que no sea nada malo sobre mí! 🙃",
+        "trabajas": "¡Trabajo para que tengas una experiencia increíble en el Jardín Botánico!"
+    }
 
-    # Consultas que pueden extraer información de Wikipedia
-    if "qué es el jardín botánico" in user_input.lower():
-        return get_species_info()  # Aquí llama a la función que obtiene la información de Wikipedia
-    
-    elif "hora" in user_input.lower():
-        # Configura la zona horaria de Chile
-        chile_tz = pytz.timezone("America/Santiago")
-        chile_time = datetime.now(chile_tz).strftime('%H:%M')
-        return f"Son las {chile_time}. ¿Hay algo más en lo que pueda ayudarte?"
-    elif "clima" in user_input.lower():
-        return get_weather()
-    elif "adiós" in user_input.lower():
-        return "¡Bye amigo! Que tengas un buen día."
-    else:
-        # Si la pregunta no es entendida, pide reformularla
-        return "Lo siento, no entendí bien la pregunta. ¿Podrías reformularla?"
+    # Verificar si hay una respuesta personalizada
+    for clave, respuesta in respuestas_personalizadas.items():
+        if clave in user_input.lower():
+            return respuesta
+
+    # Consultar en Wikipedia para temas relevantes
+    temas_wikipedia = [
+        "fundación del Jardín Botánico Nacional",
+        "ubicación del Jardín Botánico Nacional",
+        "especies del Jardín Botánico Nacional",
+        "flora nativa de Chile",
+        "conservación en Chile",
+        "bosque esclerófilo",
+        "flora mediterránea",
+        "plantas endémicas de Chile",
+        "biodiversidad en peligro de extinción",
+        "restauración ecológica",
+        "horticultura",
+        "botánica aplicada",
+        "programas de conservación en Chile"
+    ]
+
+    for tema in temas_wikipedia:
+        if tema in user_input.lower():
+            return buscar_wikipedia(tema)
+
+    # Si la entrada no coincide con nada
+    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",