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()