Try to realign elements to take less vertical space.

app.py

@@ -35,11 +35,21 @@ st.markdown(
 )
 
 #st.title("Adrega AI Help")
-st.session_state.include_context = st.checkbox('Search in Help')
+col1, col2, col3 = st.columns([3, 1, 1])
+
+with col1:
+    st.text_input('Ask me a question', key='user_input', on_change=handle_submit)
+
+with col2: 
+    st.session_state.include_context = st.checkbox('Search in Help')
 
 if 'conversation' not in st.session_state:
     st.session_state.conversation = ""
 
+with col3:
+    if st.button("Ask"): 
+        handle_submit()
+
 def handle_submit():
     user_input = st.session_state.user_input
     if user_input:
@@ -66,10 +76,5 @@ def handle_submit():
         st.session_state.conversation = f"<p><strong>User:</strong> {user_input}</p><p><strong>Adrega AI:</strong> {answer}</p>" + st.session_state.conversation
     else:
         st.write("Please enter a question.")
-
-
-st.text_input('Ask me a question', key='user_input', on_change=handle_submit)
-if st.button("Ask"): 
-    handle_submit()
     
 st.markdown(f'<div class="scrollable-div">{st.session_state.conversation}</div>', unsafe_allow_html=True)