Update app.py

app.py

@@ -7,6 +7,7 @@ import uuid
 import json
 import base64
 from io import BytesIO
+from datetime import datetime
 from oss2.credentials import EnvironmentVariableCredentialsProvider,StaticCredentialsProvider
 
 # 下载网络图片
@@ -18,6 +19,14 @@ def download_image(url):
     except requests.exceptions.RequestException as e:
         raise RuntimeError(f"图片下载失败: {e}")
 
+
+
+def formatNow():
+    now = datetime.now()
+    return now.strftime("%Y-%m-%d %H:%M:%S")
+
+
+
 def remove_metadata(image, output_path=None):
     # 获取图片的格式
     img_format = image.format.lower()
@@ -87,6 +96,8 @@ def run(params):
     model_type = params['model_type']
     compress = params.get('compress',False)
     quality = params.get('quality',85)
+    env_text = params.get('env','-')
+    logPrefix = f"{formatNow()}-{env_text}-"
     # 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()
@@ -100,9 +111,9 @@ def run(params):
             output_image_path = compress_image(downloaded_image,result_image)
         else:
             output_image_path = remove_metadata(downloaded_image,result_image)
-            print("元数据移除完成")
+            print(f"{logPrefix}元数据移除完成")
         if access_key_id and endpoint and bucket_name:
-            print('上传OSS')
+            print(f'{logPrefix}上传OSS')
             # 步骤 3：上传到阿里云 OSS
             oss_url = upload_to_oss(output_image_path, upload_filename, bucket_name, access_key_id, access_key_secret,securityToken, endpoint)
         
@@ -117,14 +128,14 @@ def run(params):
                     f"{callback_header_key}": callback_header_secret
                 })
                 print(res.status_code)
-                return f"图片上传成功，URL：{oss_url}"
+                return f"{logPrefix}图片上传成功，URL：{oss_url}"
             else:
-                return "上传到 OSS 失败"
+                return f"{logPrefix}上传到 OSS 失败"
         else:
-            print('不上传OSS')
+            print(f'{logPrefix}不上传OSS')
             with open(output_image_path,'rb') as f:
                 image_base64 = base64.b64encode(f.read())
-            print("运行完成")
+            print(f"{logPrefix}运行完成")
             delete_local_file(output_image_path)  # 删除压缩后的图片
             return image_base64
     else: