Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import zipfile
 from dotenv import load_dotenv
 from groq import Groq
 import chromadb
@@ -19,23 +20,27 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 hf_token = os.getenv("HF_TOKEN")
 
 def load_chroma_db():
-    # Load from Hugging Face dataset
     print("Using ChromaDB from Hugging Face dataset...")
-    # Download the database file from Hugging Face with authentication
-    db_file_path = hf_hub_download(
+    
+    # Download the zipped database from Hugging Face
+    zip_path = hf_hub_download(
         repo_id="Mr-Geo/chroma_db",
-        filename="chroma.sqlite3",
+        filename="chroma_db.zip",
         repo_type="dataset",
         use_auth_token=hf_token
     )
-    print(f"Downloaded database file to: {db_file_path}")
+    print(f"Downloaded database zip to: {zip_path}")
+    
+    # Extract to a temporary directory
+    extract_dir = "/tmp/chroma_db"
+    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+        zip_ref.extractall(extract_dir)
     
-    # Get the directory path of the downloaded file
-    db_directory = os.path.dirname(db_file_path)
+    print(f"Extracted database to: {extract_dir}")
     
-    # Initialize ChromaDB with the persistent client and point it to the downloaded database
+    # Initialize ChromaDB with the extracted database
     db = chromadb.PersistentClient(
-        path=db_directory,  # Use the directory where the file was downloaded
+        path=extract_dir,
         settings=Settings(
             anonymized_telemetry=False,
             allow_reset=True,