Update app.py

app.py

@@ -8,13 +8,68 @@ import time
 import gradio as gr
 import requests
 
+from pathlib import Path
+from datetime import datetime, timedelta
+
 import dashscope
-from dashscope.utils.oss_utils import check_and_upload_local
+# from dashscope.utils.oss_utils import check_and_upload_local
 
 DASHSCOPE_API_KEY = os.getenv("DASHSCOPE_API_KEY")
 dashscope.api_key = DASHSCOPE_API_KEY
 
 
+
+def get_upload_policy(api_key, model_name):
+    """获取文件上传凭证"""
+    url = "https://dashscope.aliyuncs.com/api/v1/uploads"
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    params = {
+        "action": "getPolicy",
+        "model": model_name
+    }
+    
+    response = requests.get(url, headers=headers, params=params)
+    if response.status_code != 200:
+        raise Exception(f"Failed to get upload policy: {response.text}")
+    
+    return response.json()['data']
+
+def upload_file_to_oss(policy_data, file_path):
+    """将文件上传到临时存储OSS"""
+    file_name = Path(file_path).name
+    key = f"{policy_data['upload_dir']}/{file_name}"
+    
+    with open(file_path, 'rb') as file:
+        files = {
+            'OSSAccessKeyId': (None, policy_data['oss_access_key_id']),
+            'Signature': (None, policy_data['signature']),
+            'policy': (None, policy_data['policy']),
+            'x-oss-object-acl': (None, policy_data['x_oss_object_acl']),
+            'x-oss-forbid-overwrite': (None, policy_data['x_oss_forbid_overwrite']),
+            'key': (None, key),
+            'success_action_status': (None, '200'),
+            'file': (file_name, file)
+        }
+        
+        response = requests.post(policy_data['upload_host'], files=files)
+        if response.status_code != 200:
+            raise Exception(f"Failed to upload file: {response.text}")
+    
+    return f"oss://{key}"
+
+def upload_file_and_get_url(api_key, model_name, file_path):
+    """上传文件并获取URL"""
+    # 1. 获取上传凭证，上传凭证接口有限流，超出限流将导致请求失败
+    policy_data = get_upload_policy(api_key, model_name) 
+    # 2. 上传文件到OSS
+    oss_url = upload_file_to_oss(policy_data, file_path)
+    
+    return oss_url
+
+
 class WanAnimateApp:
     def __init__(self, url, get_url):
         self.url = url
@@ -28,8 +83,8 @@ class WanAnimateApp:
         model,
     ):
         # Upload files to OSS if needed and get URLs
-        _, image_url = check_and_upload_local(model_id, ref_img, DASHSCOPE_API_KEY)
-        _, video_url = check_and_upload_local(model_id, video, DASHSCOPE_API_KEY)
+        image_url = upload_file_and_get_url(DASHSCOPE_API_KEY, model_id, ref_img)
+        video_url = upload_file_and_get_url(DASHSCOPE_API_KEY, model_id, video)
 
         # Prepare the request payload
         payload = {