diff --git a/app/api/api_brighten.py b/app/api/api_brighten.py index c06acd9..cc5a03f 100644 --- a/app/api/api_brighten.py +++ b/app/api/api_brighten.py @@ -26,15 +26,11 @@ async def brighten(request_item: BrightenModel): 创建一个具有以下参数的请求体: - **image_url**: 提亮图片url - **brighten_value**: 提高亮度的比重 亮度因子 1.0 表示原始亮度,1.5 表示增加 50% 的亮度 - - **brighten_bucket**: 结果图保存的bucket - - **brighten_name**: 结果图保存的object name 示例参数: { "image_url": "aida-users/89/relight_image/3850e17b-3efd-4597-90ef-2a7bcd1a1a0b-0-89.png", - "brighten_value": 1.5, - "brighten_bucket": "test", - "brighten_name": "12341556-89.jpg" + "brighten_value": 1.5 } """ try: @@ -46,7 +42,7 @@ async def brighten(request_item: BrightenModel): new_image.save(image_data, format='PNG') image_data.seek(0) image_bytes = image_data.read() - req = oss_upload_image(bucket=request_item.brighten_bucket, object_name=request_item.brighten_name, image_bytes=image_bytes) + req = oss_upload_image(bucket=request_item.image_url.split('/')[0], object_name=request_item.image_url[request_item.image_url.find('/') + 1:], image_bytes=image_bytes) brighten_url = f"{req.bucket_name}/{req.object_name}" logger.info(f"run time is : {time.time() - start_time}") except Exception as e: @@ -57,9 +53,7 @@ async def brighten(request_item: BrightenModel): if __name__ == '__main__': request_item = BrightenModel(image_url="aida-users/89/relight_image/3850e17b-3efd-4597-90ef-2a7bcd1a1a0b-0-89.png", - brighten_value=1.5, - brighten_bucket="test", - brighten_name="12341556-89.png") + brighten_value=1.5) image = oss_get_image(bucket=request_item.image_url.split('/')[0], object_name=request_item.image_url[request_item.image_url.find('/') + 1:], data_type="PIL") new_image = increase_brightness(image, request_item.brighten_value) new_image.show() diff --git a/app/schemas/brighten.py b/app/schemas/brighten.py index c25c6d0..e407905 100644 --- a/app/schemas/brighten.py +++ b/app/schemas/brighten.py @@ -4,5 +4,3 @@ from pydantic import BaseModel class BrightenModel(BaseModel): image_url: str brighten_value: float - brighten_bucket: str - brighten_name: str