matheus-erthal commited on
Commit
6974862
1 Parent(s): 8266cc1

Adicionando mensagem amigável

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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": "Images cropped successfully"}
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
- return AwsService.s3_client().list_objects(Bucket=bucket, Prefix=prefix)['Contents']
 
 
 
 
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()