Update app.py

app.py

@@ -460,7 +460,7 @@ def handle_submit(user_query: str):
             st.rerun()
 
 def create_chat_input():
-    """Create chat input with enhanced configuration and clear context button"""
+    """Create chat input with enhanced configuration and combined clear button"""
     with st.form(key="chat_form", clear_on_submit=True):
         st.markdown("""
             <label for="query_input" style="font-size: 1.2rem; font-weight: 600; margin-bottom: 1rem; display: block;">
@@ -476,7 +476,7 @@ def create_chat_input():
             placeholder="เช่น: วิชาเลือกมีอะไรบ้าง?"
         )
         
-        col1, col2, col3 = st.columns([5, 3, 2])
+        col1, col2 = st.columns([5, 5])
         
         with col1:
             submitted = st.form_submit_button(
@@ -486,15 +486,8 @@ def create_chat_input():
             )
         
         with col2:
-            clear_button = st.form_submit_button(
-                "🗑️ ล้างประวัติการแสดง",
-                type="secondary",
-                use_container_width=True
-            )
-            
-        with col3:
-            clear_context_button = st.form_submit_button(
-                "🧠 ล้างบริบทสนทนา",
+            clear_all_button = st.form_submit_button(
+                "🗑️ ล้างประวัติและบริบทสนทนา",
                 type="secondary",
                 use_container_width=True
             )
@@ -502,13 +495,13 @@ def create_chat_input():
         if submitted:
             handle_submit(query)
         
-        if clear_button:
+        if clear_all_button:
+            # Clear chat history
             st.session_state.chat_history = []
-            st.rerun()
-            
-        if clear_context_button:
+            # Clear conversation context
             clear_conversation_context()
-            st.info("ล้างบริบทสนทนาแล้ว แต่ยังคงประวัติการแสดงสนทนา")
+            st.info("ล้างประวัติและบริบทสนทนาแล้ว")
+            st.rerun()
 
 def main():
     # Page config