Update app.py

app.py

@@ -34,31 +34,11 @@ with st.container():
 # Safe way to access secrets
 GOOGLE_API_KEY = "AIzaSyBm0GOvYox4OyRG1WFOK7FT5fnNCHfubns"
 
+embedding = GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=api_key)
+vectorstore = Chroma(persist_directory=chroma_db, embedding_function=embedding)
+
 
-# Initialize Chroma DB
-@st.cache_resource
-def initialize_vectorstore(api_key):
-    embedding = GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=api_key)
-    zip_path = "src/chroma_db1.zip"
-    extract_dir = "src/chroma_db2"
-    if os.path.exists(zip_path):
-        try:
-            with zipfile.ZipFile(zip_path, 'r') as zip_ref:
-                zip_ref.extractall(extract_dir)
-            vectorstore = Chroma(persist_directory=extract_dir, embedding_function=embedding)
-            if vectorstore._collection.count() > 0:
-                return vectorstore
-            else:
-                st.error("Chroma DB is empty after extraction.")
-        except Exception as e:
-            st.error(f"Failed to load Chroma DB: {str(e)}")
-    else:
-        st.error(f"`chroma_db1.zip` not found at {zip_path}")
-    return None
 
-retriever = None
-vectorstore = None
-llm, parser = None, None
 
 if GOOGLE_API_KEY:
     vectorstore = initialize_vectorstore(GOOGLE_API_KEY)