Update app.py

app.py

@@ -10,6 +10,7 @@ from google.adk.models.lite_llm import LiteLlm
 import uuid
 import asyncio
 
+
 #download_files_from_drive_tool = FunctionTool(func=download_files_from_drive)
 # --- Definición de Mod
 
@@ -130,19 +131,13 @@ def validar_plan_de_afiliado(plan: str, practica: str) -> Dict[str, Any]:
         "comment": f"Validación correcta. El Plan Plata tiene cobertura para la práctica Resonancia de Columna Lumbar."
     }
 
-APP_NAME = "predoc_app"
-
-USER_ID = str(uuid.uuid4())
-SESSION_ID = str(uuid.uuid4())
-
 
+APP_NAME = "predoc_app"
 
+# Definición del agente
 root_agent = LlmAgent(
-    #model="gemini-2.5-flash",
     model=LiteLlm(model="openai/gpt-4.1"),
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.0,
-    ),
+    generate_content_config=types.GenerateContentConfig(temperature=0.0),
     name="PreDoc",
     instruction=instruction,
     description="Asistente médico para renovación de recetas",
@@ -151,99 +146,76 @@ root_agent = LlmAgent(
 
 session_service = InMemorySessionService()
 
-async def initialize():
-    global session
-    session = await session_service.create_session(
-        app_name=APP_NAME, 
-        user_id=USER_ID, 
-        session_id=SESSION_ID
-    )
-
-# Ejecutar la función async (esto va al final del archivo principal o en tu entrypoint)
-asyncio.run(initialize())
-runner = Runner(agent=root_agent, app_name=APP_NAME, session_service=session_service)
-
-
-
-
-def call_agent_text(query):
-   
-   content = types.Content(role='user', parts=[types.Part(text=query)])
-   events = runner.run(user_id=USER_ID, session_id=SESSION_ID, new_message=content)
-
-   for event in events:
-       print(event)
-       if event.is_final_response():
-           final_response = event.content.parts[0].text
-           print("Agent Response: ", final_response)
-   return final_response
-
-def call_agent_image(query):
-   images=[]
-   for q in query:
-    with open(q, 'rb') as f:
-        image_bytes = f.read()
-    images.append(types.Part.from_bytes(
-          data=image_bytes,
-          mime_type='image/jpeg',
-        ))
-
-
-   content = types.Content(role='user', parts=images)
-   events = runner.run(user_id=USER_ID, session_id=SESSION_ID, new_message=content)
-
-   for event in events:
-       print(event)
-       if event.is_final_response():
-           final_response = event.content.parts[0].text
-           print("Agent Response: ", final_response)
-   return final_response
-
-def call_agent_both(image,text):
-   with open(image[0], 'rb') as f:
-      image_bytes = f.read()
-
-   content = types.Content(
-    role='user',
-    parts=[
-        types.Part.from_bytes(data=image_bytes, mime_type='image/jpeg'),
-        types.Part(text=text)
-    ]
-)
-
-   print("CONTENT",content)
-   events = runner.run(user_id=USER_ID, session_id=SESSION_ID, new_message=content)
-
-   for event in events:
-       print(event)
-       if event.is_final_response():
-           final_response = event.content.parts[0].text
-           print("Agent Response: ", final_response)
-   return final_response
-
-def respond(message, history):
-    num_files = len(message["files"])
-    print("FILE", message)
+# Función principal con state
+def respond(message, history, state):
+    # Inicializamos state si es la primera vez
+    if state is None:
+        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}")
+
+        # Crear sesión ADK de forma asíncrona
+        async def create_session():
+            await session_service.create_session(
+                app_name=APP_NAME,
+                user_id=user_id,
+                session_id=session_id
+            )
+        asyncio.run(create_session())
+
+    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):
+        content = types.Content(role='user', parts=[types.Part(text=query)])
+        events = runner.run(user_id=user_id, session_id=session_id, new_message=content)
+        for event in events:
+            if event.is_final_response():
+                return event.content.parts[0].text
+        return "No se obtuvo respuesta."
+
+    def call_agent_image(query):
+        images = []
+        for q in query:
+            with open(q, 'rb') as f:
+                image_bytes = f.read()
+            images.append(types.Part.from_bytes(data=image_bytes, mime_type='image/jpeg'))
+
+        content = types.Content(role='user', parts=images)
+        events = runner.run(user_id=user_id, session_id=session_id, new_message=content)
+        for event in events:
+            if event.is_final_response():
+                return event.content.parts[0].text
+        return "No se obtuvo respuesta."
+
+    def call_agent_both(image, text):
+        with open(image[0], 'rb') as f:
+            image_bytes = f.read()
+        content = types.Content(
+            role='user',
+            parts=[
+                types.Part.from_bytes(data=image_bytes, mime_type='image/jpeg'),
+                types.Part(text=text)
+            ]
+        )
+        events = runner.run(user_id=user_id, session_id=session_id, new_message=content)
+        for event in events:
+            if event.is_final_response():
+                return event.content.parts[0].text
+        return "No se obtuvo respuesta."
+
+    # Dispatcher según tipo de input
     if message['text'] != '' and len(message['files']) > 0:
-      #ambas
-      print("AMBAS")
-      res = call_agent_both(message['files'],message['text'])
-      return res
-      
+        return call_agent_both(message['files'], message['text']), state
     elif message['text'] == '' and len(message['files']) > 0:
-      #imagenes
-      res = call_agent_image(message['files'])
-      return res
-
-    elif message['text'] !='' and len(message['files']) == 0:
-      #texto
-      res = call_agent_text(message['text'])
-      return res
+        return call_agent_image(message['files']), state
+    elif message['text'] != '' and len(message['files']) == 0:
+        return call_agent_text(message['text']), state
     else:
-      #vacio
-      return "Escribe algo para que pueda contestarte."
-    
-
-demo = gr.ChatInterface(fn=respond, title="Agente Revisor", multimodal=True)
+        return "Escribe algo para que pueda contestarte.", state
 
-demo.launch(debug=True)
+# Gradio interface con estado
+demo = gr.ChatInterface(fn=respond, title="Agente Revisor", multimodal=True, state=gr.State())
+demo.launch(debug=True)