Refactor app.py for chat interface and update agent logic 2

agent.py

@@ -41,7 +41,7 @@ class BasicAgent:
         self.model = HfApiModel(
             model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud",
             provider="together",
-            use_auth_token=True,
+            use_auth_token=True
         )
 
         # Initialize agent once
@@ -49,7 +49,7 @@ class BasicAgent:
             tools=[tls.search_tool, tls.calculate_cargo_travel_time],
             model=self.model,  # <-- use self.model instead of re-instantiating
             additional_authorized_imports=["pandas"],
-            max_steps=20,
+            max_steps=20
         )
 
         self.chat_history = []  # Optional: for memory support later

app.py

@@ -146,8 +146,8 @@ def test_init_agent_for_chat(text_input, history):
 # --- Agent Logic for Chat Interface ---
 def handle_chat(message, history):
     try:
-        agent = BasicAgent()
-        response = agent(message)
+        basicAgent = agent.BasicAgent()
+        response = basicAgent(message)
         return response
     except Exception as e:
         return f"[ERROR] Agent failed: {str(e)}"