SoyAlekZen's picture
Upload folder using huggingface_hub
4fa2f73 verified
Raw
History Blame Contribute Delete
1.87 kB
import gradio as gr
import requests
# Función para generar la respuesta usando tu API
def generate_response_api(message, thread_id, assistant_id):
base_url = "https://enjambre-api-mue2676eoa-uc.a.run.app"
url = f"{base_url}/v1/agents/{assistant_id}/completion"
response = requests.post(url, json={
"message": message,
"model": "gpt-3.5-turbo-0613",
"threadId": thread_id
})
response_json = response.json()
return response_json["message"], response_json["threadId"]
# Función para manejar el chat, ahora acepta dos argumentos
def chat_function(message, history):
global thread_id
assistant_id = "asst_vsm9RJaiFiWAcMFmwKF2u6Jd" # ID de tu asistente
response, new_thread_id = generate_response_api(message, thread_id, assistant_id)
thread_id = new_thread_id # Actualiza el thread_id global
return response
# Inicializa la variable global
thread_id = None
# Configuración de ChatInterface
demo = gr.ChatInterface(
fn=chat_function,
title="Proyectos MAC",
description="Soy un asistente virtual que conoce el proyecto de Magaña Arevalo Contadores a la perfección y te puedo ayudar con tus dudas o estrategias.",
theme="soft",
examples=[
"¿Cuál es el cronograma detallado de implementación para la transformación digital de MAC?",
"¿Qué capacitaciones están incluidas en el proyecto para el personal de MAC?",
"¿Cómo se integrarán las plataformas bancarias y el CRM en nuestra infraestructura actual?",
"¿Cuáles son los KPIs clave que los Dashboards de BI nos ayudarán a monitorear?",
"¿Qué medidas de soporte y mantenimiento ofrece Geek Studio tras la finalización del proyecto?"
],
retry_btn=None,
undo_btn=None,
submit_btn="Enviar",
stop_btn="Detener"
).queue()
if __name__ == "__main__":
demo.launch(share=True)