Update app.py

app.py

@@ -131,6 +131,23 @@ def upload_image_to_r2(image, account_id, access_key, secret_key, bucket_name):
         buffer.seek(0)
         s3.upload_fileobj(buffer, bucket_name, image_file)
         print("upload finish", image_file)
+
+        # start to generate thumbnail
+        thumbnail = image.copy()
+        thumbnail_width = 256
+        aspect_ratio = image.height / image.width
+        thumbnail_height = int(thumbnail_width * aspect_ratio)
+        thumbnail = thumbnail.resize((thumbnail_width, thumbnail_height), Image.LANCZOS)
+        
+        # Generate the thumbnail image filename
+        thumbnail_file = image_file.replace(".png", "_thumbnail.png")
+        
+        # Save thumbnail to buffer and upload
+        thumbnail_buffer = BytesIO()
+        thumbnail.save(thumbnail_buffer, "PNG")
+        thumbnail_buffer.seek(0)
+        s3.upload_fileobj(thumbnail_buffer, bucket_name, thumbnail_file)
+        print("upload thumbnail finish", thumbnail_file)
         
     return image_file