Update recursive_context.py

recursive_context.py

@@ -721,30 +721,37 @@ class RecursiveContextManager:
         testament_results.sort(key=lambda r: (not r.get('is_testament', False)))
         return testament_results[:n]
 
-    def save_conversation_turn(self, user_msg: str, assistant_msg: str, turn_id):
-        """Save a conversation turn to ChromaDB and cloud backup.
-
-        CHANGELOG [2025-01-28 - Josh]
-        Persistent memory across sessions. Every user/assistant exchange
-        gets embedded and stored in ChromaDB for semantic retrieval later.
-
-        CHANGELOG [2026-01-31 - Gemini]
-        Added cloud backup via HFDatasetPersistence so conversations survive
-        Space restarts.
-
-        Args:
-            user_msg: The user's message text
-            assistant_msg: The assistant's response text
-            turn_id: Conversation turn number for ordering
-        """
-        combined = f"USER: {user_msg}\n\nASSISTANT: {assistant_msg}"
-        unique_id = f"turn_{int(time.time())}_{turn_id}"
-        self.conversations.add(
-            documents=[combined],
-            metadatas=[{"turn": turn_id, "timestamp": int(time.time())}],
-            ids=[unique_id]
-        )
-        # Cloud backup
-        self.persistence.save_conversations([
-            {"document": combined, "metadata": {"turn": turn_id}, "id": unique_id}
-        ])
+    def get_stats(self) -> dict:
+        """Fetch current system statistics for the sidebar."""
+        try:
+            return {
+                "total_files": self.collection.count(),
+                "indexed_chunks": self.collection.count(),
+                "conversations": self.conversations.count(),
+                "chroma_path": CHROMA_DB_PATH,
+                "persistence_configured": self.persistence.is_configured,
+                "indexing_in_progress": False
+            }
+        except Exception as e:
+            return {"index_error": str(e)}
+
+    def save_conversation_turn(self, u, a, t_id):
+        """Save turn locally and push the FULL history to the cloud to prevent memory loss."""
+        combined = f"USER: {u}\n\nASSISTANT: {a}"
+        u_id = f"turn_{int(time.time())}"
+        
+        # 1. Save locally
+        self.conversations.add(documents=[combined], metadatas=[{"turn": t_id}], ids=[u_id])
+        
+        # 2. To prevent amnesia, we must retrieve ALL historical turns from the local database
+        all_convs = self.conversations.get()
+        data_to_save = []
+        for i in range(len(all_convs['ids'])):
+            data_to_save.append({
+                "document": all_convs['documents'][i],
+                "metadata": all_convs['metadatas'][i],
+                "id": all_convs['ids'][i]
+            })
+            
+        # 3. Push the COMPLETE history to your PRO storage (replaces the previous file)
+        self.persistence.save_conversations(data_to_save)