Update app,py

app.py

@@ -19,8 +19,6 @@ 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
-if 'input_cleared' not in st.session_state:
-    st.session_state.input_cleared = False  # Ensure input box resets after submission
 
 # THEN: Import your modules
 from rag_engine import process_query, load_model
@@ -41,28 +39,28 @@ st.markdown("""
 }
 
 /* ✅ Initially green border */
-div[data-baseweb="input"] {
+input[type="text"] {
     border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
     transition: border 0.3s ease-in-out;
 }
 
 /* 🔴 Change to red while typing */
-div[data-baseweb="input"]:focus-within {
+input[type="text"]:focus {
     border: 2px solid #FF5722 !important;
 }
 
 /* ✅ Revert to green after submission */
-div.answered-query {
+.answered-query input[type="text"] {
     border: 2px solid #4CAF50 !important;
 }
 
-/* ✅ Green-bordered button */
+/* ✅ Green-bordered button (not filled) */
 .stButton>button {
     border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
-    color: white !important;
-    background-color: #4CAF50 !important;
+    color: black !important;
+    background-color: white !important;
 }
 </style>
 <div class="main-title">Indian Spiritual Texts Q&A</div>
@@ -100,37 +98,29 @@ elif st.session_state.init_time is not None:
         init_message.empty()
         st.session_state.init_time = None  # Reset timer after clearing
 
-# Handle form submission - ensures input is cleared and focus is removed
-if 'form_submitted' not in st.session_state:
-    st.session_state.form_submitted = False
-
 # 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.form_submitted = True
         st.session_state.query_answered = False  # Mark as not yet answered
-        st.session_state.input_cleared = True  # Flag to clear input
+        st.session_state.query_input = ""  # Clear input box after submission
 
-# Create a form for handling the user input
+# ✅ Create a form for handling user input
 with st.form(key="query_form"):
-    # Apply CSS class dynamically based on submission status
-    css_class = "answered-query" if st.session_state.query_answered else ""
+    # Input field with dynamic border behavior
     query = st.text_input(
-        "Ask your question:", key="query_input", value="", placeholder="Type here..."
+        "Ask your question:",
+        key="query_input",
+        value="",
+        placeholder="Type here..."
     )
 
-    # Submit button within the form
+    # Submit button
     submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
-# ✅ Reset input field after submission
-if st.session_state.input_cleared:
-    st.session_state.input_cleared = False
-    st.rerun()
-
-# Display the current question if there is one
-if st.session_state.last_query:
+# ✅ Display the last question after submission
+if st.session_state.submit_clicked and st.session_state.last_query:
     st.markdown("### Current Question:")
     st.info(st.session_state.last_query)
 
@@ -141,12 +131,10 @@ with col1:
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# Only process the query if explicitly submitted
+# ✅ Process the query if submitted
 if st.session_state.submit_clicked and st.session_state.last_query:
-    # Reset the submit flag
-    st.session_state.submit_clicked = False
+    st.session_state.submit_clicked = False  # Reset flag
 
-    # Process the query
     with st.spinner("Processing your question..."):
         try:
             result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
@@ -158,8 +146,7 @@ if st.session_state.submit_clicked and st.session_state.last_query:
             for citation in result["citations"].split("\n"):
                 st.write(citation)
 
-            # ✅ Mark the query as answered
-            st.session_state.query_answered = True
+            st.session_state.query_answered = True  # ✅ Mark query as answered
 
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")