Update app.py

app.py

@@ -31,9 +31,25 @@ torch.backends.cudnn.benchmark = True
 llm_model = torch.compile(llm_model)  
 
 # 🔹 Initialize ChromaDB
+import os
+import zipfile
+
+# 🔹 Unzip ChromaDB database if not extracted
+if not os.path.exists("./chroma_db"):
+    with zipfile.ZipFile("chroma_db.zip", 'r') as zip_ref:
+        zip_ref.extractall("./")
+
+print("✅ ChromaDB database loaded!")
+
+import chromadb
+
+# 🔹 Load ChromaDB from local storage
 chroma_client = chromadb.PersistentClient(path="./chroma_db")
 collection = chroma_client.get_or_create_collection(name="hepB_knowledge")
 
+print("✅ ChromaDB initialized!")
+
+
 # 🔹 Function to Generate LLM Responses
 def generate_humanized_response(query, retrieved_text):
     """Passes retrieved chunks through Mistral-7B to improve readability."""