chat / app.py
JuanMa360's picture
bugfix: function
ab558ef
raw
history blame
1.19 kB
import time
import gradio as gr
import requests
import json
# Funci贸n que hace la solicitud a la API y muestra la respuesta progresivamente
def slow_api_response(message, history):
url = "https://data-monopolio.dev.dd360.mx/ai-assistant/v1"
# Crea el payload con el mensaje proporcionado
payload = json.dumps({
"query": message,
"sessionId": "12345",
"numMessages": "1",
"userId": "user-01",
"property_id": "c0ced2220b87fc23762facf617157a4f"
})
headers = {
'Content-Type': 'application/json'
}
try:
# Realiza la solicitud POST a la API
response = requests.request("POST", url, headers=headers, data=payload)
api_response = response.text
# Simula una respuesta progresiva
for i in range(len(api_response)):
time.sleep(0.05) # Simula un retraso
yield api_response[:i + 1] # Devuelve la respuesta progresiva
except requests.RequestException as e:
yield f"Error: {str(e)}"
# Interfaz Gradio usando la funci贸n slow_api_response
demo = gr.ChatInterface(fn=slow_api_response)
if __name__ == "__main__":
demo.launch()