Update app.py

app.py

@@ -1,17 +1,17 @@
 import gradio as gr
-
-from agents.process_aware_agent import build_agent  # adapt to your actual API
-
-agent = create_agent_graph()
+from agents.process_aware_agent import run_agent
 
 def chat_fn(message, history):
-    # adapt this to however your agent is invoked
-    result = agent.invoke({"input": message})
-    if isinstance(result, dict):
-        return result.get("output") or result.get("answer") or str(result)
-    return str(result)
+    try:
+        return run_agent(message)
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-demo = gr.ChatInterface(fn=chat_fn, title="Process-Aware LangGraph Agent")
+demo = gr.ChatInterface(
+    fn=chat_fn,
+    title="Process-Aware LangGraph Agent",
+    description="Chat with a process-aware LangGraph agent."
+)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)