Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from langchain_core.messages import HumanMessage
 from agents import create_agent_flow
+from langchain_core.messages import HumanMessage
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -22,7 +22,7 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
       
         question = [HumanMessage(content=question)]
-        question_ask = self.graph.invoke({"messages": question})
+        question_ask = self.agent.invoke({"messages": question})
         response = question_ask['messages'][-1].content
         print(f"Agent returning fixed answer: {response}")
         return response[8:]