adding new changes for generation and permisions

generate.py

@@ -21,9 +21,13 @@ pipe = StableDiffusionPipeline.from_pretrained(
 
 pipe = pipe.to(device)
 
-s3 = boto3.resource('s3')
+s3 = boto3.resource('s3',
+                    aws_access_key_id=AWS_ACCESS_KEY_ID,
+                    aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
 
-s3_client = boto3.client('s3')
+s3_client = boto3.client('s3',
+                    aws_access_key_id=AWS_ACCESS_KEY_ID,
+                    aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
 
 
 def text_to_image(prompt, save_as, key_id):
@@ -46,13 +50,21 @@ def text_to_image(prompt, save_as, key_id):
 
         # Full path of the file in the bucket
         s3_key = "public/" + image_name
+        print('Saving image to s3')
 
         # Upload the image to the S3 bucket
         s3.upload_fileobj(image_buffer, S3_BUCKET_NAME, s3_key)
+        print('Image saved to s3')
 
     def generator_image(prompt):
         prompt = prompt
-        image = pipe(prompt).images[0]
+        print('Starting to generate the image ...')
+        try:
+            image = pipe(prompt).images[0]
+        except Exception as e:
+            print('Error: ', e)
+
+        print('Image generation completed')
 
         # Save the image in S3
         save_image_to_s3(image)