Update app.py

app.py

@@ -49,7 +49,7 @@ div[data-baseweb="input"]:focus-within {
 .stButton>button {
     background-color: #4CAF50 !important;
     color: white !important;
-    border: none !important;
+    border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
     padding: 8px 16px;
     font-size: 16px;
@@ -74,24 +74,25 @@ if not st.session_state.initialized:
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
-# 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
+# ✅ Create a form for user input
+with st.form(key="query_form"):
+    query = st.text_input(
+        "Ask your question:",
+        key="query_input",
+        placeholder="Type here...",
+    )
+    submit_button = st.form_submit_button("Get Answer")
+
+# ✅ Handle submission **without st.rerun()**
+if submit_button:
+    if query.strip():
+        st.session_state.last_query = query.strip()
         st.session_state.submit_clicked = True
-        st.session_state.query_answered = True  # ✅ Ensure it resets to green
-        st.session_state.query_input = ""  # ✅ Clear input
-        st.experimental_rerun()  # ✅ Refresh UI properly
+        st.session_state.query_answered = True  # ✅ Ensure text box resets to green
 
 # ✅ 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 user input
-with st.form(key="query_form"):
-    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 styling div
 
 # ✅ Display the last question after submission
@@ -107,7 +108,7 @@ with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
 # ✅ Process the query if submitted
-if st.session_state.submit_clicked and st.session_state.last_query:
+if st.session_state.submit_clicked:
     st.session_state.submit_clicked = False  # Reset flag
 
     with st.spinner("Processing your question..."):
@@ -121,9 +122,6 @@ if st.session_state.submit_clicked and st.session_state.last_query:
             for citation in result["citations"].split("\n"):
                 st.write(citation)
 
-            st.session_state.query_answered = True  # ✅ Mark query as answered
-            st.experimental_rerun()  # ✅ Proper rerun to refresh UI
-
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")