Updated the trigger toggle and minor changes to the Readme file

Updated the trigger toggle and minor changes to the Readme file

Author: hi4rahul

README.md

@@ -77,14 +77,14 @@ The demo application is deployed as an [AWS CloudFormation](https://aws.amazon.c
 #### Accessing the Application
 
 After deployment, the application can be used by configuring the following services.
-1. The event that triggers the step Machine is defaulted to run every hour and is set to "Enabled" at the start. This event can be modified and set to "Disabled" if the use case does not need to trigger the step machine automatically. The default hourly run schedule can also be modified as per use case. When modifying the run schedule, a key point to bear in mind will be to check the pricing for Amazon Rekognition Custom Labels - which is currently priced per hour.
+1. The event that triggers the Step Machine is defaulted to run every hour and is set to "Enabled" at the start. This event can be modified and set to "Disabled" if the use case does not need to trigger the step machine automatically. The default hourly run schedule can also be modified as per use case. When modifying the run schedule, a key point to bear in mind will be to check the pricing for Amazon Rekognition Custom Labels - which is currently priced per hour. The step machine is configured to shut down the model after 1 hour by default.
 2. The Source S3 bucket should be used as the destination bucket for all the images to be used for inferencing. 
 3. The Final S3 bucket holds the images that have been processed along with the inferenced custom label jsons.
 
 
 ### Remove the application
 
-To remove the application open the AWS CloudFormation Console, click the PPE Demo project, right-click and select "_Delete Stack_". Your stack will take some time to be deleted. You can track its progress in the "Events" tab. When it is done, the status will change from "_DELETE_IN_PROGRESS_" to "_DELETE_COMPLETE_". It will then disappear from the list.
+To remove the application open the AWS CloudFormation Console, click on the name of the project, right-click and select "_Delete Stack_". Your stack will take some time to be deleted. You can track its progress in the "Events" tab. When it is done, the status will change from "_DELETE_IN_PROGRESS_" to "_DELETE_COMPLETE_". It will then disappear from the list. Please note that the Amazon S3 buckets will only be deleted if they are empty.
 
 
 ## License

functions/toggle_trigger/app.py

@@ -43,26 +43,28 @@ def lambda_handler(event, context):
             print(e)
 
     # State (string) -- The state of the event source mapping. It can be one of the following: Creating , Enabling , Enabled , Disabling , Disabled , Updating , or Deleting .
-    running_states = ["Enabling", "Enabled"]
+    running_states = ["Enabling", "Enabled", "Disabled", "Disabling"]
     if response['State'] in running_states:
         # Disable
-        try:
-            response = lambda_client.update_event_source_mapping(
-            UUID = analyse_lambda_uuid,
-                Enabled = False
-            )
-        except Exception as e:
-            print(e)
-    else:
-        # Enable
-        try:
-            response = lambda_client.update_event_source_mapping(
+        if (event[0]['Action'] == 'disable'): 
+            try:
+                response = lambda_client.update_event_source_mapping(
                 UUID = analyse_lambda_uuid,
-                Enabled = True
-            )
-        except Exception as e:
-            print(e)
+                    Enabled = False
+                )
+            except Exception as e:
+                print(e)
+        else:
+            # Enable
+            if (event[0]['Action'] == 'enable'): 
+                try:
+                    response = lambda_client.update_event_source_mapping(
+                        UUID = analyse_lambda_uuid,
+                        Enabled = True
+                    )
+                except Exception as e:
+                    print(e)
+    else:
+        print("Current state is:", response['State'])
 
     return response['State']
-    # return {'status': '200'}
-

template.yaml

@@ -12,7 +12,7 @@ Parameters:
 Mappings:
   Solution:
     Constants:
-      Version: 'v0.2'
+      Version: 'v0.3'
 
 Resources:
   SourceS3Bucket:
@@ -107,10 +107,14 @@ Resources:
             Next: Disable SQS Trigger
           Enable SQS Trigger:
             Type: Task
+            Parameters: 
+              - Action: enable
             Resource: !GetAtt ToggleTriggerFunction.Arn
             Next: Keep Model running for 1 hr
           Disable SQS Trigger:
             Type: Task
+            Parameters: 
+              - Action: disable
             Resource: !GetAtt ToggleTriggerFunction.Arn
             Next: Stop Model
           Stop Model:
@@ -122,7 +126,7 @@ Resources:
           Type: Schedule # More info about Schedule Event Source: https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-property-statemachine-schedule.html
           Properties:
             Description: Schedule to run the state machine every hour
-            Enabled: True # This schedule can be disabled based on the use case to avoid incurring charges.
+            Enabled: False # This schedule can be disabled based on the use case to avoid incurring charges.
             Schedule: "rate(1 hour)"
       Policies: # Find out more about SAM policy templates: https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/serverless-policy-templates.html
         - LambdaInvokePolicy: