Update app.py

app.py

@@ -46,6 +46,50 @@ def upload_to_oss(file_path, object_name, bucket_name, access_key_id, access_key
         print(f"OSS上传失败: {e}")
         return None
 
+def compress_image(image, output_path, quality=85):
+    """
+    根据图片格式自动选择压缩方法并压缩图片，保留元数据。
+    
+    :param image: PIL Image 对象
+    :param output_path: 输出压缩后的图片文件路径
+    :param quality: 对于有损压缩的图片格式（JPEG, WebP）设置压缩质量，范围 1-100
+    """
+    # 获取图片的实际格式
+    img_format = image.format.lower()
+
+
+    if img_format == 'jpeg':
+        # 压缩 JPEG，同时保留 EXIF 数据（如果有）
+        image.save(output_path, "JPEG", quality=quality)
+        print(f"Compressed JPEG with quality={quality}, EXIF data preserved.")
+    
+    elif img_format == 'png':
+        # 压缩 PNG，optimize=True 启用无损压缩
+        image.save(output_path, "PNG", optimize=True)
+        print("Compressed PNG with lossless compression")
+    
+    elif img_format == 'gif':
+        # 压缩 GIF，使用调色板模式减少颜色
+        image = image.convert("P", palette=Image.ADAPTIVE)
+        image.save(output_path, optimize=True)
+        print("Compressed GIF with optimized color palette")
+    
+    elif img_format == 'tiff':
+        # 压缩 TIFF，使用 tiff_deflate 进行无损压缩
+        image.save(output_path, "TIFF", compression="tiff_deflate")
+        print("Compressed TIFF using deflate compression")
+    
+    elif img_format == 'webp':
+        # WebP 有损压缩
+        image.save(output_path, "WEBP", quality=quality)
+        print(f"Compressed WebP with quality={quality}")
+    
+    else:
+        raise Exception(f"Unsupported image format: {img_format}")
+        return
+    
+    print(f"Image saved to {output_path}")
+    
 # 删除本地文件
 def delete_local_file(file_path):
     if os.path.exists(file_path):
@@ -69,6 +113,8 @@ def run(params):
     callback_header_key = params['callback_header_key']
     callback_header_secret = params['callback_header_secret']
     model_type = params['model_type']
+    compress = params.get('compress',false)
+    quality = params.get('quality',85)
     # url,version_id,access_key_id,access_key_secret,securityToken,endpoint,bucket_name,upload_filename,callback_url,callback_header_key,callback_header_secret
     # 网络图片 URL 和本地保存路径
     tmp_file_name = uuid.uuid4()
@@ -78,8 +124,11 @@ def run(params):
     downloaded_image = download_image(url)
     
     if downloaded_image:
-        output_image_path = remove_metadata(downloaded_image,result_image)
-        print("元数据移除完成")
+        if compress:
+            output_image_path = compress_image(downloaded_image,result_image)
+        else:
+            output_image_path = remove_metadata(downloaded_image,result_image)
+            print("元数据移除完成")
         if access_key_id and endpoint and bucket_name:
             print('上传OSS')
             # 步骤 3：上传到阿里云 OSS