Update app.py

app.py

@@ -273,38 +273,43 @@ def handle_submit():
 
 def create_chat_input():
     """Create the chat input section with form handling"""
-    with st.form(key="chat_form", clear_on_submit=True):
-        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 คือวันที่เท่าไร?"
-        )
-        
-        # Button layout
-        col1, col2, col3 = st.columns([1, 1, 4])
-        
-        with col1:
-            submit_button = st.form_submit_button(
+    # Column layout for the entire input section
+    input_col, clear_col = st.columns([4, 1])
+    
+    with input_col:
+        # Create the form for chat input
+        with st.form(key="chat_form", clear_on_submit=True):
+            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 คือวันที่เท่าไร?"
+            )
+            
+            # Submit button in form
+            submitted = st.form_submit_button(
                 "📤 ส่งคำถาม",
                 type="primary",
-                use_container_width=True,
-                on_click=handle_submit
+                use_container_width=True
             )
-        
-        with col2:
-            if st.button("🗑️ ล้างประวัติ", type="secondary", use_container_width=True):
-                st.session_state.context_memory = []
-                st.session_state.chat_history = []
-                st.rerun()
+            
+            if submitted:
+                handle_submit()
+    
+    # Clear history button outside the form
+    with clear_col:
+        if st.button("🗑️ ล้างประวัติ", type="secondary", use_container_width=True):
+            st.session_state.context_memory = []
+            st.session_state.chat_history = []
+            st.rerun()
 
 def main():
     # Page config
@@ -352,7 +357,7 @@ def main():
     chat_col, info_col = st.columns([7, 3])
     
     with chat_col:
-        # Display chat history
+        # Display chat history first
         for i, (role, content) in enumerate(st.session_state.chat_history):
             if role == "user":
                 st.markdown(f"""
@@ -383,40 +388,10 @@ def main():
                                     {doc.content}
                                 </div>
                             """, unsafe_allow_html=True)
-
-        # Create chat input using the new form-based approach
+        
+        # Create chat input using the new implementation
         create_chat_input()
 
-        # 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 with enter key handling
-            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.context_memory = []
-                    st.session_state.chat_history = []
-                    # st.rerun()
-
     # Sidebar info column
     with info_col:
         st.markdown("""