Update app.py

app.py

@@ -148,12 +148,12 @@ root_agent = LlmAgent(
 session_service = InMemorySessionService()
 
 # Esta función se conecta a ChatInterface
-def wrapped_respond(message, history, state=None):
-    print("STATE",state)
-    if state is None:
+def respond(message, history):
+    # Detectar si es inicio de conversación (history vacío o con 0 mensajes)
+    print("HISTORY",history)
+    if history is None or len(history) == 0:
         user_id = str(uuid.uuid4())
         session_id = str(uuid.uuid4())
-        state = {"user_id": user_id, "session_id": session_id}
         print(f"🔄 Nueva sesión: {session_id}")
 
         async def create_session():
@@ -163,9 +163,16 @@ def wrapped_respond(message, history, state=None):
                 session_id=session_id
             )
         asyncio.run(create_session())
+        # Guardar en algún lado user_id y session_id para usar en las próximas llamadas
+        # Por simplicidad acá lo guardamos en variables globales
+        global CURRENT_USER_ID, CURRENT_SESSION_ID
+        CURRENT_USER_ID = user_id
+        CURRENT_SESSION_ID = session_id
+    else:
+        # Usar IDs existentes
+        user_id = CURRENT_USER_ID
+        session_id = CURRENT_SESSION_ID
 
-    user_id = state["user_id"]
-    session_id = state["session_id"]
     runner = Runner(agent=root_agent, app_name=APP_NAME, session_service=session_service)
 
     def call_agent_text(query):
@@ -217,6 +224,6 @@ def wrapped_respond(message, history, state=None):
         return "Escribe algo para que pueda contestarte."
 
 # Inicializamos demo sin el argumento state
-demo = gr.ChatInterface(fn=wrapped_respond, title="Agente Revisor", multimodal=True)
+demo = gr.ChatInterface(fn=respond, title="Agente Revisor", multimodal=True)
 
 demo.launch(debug=True)