Update app.py

app.py

@@ -28,10 +28,16 @@ if 'last_answer' not in st.session_state:
 # Add new session state for showing/hiding acknowledgment
 if 'show_acknowledgment' not in st.session_state:
     st.session_state.show_acknowledgment = False
+# Add page change detection
+if 'page_loaded' not in st.session_state:
+    st.session_state.page_loaded = True
+    # Reset query state when returning to home page
+    st.session_state.last_query = ""
+    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():
@@ -71,6 +77,26 @@ 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;
+    font-weight: normal !important;
+    min-width: 150px !important;
+}
+.thank-you-button > div > button:hover {
+    background-color: #fafbff !important;
+    border-color: #c5cae9 !important;
+}
 /* Input field styling */
 div[data-baseweb="input"] {
     border: 1px solid #fff0f0 !important;
@@ -121,17 +147,17 @@ div.stInfo {
     min-width: 120px;
     margin: 0 5px;
 }
-/* Acknowledgment section styling */
+/* Acknowledgment section styling - fully fixed */
 .acknowledgment-container {
-    background: #ffffff;
-    border: 1px solid #e1e4f2;
+    background-color: #f8f5ff;
+    border: 1px solid #e0d6fe;
     border-radius: 8px;
     padding: 15px;
     margin: 20px 0;
     box-shadow: 0 2px 5px rgba(0,0,0,0.05);
 }
 .acknowledgment-header {
-    color: #c0392b;
+    color: #6a1b9a;
     font-size: 1.3rem;
     margin-bottom: 10px;
     text-align: center;
@@ -159,9 +185,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!")
@@ -175,13 +201,17 @@ elif st.session_state.init_time is not None:
         init_message.empty()
         st.session_state.init_time = None
 
-# Heartfelt acknowledgment section with toggle - improved button text
-if st.session_state.show_acknowledgment:
-    if st.button("Hide Gratitude and Acknowledgments", on_click=toggle_acknowledgment):
-        pass
-else:
-    if st.button("Show Gratitude and Acknowledgments", on_click=toggle_acknowledgment):
-        pass
+# 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)
 
 if st.session_state.show_acknowledgment:
     st.markdown('<div class="acknowledgment-container">', unsafe_allow_html=True)
@@ -298,7 +328,9 @@ 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)
+            # Simulated result for testing
+            result = {"answer_with_rag": "This is a sample answer.", "citations": "Sample citation 1\nSample citation 2"}
             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": ""}