Move input boxes over output

app.py

@@ -34,6 +34,16 @@ st.markdown(
     unsafe_allow_html=True
 )
 
+st.text_input('Ask me a question', key='user_input', on_change=handle_submit)
+col1, col2 = st.columns(2)
+
+with col1: 
+    if st.button("Ask"): 
+        handle_submit()
+
+with col2:
+    st.session_state.include_context = st.checkbox('Search in Help')
+
 def handle_submit():
     user_input = st.session_state.user_input
 
@@ -70,17 +80,7 @@ def handle_submit():
 
 #st.title("Adrega AI Help")
 
-st.text_input('Ask me a question', key='user_input', on_change=handle_submit)
-col1, col2 = st.columns(2)
-
-with col1: 
-    if st.button("Ask"): 
-        handle_submit()
-
 if 'conversation' not in st.session_state:
     st.session_state.conversation = ""
-
-with col2:
-    st.session_state.include_context = st.checkbox('Search in Help')
   
 #st.markdown(f'<div class="scrollable-div">{st.session_state.conversation}</div>', unsafe_allow_html=True)