Update app.py

app.py

@@ -18,12 +18,15 @@ pipe = StableDiffusionPipeline.from_pretrained(
 pipe = pipe.to(device)
 
 def text_to_image(summary, image_name):
+    
     # Crea una instancia del cliente de S3
     s3 = boto3.client('s3',
                   aws_access_key_id=AWS_ACCESS_KEY_ID,
                   aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
 
-    def save_image_to_s3(image, image_name):
+    image_name = '-'.join(image_name.split()) + ".webp"
+    
+    def save_image_to_s3(image):
         # Crea un objeto de BytesIO para almacenar la imagen
         image_buffer = BytesIO()
         image.save(image_buffer, format='WEBP')
@@ -35,18 +38,17 @@ def text_to_image(summary, image_name):
         # Sube la imagen al bucket de S3
         s3.upload_fileobj(image_buffer, S3_BUCKET_NAME, s3_key)
     
-    def generator_image(summary, image_name):
+    def generator_image(summary):
         prompt = summary
         image = pipe(prompt).images[0]
         
         # Guarda la imagen en S3
-        save_image_to_s3(image, image_name)
-    
-        return image_name
+        save_image_to_s3(image)
     
     # generate image
-    image_name = '-'.join(image_name.split()) + ".webp"
-    return generator_image(summary, image_name)
+    generator_image(summary)
+    
+    return image_name