Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
from datetime import datetime | |
import pytz | |
import requests | |
# Crear el chatbot usando el modelo de transformers | |
chatbot = pipeline("text-generation", model="ajustefino_model") | |
# 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¤t_weather=true" | |
response = requests.get(url) | |
if response.status_code == 200: | |
data = response.json() | |
temp = data["current_weather"]["temperature"] | |
windspeed = data["current_weather"]["windspeed"] | |
return f"El clima es de {temp}°C con una velocidad de viento de {windspeed} km/h." | |
else: | |
return "Lo siento, no pude obtener el clima en este momento." | |
# Función para obtener la respuesta del bot | |
def get_response(user_input): | |
# 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 "El Jardín está abierto de [9:00 hasta las 17:30]. ¡Espero que vengas!" | |
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 :( " | |
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: | |
# Respuesta por defecto usando el modelo GPT-2 | |
response = chatbot(user_input, max_length=50) | |
return response[0]['generated_text'] | |
# 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!") | |
# Lanzar la interfaz | |
iface.launch() | |