Update app/rag_setup.py

app/rag_setup.py

@@ -95,20 +95,24 @@ class TFIDFEmbeddingFunction:
 
 # ChromaDB setup - using in-memory storage for simplicity and compatibility with Hugging Face.
 logger.info("Initializing ChromaDB with in-memory storage.")
-try:
-    client = chromadb.Client()
-    embedding_function = TFIDFEmbeddingFunction()
-    
-    collection = client.get_or_create_collection(
-        name="context_aware_collection",
+client = chromadb.Client()
+embedding_function = TFIDFEmbeddingFunction()
+COLLECTION_NAME = "context_aware_collection"
+
+def clear_and_recreate_collection():
+    """Deletes and re-creates the ChromaDB collection to ensure it's empty."""
+    logger.info(f"Deleting and re-creating collection '{COLLECTION_NAME}'...")
+    try:
+        client.delete_collection(name=COLLECTION_NAME)
+    except Exception as e:
+        logger.warning(f"Failed to delete collection (it may not exist): {e}")
+    return client.get_or_create_collection(
+        name=COLLECTION_NAME,
         embedding_function=embedding_function
     )
-    
-    logger.info("ChromaDB collection initialized successfully.")
-    
-except Exception as e:
-    logger.error(f"Error initializing ChromaDB: {e}")
-    raise RuntimeError(f"Failed to initialize ChromaDB: {e}")
+
+collection = clear_and_recreate_collection()
+logger.info("ChromaDB collection initialized successfully.")
 
 class OpenRouterLLM:
     """A wrapper for interacting with the OpenRouter LLM API."""