Update app.py

app.py

@@ -18,11 +18,17 @@ class SentenceTransformerEmbeddings:
 class LegalAssistant:
     def __init__(self):
         try:
-            # Verify ChromaDB content first
+            # Initialize and verify ChromaDB content
+            logger.info("Initializing LegalAssistant...")
+            from test_embeddings import test_chromadb_content, initialize_chromadb
+            
+            # Try to verify content, if fails, try to initialize
             if not test_chromadb_content():
-                raise ValueError("ChromaDB content verification failed")
+                logger.warning("ChromaDB verification failed, attempting to initialize...")
+                if not initialize_chromadb():
+                    raise ValueError("Failed to initialize ChromaDB")
             
-            # Initialize ChromaDB
+            # Initialize ChromaDB client
             base_path = os.path.dirname(os.path.abspath(__file__))
             chroma_path = os.path.join(base_path, 'chroma_db')
             
@@ -35,6 +41,8 @@ class LegalAssistant:
                 embedding_function=self.embedding_function
             )
             
+            logger.info(f"Collection loaded with {self.collection.count()} documents")
+            
             # Initialize Mistral AI client
             self.mistral_client = OpenAI(
                 api_key=os.environ.get("MISTRAL_API_KEY", "dfb2j1YDsa298GXTgZo3juSjZLGUCfwi"),