app.py

import gradio as gr
import requests
import os


# Marcadores de posición que serán reemplazados por TemplateGen
assistant_id = "asst_vsm9RJaiFiWAcMFmwKF2u6Jd"
project_id = "74f3g1h8ccc53ddfecj733g8"
title = "Innovación y Desarrollo"
description = "Explora las últimas innovaciones y estrategias de desarrollo para impulsar tu negocio hacia el futuro."
theme = "dark"
examples = ["¿Cómo podemos aplicar la inteligencia artificial en nuestros procesos?", "¿Qué tecnologías emergentes deberíamos considerar?", "¿Cuál es la mejor manera de fomentar la innovación en nuestro equipo?", "¿Cómo medimos el éxito de nuestros proyectos de innovación?", "¿Qué recursos están disponibles para equipos de innovación?"]
# Inicializa la variable global
thread_id = None


def process_body(message: str, thr_id):
    if thr_id == None:
        return {
            "message": message,
            "model": "gpt-3.5-turbo-0613"
        }
    return {
        "message": message,
        "model": "gpt-3.5-turbo-0613",
        "threadId": thr_id
    }

# Función para generar la respuesta usando tu API
def generate_response_api(message, thread_id, assistant_id):
    assistant_id = "asst_vsm9RJaiFiWAcMFmwKF2u6Jd"  # Variable dinámica
    project_id = "74f3g1h8ccc53ddfecj733g8"  # Variable dinámica
    base_url = "https://enjambre-api-mue2676eoa-uc.a.run.app"
    url = f"{base_url}/v1/agents/{assistant_id}/completion"
    response = requests.post(url, json=process_body(message, thread_id))
    response_json = response.json()
    return response_json["message"], response_json["threadId"]

# Aquí continúa tu lógica de Gradio...

def chat_function(message, history):
    global thread_id
    assistant_id   # ID de tu asistente
    response, new_thread_id = generate_response_api(message, thread_id, assistant_id)
    if new_thread_id != None:
        thread_id = new_thread_id  # Actualiza el thread_id global
    return response

def set_thread_id(thr_id:str):
    global thread_id
    if thr_id == '':
        return 
    thread_id = thr_id

# Configuración de ChatInterface usando valores dinámicos
demo=gr.ChatInterface(
        fn=chat_function,
        title=title,
        description=description,
        theme=theme,        
        examples=examples,
        retry_btn=None,
        undo_btn=None,    
        submit_btn="Enviar",
        stop_btn="Detener"
    ).queue()


if __name__ == "__main__":
    demo.launch(share=True)