Update app.py

app.py

@@ -280,10 +280,10 @@ def main():
                 key="query_input",
                 value="" if 'clear_input' in st.session_state and st.session_state.clear_input else st.session_state.get('query_input', '')
             )
-
-            # Button layout inside main() function
-            col1, col2, col3 = st.columns([1, 1, 4])
         
+            # Button layout
+            col1, col2, col3 = st.columns([1, 1, 4])
+            
             with col1:
                 send_query = st.button(
                     "📤 ส่งคำถาม",
@@ -291,7 +291,7 @@ def main():
                     use_container_width=True,
                     key="send_query_button"
                 )
-        
+            
             with col2:
                 clear_history = st.button(
                     "🗑️ ล้างประวัติ",
@@ -299,6 +299,7 @@ def main():
                     use_container_width=True,
                     key="clear_history_button"
                 )
+        
             # Process query
             if send_query and query:
                 if st.session_state.pipeline is None:
@@ -323,21 +324,26 @@ def main():
                         
                         with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
                             st.json(result["query_info"])
-
+                        
                         # Clear the input after successful submission
                         st.session_state.clear_input = True
                         st.rerun()
                         
                 except Exception as e:
                     st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
-            
+                
             elif send_query and not query:
                 st.warning("⚠️ กรุณาระบุคำถาม")
-
-        # Handle clear history action
-        if clear_history:
-            st.session_state.chat_history = []
-            st.rerun()
+        
+            # Handle clear history action
+            if clear_history:
+                st.session_state.chat_history = []
+                st.session_state.clear_input = True  # Also clear input when clearing history
+                st.rerun()
+        
+            # Reset clear_input flag at the end of each render
+            if 'clear_input' in st.session_state:
+                del st.session_state.clear_input
     
     with info_col:
         # System information