Update app.py

app.py

@@ -273,53 +273,60 @@ def main():
                     </span>
                 </label>
             """, unsafe_allow_html=True)
-            
-            # Function to handle form submission
-            def on_submit():
+
+            # Callback function to clear input
+            def clear_input():
+                st.session_state.query_input = ""
                 st.session_state.chat_history = []
-                st.session_state.submitted = True
-                
-            # Initialize the form submit state if not present
-            if 'submitted' not in st.session_state:
-                st.session_state.submitted = False
                 
-            # Create form
-            with st.form(key='query_form'):
-                query = st.text_input(
-                    "",
-                    placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
-                    key="query_input"
+            # Initialize the submit state if not present
+            if 'submit_state' not in st.session_state:
+                st.session_state.submit_state = False
+
+            # Text input with callback
+            query = st.text_input(
+                "",
+                placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
+                key="query_input",
+                on_change=clear_input if st.session_state.get('submit_state', False) else None
+            )
+
+            # Button layout
+            col1, col2, col3 = st.columns([1, 1, 4])
+            
+            with col1:
+                send_query = st.button(
+                    "📤 ส่งคำถาม",
+                    type="primary",
+                    use_container_width=True,
+                    key="send_query_button"
+                )
+            
+            with col2:
+                clear_history = st.button(
+                    "🗑️ ล้างประวัติ",
+                    type="secondary",
+                    use_container_width=True,
+                    key="clear_history_button",
+                    on_click=clear_input
                 )
-                
-                col1, col2, col3 = st.columns([1, 1, 4])
-                
-                with col1:
-                    submit_button = st.form_submit_button(
-                        "📤 ส่งคำถาม",
-                        type="primary",
-                        on_click=on_submit,
-                        use_container_width=True
-                    )
-                
-                with col2:
-                    clear_button = st.form_submit_button(
-                        "🗑️ ล้างประวัติ",
-                        type="secondary",
-                        on_click=lambda: setattr(st.session_state, 'chat_history', []),
-                        use_container_width=True
-                    )
 
-            # Process query when form is submitted
-            if st.session_state.submitted and query:
+            # Process query
+            if send_query and query:
                 if st.session_state.pipeline is None:
                     st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
                 else:
                     try:
                         with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
-                            # Add new query to history
-                            add_to_history("user", query)
+                            # Set submit state to trigger input clearing
+                            st.session_state.submit_state = True
                             
+                            # Process the query
                             result = st.session_state.pipeline.process_query(query)
+                            
+                            # Clear previous history and add new QA pair
+                            st.session_state.chat_history = []
+                            add_to_history("user", query)
                             add_to_history("assistant", result["answer"])
                             
                             with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
@@ -334,17 +341,15 @@ def main():
                             with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
                                 st.json(result["query_info"])
                             
-                            # Reset submit state
-                            st.session_state.submitted = False
                             st.rerun()
                             
                     except Exception as e:
                         st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
             
-            elif st.session_state.submitted and not query:
+            elif send_query and not query:
                 st.warning("⚠️ กรุณาระบุคำถาม")
-                st.session_state.submitted = False
-    
+                st.session_state.submit_state = False
+
     with info_col:
         # System information
         st.markdown("""