Update app.py

app.py

@@ -262,31 +262,28 @@ def main():
     """, unsafe_allow_html=True)
 
     chat_col, info_col = st.columns([7, 3])
-
-    def clear_text():
-        st.session_state.query_input = ''
     
-    def handle_submit():
-        query = st.session_state.query_input
-        if query:
+    def submit():
+        if st.session_state.query_input:
             if st.session_state.pipeline is None:
                 st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
                 return
                 
             try:
                 with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
-                    result = st.session_state.pipeline.process_query(query)
+                    result = st.session_state.pipeline.process_query(st.session_state.query_input)
                     
                     # Update session state
-                    st.session_state.chat_history.append(("user", query))
+                    st.session_state.chat_history.append(("user", st.session_state.query_input))
                     st.session_state.chat_history.append(("assistant", {
                         "answer": result["answer"],
                         "documents": result["documents"],
                         "query_info": result["query_info"]
                     }))
                     
-                    # Clear input
-                    clear_text()
+                    # Reset the input
+                    st.session_state.query_input = ""
+                    st.rerun()
                     
             except Exception as e:
                 st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
@@ -294,36 +291,7 @@ def main():
             st.warning("⚠️ กรุณาระบุคำถาม")
 
     with chat_col:
-        # Input section at the top
-        with st.container():
-            st.markdown("""
-                <label for="query_input" style="font-size: 1.2rem; font-weight: 600; margin-bottom: 1rem; display: block;">
-                    <span style="color: #ffffff; border-left: 4px solid #ffffff; padding-left: 0.8rem;">
-                        โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:
-                    </span>
-                </label>
-            """, unsafe_allow_html=True)
-            
-            # Text input
-            query = st.text_input(
-                "",
-                key="query_input",
-                placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
-            )
-
-            # Button layout
-            col1, col2, col3 = st.columns([1, 1, 4])
-            
-            with col1:
-                if st.button("📤 ส่งคำถาม", type="primary", use_container_width=True):
-                    handle_submit()
-            
-            with col2:
-                if st.button("🗑️ ล้างประวัติ", type="secondary", use_container_width=True):
-                    st.session_state.chat_history = []
-                    st.rerun()
-
-        # Chat history display
+        # Chat history display first
         for i, (role, content) in enumerate(st.session_state.chat_history):
             if role == "user":
                 st.markdown(f"""
@@ -353,7 +321,36 @@ def main():
                 with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
                     st.json(content['query_info'])
 
-    # Info column content remains the same
+        # Input section at the bottom
+        with st.container():
+            st.markdown("""
+                <label for="query_input" style="font-size: 1.2rem; font-weight: 600; margin-bottom: 1rem; display: block;">
+                    <span style="color: #ffffff; border-left: 4px solid #ffffff; padding-left: 0.8rem;">
+                        โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:
+                    </span>
+                </label>
+            """, unsafe_allow_html=True)
+            
+            # Text input
+            st.text_input(
+                "",
+                key="query_input",
+                placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
+                on_change=submit
+            )
+
+            # Button layout
+            col1, col2, col3 = st.columns([1, 1, 4])
+            
+            with col1:
+                st.button("📤 ส่งคำถาม", type="primary", use_container_width=True, on_click=submit)
+            
+            with col2:
+                if st.button("🗑️ ล้างป���ะวัติ", type="secondary", use_container_width=True):
+                    st.session_state.chat_history = []
+                    st.rerun()
+
+    # Info column remains the same
     with info_col:
         st.markdown("""
             <div style="background-color: #F9FAFB; padding: 1.5rem; border-radius: 12px; margin-bottom: 2rem;">