Merge pull request #4 from marlowetutor/refactor/DEVEVANG-100-python-…

…rewrite Refactor/devevang 100 python rewrite
GenesysCloudBlueprints · Oct 3, 2024 · 3143da2 · 3143da2
2 parents 3c4a412 + 25fd35e
commit 3143da2
Show file tree

Hide file tree

Showing 31 changed files with 367 additions and 20,114 deletions.
diff --git a/blueprint/aws-classifier-lambda/package-lock.json b/blueprint/aws-classifier-lambda/package-lock.json
diff --git a/blueprint/aws-classifier-lambda/package.json b/blueprint/aws-classifier-lambda/package.json
diff --git a/blueprint/aws-classifier-lambda/serverless.ts b/blueprint/aws-classifier-lambda/serverless.ts
diff --git a/blueprint/aws-classifier-lambda/src/functions/emailresolver/handler.ts b/blueprint/aws-classifier-lambda/src/functions/emailresolver/handler.ts
diff --git a/blueprint/aws-classifier-lambda/src/functions/emailresolver/index.ts b/blueprint/aws-classifier-lambda/src/functions/emailresolver/index.ts
diff --git a/blueprint/aws-classifier-lambda/src/functions/emailresolver/mock_401K.json b/blueprint/aws-classifier-lambda/src/functions/emailresolver/mock_401K.json
diff --git a/blueprint/aws-classifier-lambda/src/functions/emailresolver/mock_529.json b/blueprint/aws-classifier-lambda/src/functions/emailresolver/mock_529.json
diff --git a/blueprint/aws-classifier-lambda/src/functions/emailresolver/mock_IRA.json b/blueprint/aws-classifier-lambda/src/functions/emailresolver/mock_IRA.json
diff --git a/blueprint/aws-classifier-lambda/src/functions/emailresolver/schema.ts b/blueprint/aws-classifier-lambda/src/functions/emailresolver/schema.ts
diff --git a/blueprint/aws-classifier-lambda/src/functions/index.ts b/blueprint/aws-classifier-lambda/src/functions/index.ts
diff --git a/blueprint/aws-classifier-lambda/src/libs/apiGateway.ts b/blueprint/aws-classifier-lambda/src/libs/apiGateway.ts
diff --git a/blueprint/aws-classifier-lambda/src/libs/handlerResolver.ts b/blueprint/aws-classifier-lambda/src/libs/handlerResolver.ts
diff --git a/blueprint/aws-classifier-lambda/src/libs/lambda.ts b/blueprint/aws-classifier-lambda/src/libs/lambda.ts
diff --git a/blueprint/aws-classifier-lambda/tsconfig.json b/blueprint/aws-classifier-lambda/tsconfig.json
diff --git a/blueprint/aws-classifier-lambda/tsconfig.paths.json b/blueprint/aws-classifier-lambda/tsconfig.paths.json
diff --git a/blueprint/aws-classifier-lambda/webpack.config.js b/blueprint/aws-classifier-lambda/webpack.config.js
diff --git a/blueprint/aws-comprehend/EmailClassifierBucketAccessRole-Permissions.json b/blueprint/aws-comprehend/EmailClassifierBucketAccessRole-Permissions.json
@@ -16,7 +16,7 @@
         "s3:ListBucket"
       ],
       "Resource": [
-        "arn:aws:s3:::nintca-test-bucket"
+        "arn:aws:s3:::inintca-test-bucket"
       ],
       "Effect": "Allow"
     }

diff --git a/blueprint/aws-sam-email-classifier/__init__.py b/blueprint/aws-sam-email-classifier/__init__.py
diff --git a/blueprint/aws-sam-email-classifier/emailclassifier/app.py b/blueprint/aws-sam-email-classifier/emailclassifier/app.py
@@ -0,0 +1,56 @@
+import json
+import boto3
+import logging
+import os
+
+logger = logging.getLogger()
+logger.setLevel("INFO")
+
+def lambda_handler(event, context):
+
+    try:
+        account_id = os.getenv('AWS_ACCOUNT_ID')
+        region = os.getenv('AWS_REGION')
+        confidence_threshold = os.getenv('COMPREHEND_CONFIDENCE_THRESHOLD')
+        if confidence_threshold == None:
+            raise Exception("Please set the confidence threshold in the environment variables.")
+
+        request_body_string = event.get("body")
+        if request_body_string == None:
+            raise Exception("Please use the correct format for the request body.")
+
+        comprehend_object = boto3.client("comprehend")
+        endpoint_arn = f'arn:aws:comprehend:{region}:{account_id}:document-classifier-endpoint/emailclassifier'
+        response_object = comprehend_object.classify_document(   
+            Text=request_body_string,
+            EndpointArn=endpoint_arn
+        )
+
+        logger.info(json.dumps(response_object))
+        classes_list = response_object.get("Classes")
+
+        if (len(classes_list) > 0):
+            # get the first item with the highest confidence/score
+            queue_name = classes_list[0].get("Name")
+            confidence = classes_list[0].get("Score")
+
+            # confidence threshold reached
+            if (float(confidence) > float(confidence_threshold)):
+                response_body = {"QueueName": queue_name , "Confidence": confidence}
+
+            # confidence threshold not reached, then an empty string is returned.
+            if (float(confidence) < float(confidence_threshold)):
+                response_body = {"Message": "Classification results did not meet the confidence threshold."}
+
+        return {
+            "statusCode": 200,
+            "body": json.dumps(response_body),
+        }
+
+    except Exception as err:
+        e = f'Error: {str(err)}'
+        logger.error(e)
+        return {
+            "statusCode": 502,
+            "body": json.dumps({"Message":e}),
+        }
diff --git a/blueprint/aws-sam-email-classifier/emailclassifier/requirements.txt b/blueprint/aws-sam-email-classifier/emailclassifier/requirements.txt
@@ -0,0 +1,2 @@
+requests
+boto3