Update app.py

app.py

@@ -17,6 +17,10 @@ from streamlit_image_select import image_select
 if "key" not in st.session_state:
     st.session_state["key"] = uuid.uuid4().hex
 
+
+if "captcha" not in st.session_state:
+    st.session_state["captcha"] = None
+
 if "model_inputs" not in st.session_state:
     st.session_state["model_inputs"] = None
 
@@ -179,6 +183,10 @@ with train.form("training"):
             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: