Update app.py

app.py

@@ -22,6 +22,9 @@ if "key" not in st.session_state:
 if "captcha" not in st.session_state:
     st.session_state["captcha"] = None
 
+if "captcha_bool" not in st.session_state:
+    st.session_state["captcha_bool"] = False
+    
 if "model_inputs" not in st.session_state:
     st.session_state["model_inputs"] = None
 
@@ -183,22 +186,29 @@ with train.form("training"):
         if not email:
             st.markdown('Please input an email!')
         else:
-            result = generate_captcha()
-            st.session_state['captcha'] = result
-            captcha_id = st.session_state['captcha']['uuid']
-            captcha_image = st.session_state['captcha']['captcha']
-
-            st.image(captcha_image, width=300)
-            captcha_response = st.text_input("Enter the captcha response")
-            if captcha_response:
-                result = verify_captcha(captcha_id, captcha_response)
-                if 'message' in result and result['message'] == 'CAPTCHA_SOLVED':
-                    with st.spinner("Model Fine Tuning..."):
-                        st.session_state["model_inputs"]["identifier"] = st.session_state["key"]
-                        st.session_state["model_inputs"]["email"] = email
-                        train_model(st.session_state["model_inputs"])
-                        st.session_state["train_view"] = True
-                else:
-                    st.error(result['error'])
-            st.success('Done!')
+            st.session_state["captcha_bool"] = True
+
+
+if st.session_state["captcha_bool"] = True:
+    captcha_form = st.empty()
+    with captcha_form.form("captcha"):
+        result = generate_captcha()
+        st.session_state['captcha'] = result
+        captcha_id = st.session_state['captcha']['uuid']
+        captcha_image = st.session_state['captcha']['captcha']
+
+        st.image(captcha_image, width=300)
+        captcha_response = st.text_input("Enter the captcha response")
+        submitted = st.form_submit_button("Submit Captcha!")
+        if submitted:
+            result = verify_captcha(captcha_id, captcha_response)
+            if 'message' in result and result['message'] == 'CAPTCHA_SOLVED':
+                with st.spinner("Model Fine Tuning..."):
+                    st.session_state["model_inputs"]["identifier"] = st.session_state["key"]
+                    st.session_state["model_inputs"]["email"] = email
+                    train_model(st.session_state["model_inputs"])
+                    st.session_state["train_view"] = True
+            else:
+                st.error(result['error'])
+        st.success('Done!')