Update app.py

app.py

@@ -1,7 +1,7 @@
-
 import os
 import time
 import streamlit as st
+import subprocess
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain.prompts import PromptTemplate
@@ -15,7 +15,7 @@ st.set_page_config(page_title="BharatLAW", layout="centered")
 # Display logo
 col1, col2, col3 = st.columns([1, 30, 1])
 with col2:
-    st.image("https://github.com/Nike-one/BharatLAW/blob/master/images/banner.png?raw=true", use_container_width=True)
+    st.image("https://github.com/xadarsh/BharatLaw/blob/main/images/banner.png?raw=true", use_container_width=True)
 
 # Hide hamburger and default footer
 def hide_hamburger_menu():
@@ -41,15 +41,25 @@ def load_embeddings():
 
 embeddings = load_embeddings()
 
-# ✅ Add FAISS index check before loading
+# ✅ Ensure FAISS index is built if missing
 index_dir = "ipc_embed_db"
 index_faiss = os.path.join(index_dir, "index.faiss")
 index_pkl = os.path.join(index_dir, "index.pkl")
 
-if os.path.exists(index_faiss) and os.path.exists(index_pkl):
+if not (os.path.exists(index_faiss) and os.path.exists(index_pkl)):
+    st.warning("⚠️ FAISS index files missing. Attempting to create them using `Ingest.py`...")
+    try:
+        subprocess.run(["python", "Ingest.py"], check=True)
+        st.success("✅ FAISS index successfully created.")
+    except subprocess.CalledProcessError as e:
+        st.error(f"❌ Failed to create FAISS index.\n\n**Error**: {e}")
+        st.stop()
+
+# ✅ Load FAISS index
+try:
     db = FAISS.load_local(index_dir, embeddings, allow_dangerous_deserialization=True)
-else:
-    st.error("❌ FAISS index files not found! Please run the FAISS indexing script first.")
+except Exception as e:
+    st.error(f"❌ Could not load FAISS index even after attempting to build it.\n\n**Reason**: {str(e)}")
     st.stop()
 
 db_retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 3})