Update app.py

app.py

@@ -194,12 +194,6 @@ def add_to_history(role: str, message: str):
         # Get the corresponding user query (previous message)
         user_query = st.session_state.chat_history[-2][1]
         add_to_qa_history(user_query, message)
-
-# Modify the clear history button handler in main():
-if clear_history:
-    st.session_state.chat_history = []
-    clear_qa_history()  # Clear saved history
-    st.rerun()
     
 def display_chat_history():
     """Display chat history with enhanced styling"""
@@ -277,7 +271,7 @@ def main():
 
             # Button layout
             col1, col2, col3 = st.columns([1, 1, 4])
-
+        
             with col1:
                 send_query = st.button(
                     "📤 ส่งคำถาม",
@@ -285,7 +279,7 @@ def main():
                     use_container_width=True,
                     key="send_query_button"
                 )
-
+        
             with col2:
                 clear_history = st.button(
                     "🗑️ ล้างประวัติ",
@@ -373,5 +367,11 @@ def main():
             "พร้อมใช้งาน" if st.session_state.pipeline else "ไม่พร้อมใช้งาน"
         ), unsafe_allow_html=True)
 
+        # Handle clear history action
+        if clear_history:
+            st.session_state.chat_history = []
+            clear_qa_history()
+            st.rerun()
+
 if __name__ == "__main__":
     main()