Error Fix

backend/main.py

@@ -63,13 +63,19 @@ class DocumentAssistant:
     
     def reset_database(self):
         """Reset the ChromaDB database"""
-        # This is a placeholder - you'll need to implement this in your DocumentAssistant class
-        # Typically, this would involve deleting all collections or recreating the database
         try:
-            # Example implementation - adjust based on your actual DocumentAssistant implementation
-            self.vector_db = None
-            self.initialize_vectorstore()
-            return True
+            # Reset the vector database
+            if hasattr(self, 'vector_db') and self.vector_db is not None:
+                success = self.vector_db.reset_collection()
+                
+                # Also clear the SimpleDB
+                if hasattr(self, 'db') and self.db is not None:
+                    self.db.clear_all()
+                
+                return success
+            else:
+                print("Vector database not initialized")
+                return False
         except Exception as e:
             print(f"Error resetting database: {str(e)}")
             return False 

backend/vector_db.py

@@ -55,6 +55,18 @@ class ChromaVectorDB:
         
         if results and results['ids']:
             self.collection.delete(ids=results['ids'])
+    
+    def reset_collection(self):
+        """Reset the collection by deleting and recreating it"""
+        try:
+            # Delete the collection
+            self.client.delete_collection("documents")
+            # Recreate the collection
+            self.collection = self.client.get_or_create_collection("documents")
+            return True
+        except Exception as e:
+            print(f"Error resetting collection: {str(e)}")
+            return False
 
 # Create sample data
 data = pd.DataFrame({