Update app.py

app.py

@@ -4,6 +4,7 @@ import oss2
 import os
 import gradio as gr
 import uuid
+import json
 import base64
 from io import BytesIO
 from oss2.credentials import EnvironmentVariableCredentialsProvider,StaticCredentialsProvider
@@ -54,8 +55,21 @@ def delete_local_file(file_path):
         print(f"{file_path} 文件不存在。")
 
 
-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):
-    
+def run(params):
+    params = json.loads(params)
+    url = params['url']
+    version_id = params['id']
+    access_key_id = params['access_key_id']
+    access_key_secret = params['access_key_secret']
+    securityToken = params['securityToken']
+    endpoint = params['endpoint']
+    bucket_name = params['bucket_name']
+    upload_filename = params['upload_filename']
+    callback_url = params['callback_url']
+    callback_header_key = params['callback_header_key']
+    callback_header_secret = params['callback_header_secret']
+    model_type = params['model_type']
+    # 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()
     result_image = f'{tmp_file_name}_compressed_image.jpg'
@@ -75,6 +89,8 @@ def run(url,version_id,access_key_id,access_key_secret,securityToken,endpoint,bu
             if oss_url:
                 res = requests.post(callback_url,json={
                     'version_id': version_id,
+                    'model_type': model_type,
+                    'filename': upload_filename
                 },headers={
                     f"{callback_header_key}": callback_header_secret
                 })
@@ -96,63 +112,14 @@ with gr.Blocks() as demo:
     
     with gr.Column(elem_id="col-container"):
         with gr.Row():
-            url = gr.Text(
+            params = gr.Text(
                 label="图片地址",
                 show_label=False,
                 max_lines=1,
                 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",
-                placeholder="云存储 AccessKeyId",
-            ) 
-        with gr.Row():
-            access_key_secret = gr.Textbox(
-                label="AccessKeySecret",
-                placeholder="云存储 AccessKeySecret",
-            ) 
-        with gr.Row():
-            securityToken = gr.Textbox(
-                label="SecurityToken",
-                placeholder="云存储 securityToken",
-            ) 
-        with gr.Row():
-            endpoint = gr.Textbox(
-                label="Endpoint",
-                placeholder="云存储 endpoint",
-            ) 
-        with gr.Row():
-            bucket_name = gr.Textbox(
-                label="BucketName",
-                placeholder="云存储 bucket_name",
-            ) 
-        with gr.Row():
-            upload_filename = gr.Textbox(
-                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(
@@ -168,19 +135,7 @@ with gr.Blocks() as demo:
     gr.on(
         triggers=[run_button.click],
         fn = run,
-        inputs = [
-            url,
-            version_id,
-            access_key_id,
-            access_key_secret,
-            securityToken,
-            endpoint,
-            bucket_name,
-            upload_filename,
-            callback_url,
-            callback_header_key,
-            callback_header_secret
-        ],
+        inputs = [params],
         outputs = [output]
     )