Error Fix

app.py

@@ -596,7 +596,11 @@ def flush_databases():
         if success:
             result.append("✅ ChromaDB cleared successfully")
         else:
-            result.append("❌ ChromaDB reset failed - database may not be initialized")
+            # Even if reset fails, we can still reinitialize the document assistant
+            # This is a workaround that creates a fresh instance
+            global document_assistant
+            document_assistant = DocumentAssistant()
+            result.append("⚠️ ChromaDB reset partially completed - created new instance")
     except Exception as e:
         result.append(f"❌ Error clearing ChromaDB: {str(e)}")
     

backend/main.py

@@ -66,26 +66,18 @@ class DocumentAssistant:
         try:
             # Reset the vector database
             if hasattr(self, 'vector_db') and self.vector_db is not None:
-                # Get the client from vector_db
-                client = self.vector_db.client
-                
-                # Delete the collection if it exists
-                try:
-                    client.delete_collection("documents")
-                    print("Collection 'documents' deleted successfully")
-                except Exception as e:
-                    print(f"Error deleting collection: {str(e)}")
-                
-                # Recreate the collection
-                self.vector_db.collection = client.get_or_create_collection("documents")
-                print("Collection 'documents' recreated successfully")
+                # Try to reset the collection
+                success = self.vector_db.reset_collection()
                 
                 # Also clear the SimpleDB
                 if hasattr(self, 'db') and self.db is not None:
-                    self.db.clear_all()
-                    print("SimpleDB cleared successfully")
+                    try:
+                        self.db.clear_all()
+                        print("SimpleDB cleared successfully")
+                    except Exception as db_error:
+                        print(f"Error clearing SimpleDB: {str(db_error)}")
                 
-                return True
+                return success
             else:
                 print("Vector database not initialized")
                 return False

backend/vector_db.py

@@ -52,13 +52,30 @@ class ChromaVectorDB:
             self.collection.delete(ids=results['ids'])
     
     def reset_collection(self):
-        """Reset the collection by deleting and recreating it"""
+        """Reset the collection by clearing all documents"""
         try:
-            # Delete the collection
-            self.client.delete_collection("documents")
-            # Recreate the collection
-            self.collection = self.client.get_or_create_collection("documents")
-            return True
+            # Get all document IDs
+            try:
+                all_ids = self.collection.get()["ids"]
+                if all_ids:
+                    # Delete all documents
+                    self.collection.delete(ids=all_ids)
+                    print(f"Deleted {len(all_ids)} documents from collection")
+                else:
+                    print("Collection is already empty")
+                return True
+            except Exception as e:
+                print(f"Error getting or deleting documents: {str(e)}")
+                
+                # Try recreating the collection as a fallback
+                try:
+                    self.client.delete_collection("documents")
+                    self.collection = self.client.get_or_create_collection("documents")
+                    print("Collection recreated successfully")
+                    return True
+                except Exception as e2:
+                    print(f"Error recreating collection: {str(e2)}")
+                    return False
         except Exception as e:
             print(f"Error resetting collection: {str(e)}")
             return False