Update app.py

app.py

@@ -4,7 +4,7 @@ import time
 # FIRST: Set page config before ANY other Streamlit command
 st.set_page_config(page_title="Indian Spiritual RAG")
 
-# Initialize ALL session state variables right at the beginning
+# Initialize ALL session state variables
 if 'initialized' not in st.session_state:
     st.session_state.initialized = False
 if 'model' not in st.session_state:
@@ -19,6 +19,8 @@ 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
@@ -38,12 +40,6 @@ st.markdown("""
     margin-bottom: 1rem;
 }
 
-/* Styling for the submit button */
-.stButton>button {
-    border: 2px solid #FF5722 !important;
-    border-radius: 8px !important;
-}
-
 /* ✅ Initially green border */
 div[data-baseweb="input"] {
     border: 2px solid #4CAF50 !important;
@@ -60,6 +56,14 @@ div[data-baseweb="input"]:focus-within {
 div.answered-query {
     border: 2px solid #4CAF50 !important;
 }
+
+/* ✅ Green-bordered button */
+.stButton>button {
+    border: 2px solid #4CAF50 !important;
+    border-radius: 8px !important;
+    color: white !important;
+    background-color: #4CAF50 !important;
+}
 </style>
 <div class="main-title">Indian Spiritual Texts Q&A</div>
 """, unsafe_allow_html=True)
@@ -107,21 +111,22 @@ def handle_form_submit():
         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
 
 # Create a form for handling the 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 ""
     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 = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
-# Force a rerun after form submission to reset the input field and focus
-if st.session_state.form_submitted:
-    st.session_state.form_submitted = False
+# ✅ 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
@@ -153,7 +158,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
+            # ✅ Mark the query as answered
             st.session_state.query_answered = True
 
         except Exception as e: