Update app.py

app.py

@@ -31,16 +31,16 @@ def load_chroma_db():
     )
     print(f"Downloaded database zip to: {zip_path}")
     
-    # Extract to a temporary directory
-    extract_dir = "/tmp/chroma_db"
+    # Extract to a temporary directory that matches local structure
+    extract_dir = "/tmp"  # Changed to /tmp so we can create /tmp/chroma_db
     with zipfile.ZipFile(zip_path, 'r') as zip_ref:
         zip_ref.extractall(extract_dir)
     
     print(f"Extracted database to: {extract_dir}")
     
-    # Initialize ChromaDB with the extracted database
+    # Initialize ChromaDB with the extracted database - matching local path structure
     db = chromadb.PersistentClient(
-        path=os.path.join(extract_dir, "chroma_db"),  # Add chroma_db to the path
+        path="/tmp/chroma_db",  # Exactly matching the local ./chroma_db structure
         settings=Settings(
             anonymized_telemetry=False,
             allow_reset=True,
@@ -48,8 +48,12 @@ def load_chroma_db():
         )
     )
     
-    # Debug: Print collections
-    print("Available collections:", db.list_collections())
+    # Debug: Print collections and document count
+    collections = db.list_collections()
+    print("Available collections:", collections)
+    if collections:
+        collection = db.get_collection(name="website_content")
+        print(f"Found {collection.count()} documents in collection")
     
     return db