Update src/history.py

src/history.py

@@ -3,6 +3,30 @@ import os
 from typing import List, Dict
 from datetime import datetime
 
+def save_conversation(user_id: str, messages: List[Dict], file_path: str = "data/chat_history.json"):
+    """Save conversation history to JSON file"""
+    try:
+        history = load_conversation(file_path)
+        history[user_id] = messages[-10:]  # Keep last 10 messages
+        
+        with open(file_path, 'w') as f:
+            json.dump(history, f, indent=2)
+            
+    except Exception as e:
+        print(f"Error saving conversation: {e}")
+
+def load_conversation(file_path: str = "data/chat_history.json") -> Dict[str, List[Dict]]:
+    """Load conversation history from JSON file"""
+    try:
+        if not os.path.exists(file_path):
+            return {}
+            
+        with open(file_path, 'r') as f:
+            return json.load(f)
+    except Exception as e:
+        print(f"Error loading conversation: {e}")
+        return {}
+
 class ChatHistory:
     def __init__(self, storage_path: str, max_entries: int = 10):
         self.path = storage_path
@@ -12,19 +36,23 @@ class ChatHistory:
     def _initialize_storage(self):
         if not os.path.exists(self.path):
             with open(self.path, 'w') as f:
-                json.dump([], f)
+                json.dump({}, f)
     
-    def add_entry(self, user_message: str, bot_response: str):
-        with open(self.path, 'r+') as f:
-            history = json.load(f)
-            history.append({
-                "timestamp": datetime.utcnow().isoformat(),
-                "user": user_message,
-                "bot": bot_response
-            })
-            f.seek(0)
-            json.dump(history[-self.max_entries:], f, indent=2)
+    def add_entry(self, user_id: str, user_message: str, bot_response: str):
+        history = load_conversation(self.path)
+        if user_id not in history:
+            history[user_id] = []
+            
+        history[user_id].append({
+            "timestamp": datetime.utcnow().isoformat(),
+            "user": user_message,
+            "bot": bot_response
+        })
+        
+        # Keep only the last max_entries messages
+        history[user_id] = history[user_id][-self.max_entries:]
+        
+        save_conversation(user_id, history[user_id], self.path)
     
-    def get_history(self) -> List[Dict]:
-        with open(self.path, 'r') as f:
-            return json.load(f)
+    def get_history(self, user_id: str) -> List[Dict]:
+        return load_conversation(self.path).get(user_id, [])