Update app.py

app.py

@@ -114,21 +114,34 @@ def main():
         # Display chat history
         display_chat_history()
         
+        # Initialize query input in session state if not present
+        if 'current_query' not in st.session_state:
+            st.session_state.current_query = ''
+
         # Main query interface
         query = st.text_input(
             "โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:",
+            value=st.session_state.current_query,
             placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
             key="query_input"
         )
 
+        # Update current query in session state
+        st.session_state.current_query = query
+
         # Add query button and clear chat button in the same line
         col1, col2 = st.columns([1, 4])
         with col1:
             if st.button("ส่งคำถาม", type="primary", use_container_width=True):
-                process_query(query)
+                if query:  # Only process if query is not empty
+                    process_query(query)
+                    # Clear the input only after successful processing
+                    st.session_state.current_query = ''
+                    st.rerun()
         with col2:
             if st.button("ล้างประวัติการสนทนา", type="secondary", use_container_width=True):
                 st.session_state.chat_history = []
+                st.session_state.current_query = ''
                 st.rerun()
     
     with info_col:
@@ -152,5 +165,4 @@ def main():
         st.markdown(f"**สถานะระบบ:** {'🟢 พร้อมใช้งาน' if st.session_state.pipeline else '🔴 ไม่พร้อมใช้งาน'}")
 
 if __name__ == "__main__":
-    main()
-    
+    main()