Update other_function.py

other_function.py

@@ -31,17 +31,17 @@ def generate_response(query,chat_history):
             response += chunk.choices[0].delta.content
     return response.replace("###", '').replace('\nUser:','')
 class ConversationBufferMemory:
-    def __init__(self, memory_key="chat_history"):
-        self.memory_key = memory_key
-        self.buffer = []
+    def __init__(self, max_size=6):
+        self.memory = []
+        self.max_size = max_size
 
     def add_to_memory(self, interaction):
-        self.buffer.append(interaction)
+        self.memory.append(interaction)
+        if len(self.memory) > self.max_size:
+            self.memory.pop(0)  # Remove the oldest interaction
 
     def get_memory(self):
-        return "\n".join([f"Human: {entry['user']}\nAssistant: {entry['assistant']}" for entry in self.buffer])
-
-
+        return self.memory
 def predict_pest(filepath):
     CLIENT = InferenceHTTPClient(
         api_url="https://detect.roboflow.com",