Update app.py

app.py

@@ -15,109 +15,84 @@ if 'last_query' not in st.session_state:
     st.session_state.last_query = ""
 if 'submit_clicked' not in st.session_state:
     st.session_state.submit_clicked = False
-if 'init_time' not in st.session_state:
-    st.session_state.init_time = None
 if 'query_answered' not in st.session_state:
-    st.session_state.query_answered = False  # Tracks if the query has been answered
+    st.session_state.query_answered = False  # Tracks if query was answered
 
 # THEN: Import your modules
 from rag_engine import process_query, load_model
 from utils import setup_all_auth
 
-# Create a placeholder for our initialization message
+# Create a placeholder for initialization message
 init_message = st.empty()
 
-# ✅ Custom CSS for dynamic behavior
+# ✅ Custom CSS for border & button fixes
 st.markdown("""
 <style>
-/* ✅ Green by default */
-.input-container input[type="text"] {
+/* ✅ Green border by default */
+div[data-baseweb="input"] {
     border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
     transition: border 0.3s ease-in-out;
 }
 
 /* 🔴 Change to red while typing */
-.input-container input[type="text"]:focus {
+div[data-baseweb="input"]:focus-within {
     border: 2px solid #FF5722 !important;
 }
 
-/* ✅ Change back to green AFTER submission */
-.answered .input-container input[type="text"] {
+/* ✅ Ensure it turns green again AFTER submission */
+.answered div[data-baseweb="input"] {
     border: 2px solid #4CAF50 !important;
 }
 
-/* ✅ Make button a MUCH lighter green */
+/* ✅ Fully green button */
 .stButton>button {
-    background-color: #81C784 !important;
+    background-color: #4CAF50 !important;
     color: white !important;
-    border: 2px solid #66BB6A !important;
+    border: none !important;
     border-radius: 8px !important;
+    padding: 8px 16px;
+    font-size: 16px;
+    font-weight: bold;
+    cursor: pointer;
 }
 </style>
 """, unsafe_allow_html=True)
 
-# Handle initialization and success message timing
+# Handle initialization
 if not st.session_state.initialized:
     init_message.info("Hang in there! We are setting the system up for you. 😊")
 
     try:
-        # Setup authentication
         setup_all_auth()
-
-        # Load the model
         load_model()
-
-        # Mark as initialized and record time
         st.session_state.initialized = True
-        st.session_state.init_time = time.time()
-
-        # Show success message
+        st.session_state.query_answered = False  # Reset query state
         init_message.success("System initialized successfully!")
-
-        # Force rerun to start the timer for removing the message
-        time.sleep(0.1)  # Small delay to ensure message appears
-        st.rerun()
-
+        time.sleep(1)
+        init_message.empty()
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
-# Check if we need to hide the success message (after initialization)
-elif st.session_state.init_time is not None:
-    elapsed_time = time.time() - st.session_state.init_time
-    if elapsed_time >= 2.0:
-        init_message.empty()
-        st.session_state.init_time = None  # Reset timer after clearing
-
 # Function to handle form submission
 def handle_form_submit():
     if st.session_state.query_input:
         st.session_state.last_query = st.session_state.query_input
         st.session_state.submit_clicked = True
-        st.session_state.query_answered = True  # ✅ Ensure box goes back to green
-        st.session_state.query_input = ""  # Clear input box after submission
-        st.rerun()  # ✅ Force UI refresh so color resets
+        st.session_state.query_answered = True  # ✅ Ensure it resets to green
+        st.session_state.query_input = ""  # ✅ Clear input
+        st.experimental_rerun()  # ✅ Refresh UI properly
 
-# ✅ Wrap the text input inside a div to allow dynamic styling
+# ✅ Wrap the text input inside a div for styling
 css_class = "answered" if st.session_state.query_answered else ""
 st.markdown(f'<div class="{css_class}">', unsafe_allow_html=True)
 
-# ✅ Create a form for handling user input
+# ✅ Create a form for user input
 with st.form(key="query_form"):
-    # Input field with dynamic border behavior
-    st.markdown('<div class="input-container">', unsafe_allow_html=True)
-    query = st.text_input(
-        "Ask your question:",
-        key="query_input",
-        value="",
-        placeholder="Type here..."
-    )
-    st.markdown('</div>', unsafe_allow_html=True)  # ✅ Close input-container div
-
-    # Submit button
+    query = st.text_input("Ask your question:", key="query_input", placeholder="Type here...")
     submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
-st.markdown("</div>", unsafe_allow_html=True)  # ✅ Close the div for styling
+st.markdown("</div>", unsafe_allow_html=True)  # ✅ Close styling div
 
 # ✅ Display the last question after submission
 if st.session_state.last_query:
@@ -147,7 +122,7 @@ if st.session_state.submit_clicked and st.session_state.last_query:
                 st.write(citation)
 
             st.session_state.query_answered = True  # ✅ Mark query as answered
-            st.rerun()  # ✅ Refresh UI so that text box turns green again
+            st.experimental_rerun()  # ✅ Proper rerun to refresh UI
 
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")