Try to fix problems with method not being initialized in time

app.py

@@ -34,22 +34,6 @@ st.markdown(
     unsafe_allow_html=True
 )
 
-#st.title("Adrega AI 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:
@@ -76,5 +60,23 @@ 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.title("Adrega AI 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()
+
+
     
 st.markdown(f'<div class="scrollable-div">{st.session_state.conversation}</div>', unsafe_allow_html=True)