Made the agent conversational without memory

agent.py

@@ -1,7 +1,7 @@
 import os
 from dotenv import load_dotenv
 from huggingface_hub import InferenceClient
-from smolagents import CodeAgent, Model, ChatMessage, SimpleMemory
+from smolagents import CodeAgent, Model, ChatMessage
 import tools.tools as tls  # Your tool definitions
 
 load_dotenv()
@@ -61,18 +61,23 @@ class BasicAgent:
     def __init__(self):
         print("✅ BasicAgent initialized with Hugging Face chat model.")
         self.model = HuggingFaceChatModel()
-        self.memory = SimpleMemory(k=5)  # k = how many turns to remember
-
+        
         self.agent = CodeAgent(
             tools=[tls.search_tool, tls.calculate_cargo_travel_time],
             model=self.model,
-            memory=self.memory,
             additional_authorized_imports=["pandas"],
             max_steps=20,
         )
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, messages) -> str:
+    # Extract last user message from the chat history
+        if isinstance(messages, list):
+            question = messages[-1]["content"]  # Gradio gives list of dicts
+        else:
+            question = messages  # Fallback if it's just a string
+
         print(f"📥 Received question: {question[:60]}...")
         response = self.agent.run(question)
         print(f"📤 Response generated: {response[:60]}...")
         return response
+