sd

app.py

@@ -53,9 +53,18 @@ graph_builder.add_edge("agent", END)
 graph = graph_builder.compile()
 
 # 6. Función para Gradio
-def chat_with_agent(message):
-    result = graph.invoke({"input": message})
-    return result["output"]
+def agent_step(state: AgentState) -> AgentState:
+    user_input = state.get("input")
+    if not user_input:
+        return {"input": "", "output": "No se recibió entrada."}
+    travel_prompt = (
+        "Eres un agente de viajes profesional y experimentado. "
+        "Asesora, recomienda y planifica itinerarios, destinos y actividades de viaje según las preferencias del usuario. "
+        f"Usuario: {user_input}"
+    )
+    response = qwen.invoke(travel_prompt)
+    return {"input": user_input, "output": response}
+
 
 # 7. Interfaz Gradio adaptada para un agente de viajes
 iface = gr.Interface(