Update app.py

app.py

@@ -156,7 +156,15 @@ api_key = os.getenv("API_KEY")
 genai.configure(api_key=api_key)
 gemini_model = genai.GenerativeModel(model_name="gemini-2.0-flash")
 
-
+@st.cache_resource
+def load_faiss_index():
+    if not os.path.exists("faiss_index.idx") or not os.path.exists("doc_texts.npy"):
+        return None, None
+    index = faiss.read_index("faiss_index.idx")
+    doc_texts = np.load("doc_texts.npy", allow_pickle=True)
+    return index, doc_texts
+
+faiss_index, doc_texts = load_faiss_index()
 
 # Query Legal Documents
 def query_legal_documents(query: str, top_k=3):