Update app.py

app.py

@@ -274,79 +274,76 @@ 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"
-            )
-
-            # Button layout
-            col1, col2, col3 = st.columns([1, 1, 4])
-            
-            with col1:
-                send_query = st.button(
-                    "📤 ส่งคำถาม",
-                    type="primary",
-                    use_container_width=True,
-                    key="send_query_button"
-                )
-            
-            with col2:
-                clear_history = st.button(
-                    "🗑️ ล้างประวัติ",
-                    type="secondary",
-                    use_container_width=True,
-                    key="clear_history_button"
+            # Function to handle form submission
+            def on_submit():
+                st.session_state.chat_history = []
+                st.session_state.submitted = True
+                
+            # Initialize the form submit state if not present
+            if 'submitted' not in st.session_state:
+                st.session_state.submitted = False
+                
+            # Create form
+            with st.form(key='query_form'):
+                query = st.text_input(
+                    "",
+                    placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
+                    key="query_input"
                 )
-
-            # 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 = []
+                col1, col2, col3 = st.columns([1, 1, 4])
                 
-                # Add new query to history
-                add_to_history("user", query)
+                with col1:
+                    submit_button = st.form_submit_button(
+                        "📤 ส่งคำถาม",
+                        type="primary",
+                        on_click=on_submit,
+                        use_container_width=True
+                    )
                 
-                try:
-                    with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
-                        result = st.session_state.pipeline.process_query(query)
-                        add_to_history("assistant", result["answer"])
-                        
-                        with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
-                            for i, doc in enumerate(result["documents"], 1):
-                                st.markdown(f"""
-                                    <div style="padding: 1rem; background-color: #F9FAFB; border-radius: 8px; margin: 0.5rem 0;">
-                                        <strong>เอกสารที่ {i}:</strong><br>
-                                        {doc.content}
-                                    </div>
-                                """, unsafe_allow_html=True)
-                        
-                        with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
-                            st.json(result["query_info"])
-                        
-                        # Clear the input field
-                        st.session_state.query_input = ""
-                        st.rerun()
-                        
-                except Exception as e:
-                    st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
+                with col2:
+                    clear_button = st.form_submit_button(
+                        "🗑️ ล้างประวัติ",
+                        type="secondary",
+                        on_click=lambda: setattr(st.session_state, 'chat_history', []),
+                        use_container_width=True
+                    )
+
+            # Process query when form is submitted
+            if st.session_state.submitted and query:
+                if st.session_state.pipeline is None:
+                    st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
+                else:
+                    try:
+                        with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
+                            # Add new query to history
+                            add_to_history("user", query)
+                            
+                            result = st.session_state.pipeline.process_query(query)
+                            add_to_history("assistant", result["answer"])
+                            
+                            with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
+                                for i, doc in enumerate(result["documents"], 1):
+                                    st.markdown(f"""
+                                        <div style="padding: 1rem; background-color: #F9FAFB; border-radius: 8px; margin: 0.5rem 0;">
+                                            <strong>เอกสารที่ {i}:</strong><br>
+                                            {doc.content}
+                                        </div>
+                                    """, unsafe_allow_html=True)
+                            
+                            with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
+                                st.json(result["query_info"])
+                            
+                            # Reset submit state
+                            st.session_state.submitted = False
+                            st.rerun()
+                            
+                    except Exception as e:
+                        st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
             
-            elif send_query and not query:
+            elif st.session_state.submitted and not query:
                 st.warning("⚠️ กรุณาระบุคำถาม")
-
-            # Handle clear history action
-            if clear_history:
-                st.session_state.chat_history = []
-                st.session_state.query_input = ""
-                st.rerun()
+                st.session_state.submitted = False
     
     with info_col:
         # System information