Update main.py

main.py

@@ -18,11 +18,69 @@ app = FastAPI()
 #router = APIRouter()
 templates = Jinja2Templates(directory="templates")
 
-# Include the authentication router with the prefix '/auth'
-#app.include_router(auth.router, prefix="")
+from google.cloud import recaptchaenterprise_v1
+from google.cloud.recaptchaenterprise_v1 import Assessment
 
-# Include the TTS router with the prefix '/tts'
-#app.include_router(tts.router, prefix="/tts")
+def create_assessment(
+    project_id: str, recaptcha_key: str, token: str, recaptcha_action: str
+) -> Assessment:
+    """Create an assessment to analyse the risk of a UI action.
+    Args:
+        project_id: Your Google Cloud project ID.
+        recaptcha_key: The reCAPTCHA key associated with the site/app
+        token: The generated token obtained from the client.
+        recaptcha_action: Action name corresponding to the token.
+    """
+
+    client = recaptchaenterprise_v1.RecaptchaEnterpriseServiceClient()
+
+    # Set the properties of the event to be tracked.
+    event = recaptchaenterprise_v1.Event()
+    event.site_key = recaptcha_key
+    event.token = token
+
+    assessment = recaptchaenterprise_v1.Assessment()
+    assessment.event = event
+
+    project_name = f"projects/{project_id}"
+
+    # Build the assessment request.
+    request = recaptchaenterprise_v1.CreateAssessmentRequest()
+    request.assessment = assessment
+    request.parent = project_name
+
+    response = client.create_assessment(request)
+
+    # Check if the token is valid.
+    if not response.token_properties.valid:
+        print(
+            "The CreateAssessment call failed because the token was "
+            + "invalid for the following reasons: "
+            + str(response.token_properties.invalid_reason)
+        )
+        return
+
+    # Check if the expected action was executed.
+    if response.token_properties.action != recaptcha_action:
+        print(
+            "The action attribute in your reCAPTCHA tag does"
+            + "not match the action you are expecting to score"
+        )
+        return
+    else:
+        # Get the risk score and the reason(s).
+        # For more information on interpreting the assessment, see:
+        # https://cloud.google.com/recaptcha-enterprise/docs/interpret-assessment
+        for reason in response.risk_analysis.reasons:
+            print(reason)
+        print(
+            "The reCAPTCHA score for this token is: "
+            + str(response.risk_analysis.score)
+        )
+        # Get the assessment name (ID). Use this to annotate the assessment.
+        assessment_name = client.parse_assessment_path(response.name).get("assessment")
+        print(f"Assessment name: {assessment_name}")
+    return response
 
 # Dependency for verifying the user's token
 def get_current_user(token: str = Depends(verify_token)):
@@ -90,8 +148,26 @@ async def register_post(
     email: str = Form(...),
     password: str = Form(...),
     confirm_password: str = Form(...),
+    recaptcha_token: str = Form(...),  # Add this line to accept the reCAPTCHA token
     db: Session = Depends(get_db)
 ):
+    # Perform reCAPTCHA verification first
+    project_id = 'Loginauthc'  # Replace with your project ID
+    recaptcha_key = '6LdaUQIpAAAAACQFcOxakEVXK9QHpaYbic6IClNO'  # Replace with your site key
+    recaptcha_action = 'submit'  # The action you're expecting
+    
+    # Call the create_assessment function to validate the token
+    assessment = await create_assessment(
+        project_id, recaptcha_key, recaptcha_token, recaptcha_action
+    )
+    
+    # Check the assessment result
+    if not assessment or assessment.risk_analysis.score < 0.5:  # Use an appropriate risk score threshold
+        return templates.TemplateResponse("register.html", {
+            "request": request,
+            "error_message": "Captcha validation failed."
+        })
+ 
     if password != confirm_password:
         # Return to the registration page with an error
         return templates.TemplateResponse("register.html", {