Update app.py

app.py

@@ -54,7 +54,7 @@ def delete_local_file(file_path):
         print(f"{file_path} 文件不存在。")
 
 
-def run(url,access_key_id,access_key_secret,securityToken,endpoint,bucket_name,upload_filename=uuid.uuid4()):
+def run(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()
@@ -73,6 +73,12 @@ def run(url,access_key_id,access_key_secret,securityToken,endpoint,bucket_name,u
             # 步骤 4：删除本地的临时文件
             delete_local_file(result_image)  # 删除压缩后的图片
             if oss_url:
+                res = requests.post(callback_url,json={
+                    'version_id': version_id,
+                },headers={
+                    f"{callback_header_key}": callback_header_secret
+                })
+                print(res.status_code)
                 return f"图片上传成功，URL：{oss_url}"
             else:
                 return "上传到 OSS 失败"
@@ -97,6 +103,11 @@ with gr.Blocks() as demo:
                 placeholder="input image url",
                 container=False,
             ) 
+        with gr.Row():
+            version_id = gr.Textbox(
+                label="AccessKeyId",
+                placeholder="云存储 AccessKeyId",
+            ) 
         with gr.Row():
             access_key_id = gr.Textbox(
                 label="AccessKeyId",
@@ -127,6 +138,21 @@ with gr.Blocks() as demo:
                 label="uploadFielName",
                 placeholder="云存储 文件名称",
             ) 
+        with gr.Row():
+            callback_url = gr.Textbox(
+                label="uploadFielName",
+                placeholder="云存储 文件名称",
+            ) 
+        with gr.Row():
+            callback_header_key = gr.Textbox(
+                label="uploadFielName",
+                placeholder="云存储 文件名称",
+            ) 
+        with gr.Row():
+            callback_header_secret = gr.Textbox(
+                label="uploadFielName",
+                placeholder="云存储 文件名称",
+            ) 
 
         with gr.Row():
             output = gr.Textbox(
@@ -144,12 +170,16 @@ with gr.Blocks() as demo:
         fn = run,
         inputs = [
             url,
+            version_id,
             access_key_id,
             access_key_secret,
             securityToken,
             endpoint,
             bucket_name,
-            upload_filename
+            upload_filename,
+            callback_url,
+            callback_header_key,
+            callback_header_secret
         ],
         outputs = [output]
     )