Commit
•
6974862
1
Parent(s):
8266cc1
Adicionando mensagem amigável
Browse files- app.py +1 -1
- services/aws_service.py +5 -1
app.py
CHANGED
@@ -27,7 +27,7 @@ def crop_photoshoot_images(photo_shoot_id):
|
|
27 |
os.remove(filename)
|
28 |
i += 1
|
29 |
|
30 |
-
return {"message": "
|
31 |
|
32 |
iface = gr.Interface(
|
33 |
fn=crop_photoshoot_images,
|
|
|
27 |
os.remove(filename)
|
28 |
i += 1
|
29 |
|
30 |
+
return {"message": f"{len(files)} images cropped successfully"}
|
31 |
|
32 |
iface = gr.Interface(
|
33 |
fn=crop_photoshoot_images,
|
services/aws_service.py
CHANGED
@@ -15,7 +15,11 @@ class AwsService:
|
|
15 |
return AwsService.session().client('s3')
|
16 |
|
17 |
def get_files_from_s3(bucket, prefix):
|
18 |
-
|
|
|
|
|
|
|
|
|
19 |
|
20 |
def get_image_from_s3(bucket, key):
|
21 |
file_byte_string = AwsService.s3_client().get_object(Bucket=bucket, Key=key)['Body'].read()
|
|
|
15 |
return AwsService.session().client('s3')
|
16 |
|
17 |
def get_files_from_s3(bucket, prefix):
|
18 |
+
results = AwsService.s3_client().list_objects(Bucket=bucket, Prefix=prefix)
|
19 |
+
if 'Contents' in results:
|
20 |
+
return results['Contents']
|
21 |
+
else:
|
22 |
+
return []
|
23 |
|
24 |
def get_image_from_s3(bucket, key):
|
25 |
file_byte_string = AwsService.s3_client().get_object(Bucket=bucket, Key=key)['Body'].read()
|