Update app.py

app.py

@@ -1,40 +1,30 @@
-import gradio as gr
-import sys
 import os
-from ui.gradio_ui import get_ui       
-from style import custom_css          
+from dotenv import load_dotenv
+load_dotenv()
+
+import gradio as gr
+from ui.gradio_ui import get_ui
+from style import custom_css
+
+# Import RAG pipeline
 from modules.lawbot.rag_with_langchain import (
-    load_documents, 
-    split_documents, 
-    create_vectorstore, 
+    load_documents,
+    split_documents,
+    create_vectorstore,
     setup_rag_chain
 )
 
-# Global variable to hold the chain
-qa_chain = None
+def initialize_rag():
+    documents = load_documents()
+    docs = split_documents(documents)
+    vectorstore = create_vectorstore(docs)
+    return setup_rag_chain(vectorstore)
+
+qa_chain = initialize_rag()
 
-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
+demo = get_ui(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()
+demo.css = custom_css
 
 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)
+    demo.launch()