app.py

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, property_id="c0ced2220b87fc23762facf617157a4f"):
    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": "3", 
        "userId": "user-01",
        "property_id": property_id #"c0ced2220b87fc23762facf617157a4f"
    })
    
    headers = {
        'Content-Type': 'application/json'
    }

    try:
        
        response = requests.request("POST", url, headers=headers, data=payload)
        api_response = response.text
        api_response = api_response.encode().decode('unicode_escape')
        
        for i in range(len(api_response)):
            time.sleep(0.05) 
            yield api_response[:i + 1]  

    except requests.RequestException as e:
        yield f"Error: {str(e)}"

property_id_input = gr.Textbox(label="Property ID", placeholder="Ingresa el ID de la propiedad")

demo = gr.ChatInterface(fn=slow_api_response,
                        title="AI Assistant",  
                        additional_inputs=[property_id_input],
                        description="Esta app simula una conversación con un asistente virtual que conoce una propiedad.",
                        examples=["hola", "me gustaria mas informacion de la casa en venta", "acepta mascotas?"]
                        ).launch()