Update app.py

app.py

@@ -36,8 +36,8 @@ 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():
@@ -77,25 +77,20 @@ button[type="submit"],
     background-color: #fafbff !important;
     border-color: #c5cae9 !important;
 }
-/* Special styling for thank you button - Modified to match Get Answer */
-.thank-you-button {
-    display: flex;
-    justify-content: center;
-    margin: 10px 0;
-}
-.thank-you-button > div > button {
-    background-color: #fff0f0 !important;
-    color: #3f51b5 !important;
-    border: 1px solid #e1e4f2 !important;
-    border-radius: 8px !important;
-    padding: 8px 16px !important;
-    box-shadow: 0 1px 2px rgba(0,0,0,0.03) !important;
+/* Special styling for thank you button */
+.thank-you-button > button {
+    background-color: #f8e6ff !important;
+    border-radius: 8px !important; 
+    padding: 10px 20px !important;
     font-weight: normal !important;
-    min-width: 150px !important;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.1) !important;
+    transition: all 0.3s ease !important;
+    border: 1px solid #d8b9ff !important;
+    color: #6a1b9a !important;
 }
-.thank-you-button > div > button:hover {
-    background-color: #fafbff !important;
-    border-color: #c5cae9 !important;
+.thank-you-button > button:hover {
+    background-color: #f0d6ff !important;
+    transform: translateY(-1px) !important;
 }
 /* Input field styling */
 div[data-baseweb="input"] {
@@ -185,9 +180,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!")
@@ -201,17 +196,19 @@ elif st.session_state.init_time is not None:
         init_message.empty()
         st.session_state.init_time = None
 
-# Thank You button that matches Get Answer button styling
-st.markdown('<div class="thank-you-button">', unsafe_allow_html=True)
-col1, col2, col3 = st.columns([1, 2, 1])
-with col2:
-    if st.session_state.show_acknowledgment:
-        button_text = "Hide Thank You Note"
-    else:
-        button_text = "Show Thank You Note"
-    if st.button(button_text, key="thank_you_button", use_container_width=True, disabled=st.session_state.is_processing):
-        toggle_acknowledgment()
-st.markdown('</div>', unsafe_allow_html=True)
+# Heartfelt acknowledgment section with toggle - improved button text and disabled during processing
+if st.session_state.show_acknowledgment:
+    st.markdown('<div class="thank-you-button">', unsafe_allow_html=True)
+    # Debouncing applied: disabled if processing
+    if st.button("Hide Thank You Note", on_click=toggle_acknowledgment, disabled=st.session_state.is_processing):
+        pass
+    st.markdown('</div>', unsafe_allow_html=True)
+else:
+    st.markdown('<div class="thank-you-button">', unsafe_allow_html=True)
+    # Debouncing applied: disabled if processing
+    if st.button("Show Thank You Note", on_click=toggle_acknowledgment, disabled=st.session_state.is_processing):
+        pass
+    st.markdown('</div>', unsafe_allow_html=True)
 
 if st.session_state.show_acknowledgment:
     st.markdown('<div class="acknowledgment-container">', unsafe_allow_html=True)
@@ -328,9 +325,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)
-            # Simulated result for testing
-            result = {"answer_with_rag": "This is a sample answer.", "citations": "Sample citation 1\nSample citation 2"}
+            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": ""}