Update interfaces/interface.py

interfaces/interface.py

@@ -5,35 +5,28 @@ from agents.agents_nodes import agent_node, format_output, tool_node
 from utils.state_utils import AgentState
 
 def create_interface():
-    
+
     graph = StateGraph(AgentState)
     graph.add_node("agent", agent_node)
     graph.add_node("tool", tool_node)
     graph.add_node("format", format_output)
     
-    # Conditional routing
-    graph.add_conditional_edges(
-        "agent",
-        lambda state: "tool" if state["messages"][-1].tool_calls else "format",
-        {"tool": "tool", "format": "format"}
-    )
+    graph.set_entry_point("agent")
+    graph.add_edge("agent", "tool")
     graph.add_edge("tool", "format")
     graph.add_edge("format", END)
-
-
     
     app = graph.compile()
 
-
     def process_query(query: str) -> dict:
         try:
             inputs = {"messages": [HumanMessage(content=query)]}
             result = app.invoke(inputs)  
-            return result.get("output") or {"error": "No result generated"}
-            
+            return messages_to_dict(result['messages'])[2]['data']['content']
         except Exception as e:
             return {"error": f"Execution error: {str(e)}"}
 
+
     with gr.Blocks(title="Time Value of Money Calculator") as interface:
         gr.Markdown("##  Time Value of Money Calculator")
         gr.Markdown("Enter natural language queries about present/future value calculations")
@@ -52,4 +45,4 @@ def create_interface():
             inputs=input_text,      
             outputs=output_json     
         )
-    return interface
+    return interface