Update app.py

app.py

@@ -1,21 +1,40 @@
-import gradio as gr
-from ui.gradio_ui import get_ui       # Main UI structure
-from style import custom_css          # Custom CSS for styling
-
-# Import and initialize LangChain RAg chain for lawbot
-from modules.lawbot.rag_with_langchain import load_documents, split_documents, create_vectorstore, setup_rag_chain
-
-# Initialize RAg chain once at app startup
-documents = load_documents()
-docs = split_documents(documents)
-vectorstore = create_vectorstore(docs)
-qa_chain = setup_rag_chain(vectorstore)
-
-# Gradio app with custom CSS and UI
-with gr.Blocks(css=custom_css) as demo:
-    get_ui()  # Loads all tabs including PehchaanSetu and others
-
-# Launch the Gradio app
-if __name__ == "__main__":
-    url = demo.launch()
-    print(f"🚀 Saarthi app is running at: {url}")
+import gradio as gr
+import sys
+import os
+from ui.gradio_ui import get_ui       
+from style import custom_css          
+from modules.lawbot.rag_with_langchain import (
+    load_documents, 
+    split_documents, 
+    create_vectorstore, 
+    setup_rag_chain
+)
+
+# Global variable to hold the chain
+qa_chain = None
+
+def initialize_saarthi():
+    """Initialize the RAG pipeline only once."""
+    global qa_chain
+    if qa_chain is None:
+        print("--- Initializing Saarthi Legal Engine ---")
+        try:
+            documents = load_documents()
+            docs = split_documents(documents)
+            vectorstore = create_vectorstore(docs)
+            qa_chain = setup_rag_chain(vectorstore)
+            print("--- Initialization Complete ---")
+        except Exception as e:
+            print(f"❌ Initialization Error: {e}")
+            raise e
+    return qa_chain
+
+# Create the Blocks interface
+with gr.Blocks(css=custom_css) as demo:
+    # We call get_ui, but ensure your UI functions call initialize_saarthi() 
+    # when they need to perform a search.
+    get_ui()
+
+if __name__ == "__main__":
+    # Binding to 0.0.0.0 and port 7860 is required for Hugging Face Spaces
+    demo.launch(server_name="0.0.0.0", server_port=7860)