Update app.py

app.py

@@ -46,6 +46,14 @@ st.markdown("""
     overflow: hidden !important;
     text-overflow: ellipsis !important;
 }
+
+/* Form submit button specific styling */
+button[type="submit"], 
+.stFormSubmit>button, 
+[data-testid="stFormSubmitButton"]>button {
+    background-color: #fff0f0 !important;
+    color: #3f51b5 !important;
+}
 .stButton>button:hover {
     background-color: #fafbff !important;
     border-color: #c5cae9 !important;
@@ -101,18 +109,11 @@ div.stInfo {
     gap: 10px;
 }
 
-/* Form submit button styling to match other buttons */
-button[kind="formSubmit"] {
-    background-color: #fff0f0 !important;
-    color: #3f51b5 !important;
-    border: 1px solid #e1e4f2 !important;
-    border-radius: 20px !important;
-    padding: 8px 16px !important;
-    box-shadow: 0 1px 2px rgba(0,0,0,0.03) !important;
-}
-button[kind="formSubmit"]:hover {
-    background-color: #fafbff !important;
-    border-color: #c5cae9 !important;
+/* Placeholder for buttons */
+.button-placeholder {
+    min-height: 38px;
+    min-width: 120px;
+    margin: 0 5px;
 }
 </style>
 <div class="main-title">Spirituality Q&A</div>
@@ -200,13 +201,13 @@ for row_idx, row in enumerate(question_rows):
             if st.button(q, key=f"r{row_idx}_q{i}", use_container_width=True):
                 set_query(q)
 
-# Function to handle form submission with enter key handling
+# Function to handle form submission
 def handle_form_submit():
     if st.session_state.query_input and st.session_state.query_input.strip():
         st.session_state.last_query = st.session_state.query_input.strip()
         st.session_state.submit_clicked = True
-        # Clear input and focus out to reset border color
-        st.session_state.query_input = ""
+        # Increment the form key to force a reset
+        st.session_state.form_key += 1
         st.rerun()
 
 # Create a form with dynamic key for resetting
@@ -233,12 +234,7 @@ if st.session_state.submit_clicked and st.session_state.last_query:
 
     with st.spinner("Processing your question..."):
         try:
-            # result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
-            # Mock answer for testing
-            result = {
-                "answer_with_rag": "🔹 This is a placeholder answer for testing UI.",
-                "citations": "🔹 Sample Source 1\n🔹 Sample Source 2"
-            }
+            result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
             st.subheader("Answer:")
             st.write(result["answer_with_rag"])
             st.subheader("Sources:")