Update app.py

app.py

@@ -50,6 +50,37 @@ def zip_and_upload_images(identifier, uploaded_files, image_type):
     )
     return f"s3://gretel-image-synthetics/data/{identifier}/{image_type}_images.zip"
 
+CAPTCHA_ENDPOINT = "https://captcha-api.akshit.me/v2/generate"
+VERIFY_ENDPOINT = "https://captcha-api.akshit.me/v2/verify"
+
+# Create a function to generate a captcha
+def generate_captcha():
+    # Make a GET request to the API endpoint to generate a captcha
+    response = requests.get(CAPTCHA_ENDPOINT)
+
+    # If the request was successful, return the API response
+    if response.status_code == 200:
+        return response.json()
+
+    # Otherwise, return an error message
+    return {"error": "Failed to generate captcha"}
+
+# Create a function to verify the captcha
+def verify_captcha(captcha_id, captcha_response):
+    # Make a POST request to the API endpoint with the captcha ID and response
+
+    response = requests.post(
+        VERIFY_ENDPOINT, json={"uuid": captcha_id, "captcha": captcha_response}
+    )
+    print(response)
+
+    # If the request was successful, return the API response
+    if response.status_code == 200:
+        return response.json()
+
+    # Otherwise, return an error message
+    return {"error": "Failed to verify captcha"}
+
 
 def train_model(model_inputs):
     api_key = "03cdd72e-5c04-4207-bd6a-fd5712c1740e"
@@ -147,10 +178,18 @@ with train.form("training"):
         if not email:
             st.markdown('Please input an email!')
         else:
-            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
+            result = generate_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!')