Update app.py

app.py

@@ -6,7 +6,8 @@ from chromadb.config import Settings
 import torch
 from sentence_transformers import CrossEncoder
 import gradio as gr
-from datetime import datetime  # Import datetime to get current time
+from datetime import datetime 
+from huggingface_hub import hf_hub_download
 
 # Load environment variables and initialize clients
 load_dotenv()
@@ -14,6 +15,23 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
+def load_chroma_db():
+    # Load from Hugging Face dataset
+    print("Using ChromaDB from Hugging Face dataset...")
+    # Download the database file from Hugging Face
+    db_file_path = hf_hub_download(repo_id="Mr-Geo/chroma_db", filename="chroma.sqlite3")
+    # Initialize your ChromaDB with the downloaded file
+    db = chromadb.Client(path=db_file_path)
+    return db
+
+# Check if running locally
+if os.path.exists("./chroma_db/chroma.sqlite3"):
+    print("Using local ChromaDB setup...")
+    db = chromadb.Client(path="./chroma_db/chroma.sqlite3")  # Adjust the path as needed
+else:
+    # Load from Hugging Face dataset
+    db = load_chroma_db()
+
 def initialize_system():
     chroma_client = chromadb.PersistentClient(
         path="./chroma_db",