Update app.py

app.py

@@ -227,19 +227,19 @@ def submit():
             st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
             return
 
-        # Add user query to chat history FIRST
+        # Store user query
         user_query = st.session_state.query_input
         st.session_state.chat_history.append(("user", user_query))
 
-        # Show loading spinner in the chat section
-        st.session_state.chat_history.append(("assistant", "🔍 กำลังค้นหาคำตอบ..."))
-        st.rerun()  # Force rerun to update UI before running pipeline
+        # Add loading message as an assistant's response
+        st.session_state.chat_history.append(("assistant", {"answer": "🔍 กำลังค้นหาคำตอบ...", "documents": [], "query_info": {}}))
+        st.rerun()  # Force UI update
 
         try:
             # Process query
             result = st.session_state.pipeline.process_query(user_query)
 
-            # Replace loading spinner with the actual answer
+            # Replace loading message with actual response
             st.session_state.chat_history[-1] = ("assistant", {
                 "answer": result["answer"],
                 "documents": result["documents"],
@@ -248,14 +248,12 @@ def submit():
 
         except Exception as e:
             # Replace spinner with error message
-            st.session_state.chat_history[-1] = ("assistant", f"❌ เกิดข้อผิดพลาด: {str(e)}")
+            st.session_state.chat_history[-1] = ("assistant", {"answer": f"❌ เกิดข้อผิดพลาด: {str(e)}", "documents": [], "query_info": {}})
 
-        # Reset input box
+        # Reset input
         st.session_state.query_input = ""
-        st.rerun()  # Force update the UI
-
-
-        
+        st.rerun()
+     
 def main():
     # Page config
     st.set_page_config(