Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import time
+import streamlit.components.v1 as components
 
 # FIRST: Set page config before ANY other Streamlit command
 st.set_page_config(page_title="Spirituality Q&A", page_icon="🕉️")
@@ -36,13 +37,31 @@ if 'page_loaded' not in st.session_state:
     st.session_state.last_answer = None
 
 # THEN: Import your modules
-# from rag_engine import process_query, load_model, cached_load_data_files
-# from utils import setup_all_auth
+from rag_engine import process_query, load_model, cached_load_data_files
+from utils import setup_all_auth
 
 # Function to toggle acknowledgment visibility
 def toggle_acknowledgment():
     st.session_state.show_acknowledgment = not st.session_state.show_acknowledgment
 
+# Custom HTML/JS to navigate to sources page
+def navigate_to_sources():
+    components.html(
+        """
+        <script>
+        // Wait for the page to fully load
+        document.addEventListener('DOMContentLoaded', (event) => {
+            // This select the nav item for the Sources page in the sidebar
+            const sourcesLink = Array.from(document.querySelectorAll('a.css-z5fcl4')).find(el => el.innerText === 'Sources');
+            if (sourcesLink) {
+                sourcesLink.click();
+            }
+        });
+        </script>
+        """,
+        height=0,
+    )
+
 # Custom styling (pure CSS)
 st.markdown("""
 <style>
@@ -181,6 +200,7 @@ div.stInfo {
 /* Source link styling */
 .source-link {
     color: #3f51b5;
+    font-weight: bold;
     text-decoration: underline;
     cursor: pointer;
 }
@@ -211,9 +231,9 @@ if not st.session_state.initialized:
     init_message.info("Hang in there! We are setting the system up for you. 😊")
     try:
         # Setup authentication and preload heavy resources
-        # setup_all_auth()
-        # load_model()  # This uses cached_load_model via alias
-        # cached_load_data_files()  # Preload FAISS index, text chunks, and metadata
+        setup_all_auth()
+        load_model()  # This uses cached_load_model via alias
+        cached_load_data_files()  # Preload FAISS index, text chunks, and metadata
         st.session_state.initialized = True
         st.session_state.init_time = time.time()
         init_message.success("System initialized successfully!")
@@ -246,13 +266,10 @@ if st.session_state.show_acknowledgment:
     This application is merely a humble vessel for the ocean of wisdom they have shared with the world. We claim no ownership of these teachings - only profound gratitude for the opportunity to help make them more accessible.
     """)
     
-    # Create a source link that works with Streamlit navigation
-    sources_link = """
-    <div class="more-info-link">
-    For detailed information about our sources, please visit the <span class="source-link" onclick="parent.window.location.href='Sources'">Sources</span> page in the navigation menu.
-    </div>
-    """
-    st.markdown(sources_link, unsafe_allow_html=True)
+    # Link to Sources using Streamlit's built-in way
+    st.markdown('<div class="more-info-link">', unsafe_allow_html=True)
+    st.write("For detailed information about our sources, please visit the *Sources* page in the navigation menu.")
+    st.markdown('</div>', unsafe_allow_html=True)
     st.markdown('</div>', unsafe_allow_html=True)
 
 # Function to handle query selection from the common questions buttons
@@ -345,7 +362,7 @@ if st.session_state.submit_clicked and st.session_state.last_query:
     st.session_state.submit_clicked = False
     with st.spinner("Processing your question..."):
         try:
-            # result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
+            result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
             st.session_state.last_answer = result  # Store result in session state
         except Exception as e:
             st.session_state.last_answer = {"answer_with_rag": f"Error processing query: {str(e)}", "citations": ""}
@@ -364,14 +381,13 @@ if st.session_state.last_answer is not None:
 # Add helpful information
 st.markdown("---")
 
-# About section with working sources link
-about_text = """
+# About section with simpler text mention of sources
+st.markdown("""
 ### About this app
 This application uses a Retrieval-Augmented Generation (RAG) system to answer questions about spirituality based on insights from Indian spiritual texts. 
 It searches through a database of texts to find relevant passages and generates answers based on those passages. 
-For more information about the texts, see <span class="source-link" onclick="parent.window.location.href='Sources'">sources</span>.
-"""
-st.markdown(about_text, unsafe_allow_html=True)
+For more information about the texts, see *Sources* in the navigation menu.
+""")
 
 # Citation note at the bottom - improved with support message
 st.markdown('<div class="citation-note">', unsafe_allow_html=True)