Update app.py

app.py

@@ -274,13 +274,16 @@ def main():
                 </label>
             """, unsafe_allow_html=True)
             
+            # Initialize query_input in session state if not present
+            if 'query_input' not in st.session_state:
+                st.session_state.query_input = ""
+            
             query = st.text_input(
                 "",
                 placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
-                key="query_input",
-                value="" if 'clear_input' in st.session_state and st.session_state.clear_input else st.session_state.get('query_input', '')
+                key="query_input"
             )
-        
+
             # Button layout
             col1, col2, col3 = st.columns([1, 1, 4])
             
@@ -299,15 +302,19 @@ def main():
                     use_container_width=True,
                     key="clear_history_button"
                 )
-        
+
             # Process query
             if send_query and query:
                 if st.session_state.pipeline is None:
                     st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
                     return
-                    
+                
+                # Clear previous chat history
+                st.session_state.chat_history = []
+                
+                # Add new query to history
                 add_to_history("user", query)
-                    
+                
                 try:
                     with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
                         result = st.session_state.pipeline.process_query(query)
@@ -325,25 +332,21 @@ def main():
                         with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
                             st.json(result["query_info"])
                         
-                        # Clear the input after successful submission
-                        st.session_state.clear_input = True
+                        # Clear the input field
+                        st.session_state.query_input = ""
                         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.session_state.clear_input = True  # Also clear input when clearing history
+                st.session_state.query_input = ""
                 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