Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import os
+import zipfile
 
 from langchain_google_genai import GoogleGenerativeAIEmbeddings, ChatGoogleGenerativeAI
 from langchain_community.vectorstores import FAISS
@@ -9,10 +10,28 @@ from langchain_core.output_parsers import StrOutputParser
 # --- Configuration ---
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "")
 
-# Path to pre-built FAISS index in the repo
+# Paths
+FAISS_ZIP_PATH = "faiss_index.zip"
 FAISS_INDEX_PATH = "faiss_index"
 
 
+def extract_faiss_index():
+    """Extract FAISS index from zip file if needed."""
+    index_file = os.path.join(FAISS_INDEX_PATH, "index.faiss")
+    
+    # Already extracted
+    if os.path.exists(index_file):
+        return True
+    
+    # Extract from zip
+    if os.path.exists(FAISS_ZIP_PATH):
+        with zipfile.ZipFile(FAISS_ZIP_PATH, 'r') as zip_ref:
+            zip_ref.extractall(".")
+        return True
+    
+    return False
+
+
 def get_conversational_chain(api_key):
     """Create the QA chain with strict context-only answering."""
     prompt_template = """
@@ -57,6 +76,9 @@ def user_input(user_question, vector_store, api_key):
 @st.cache_resource
 def load_vector_store(_api_key):
     """Load pre-built FAISS vector store."""
+    # Extract zip if needed
+    extract_faiss_index()
+    
     embeddings = GoogleGenerativeAIEmbeddings(
         model="models/embedding-001", 
         google_api_key=_api_key
@@ -188,11 +210,11 @@ def main():
         st.info("Please add GOOGLE_API_KEY to your Hugging Face Space secrets.")
         st.stop()
 
-    # Check if FAISS index exists
+    # Check if FAISS index or zip exists
     index_file = os.path.join(FAISS_INDEX_PATH, "index.faiss")
-    if not os.path.exists(index_file):
-        st.error(f"FAISS index not found at: {FAISS_INDEX_PATH}/")
-        st.info("Please upload index.faiss and index.pkl to the faiss_index folder.")
+    if not os.path.exists(index_file) and not os.path.exists(FAISS_ZIP_PATH):
+        st.error("FAISS index not found!")
+        st.info("Please upload faiss_index.zip or the faiss_index folder to your Space.")
         st.stop()
 
     # Load vector store (cached)