Update app.py

app.py

@@ -230,26 +230,17 @@ def submit():
         st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
         return
 
-    # Store user query
     user_query = st.session_state.query_input
     st.session_state.chat_history.append(("user", user_query))
 
-    # Avoid infinite reruns
-    if "processing_query" in st.session_state and st.session_state.processing_query:
-        return
-
-    st.session_state.processing_query = True
-
-    # Display a loading message
+    # Display loading message
     st.session_state.chat_history.append(("assistant", {
         "answer": "🔍 กำลังค้นหาคำตอบ...",
         "documents": [],
         "query_info": {}
     }))
-    
-    # **No `st.rerun()` here yet! Just update chat**
+
     st.session_state.query_input = ""  # Clear input after submission
-    st.rerun()  # This will force a refresh once
 
     try:
         # Process the query
@@ -262,17 +253,18 @@ def submit():
             "query_info": result["query_info"]
         })
 
+        # Save QA history
+        add_to_qa_history(user_query, result["answer"])
+
     except Exception as e:
-        # Replace with an error message if query fails
         st.session_state.chat_history[-1] = ("assistant", {
             "answer": f"❌ เกิดข้อผิดพลาด: {str(e)}",
             "documents": [],
             "query_info": {}
         })
+        st.error(f"Query processing error: {e}")
 
-    # Reset processing flag
-    st.session_state.processing_query = False
-    st.rerun()  # Refresh UI once processing is done
+    st.rerun()
 
 def main():
     # Page config