update api

caller.py

@@ -3,12 +3,17 @@
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 
-from PIL import Image, ImageFilter
-from loguru import logger
 import requests
 import base64
 import json
 import io
+import os
+import hashlib
+import hmac
+import time
+import random
+from PIL import Image, ImageFilter
+from loguru import logger
 
 # 接口 URL
 t2i_url = 'https://magicarena.bytedance.com/api/evaluate/v1/algo/process'
@@ -19,6 +24,31 @@ headers = {
     'X-USE-PPE': '1'
 }
 
+APP_KEY = os.environ['APP_KEY']
+SECRET_KEY = os.environ["SECRET_KEY"]
+
+def get_auth(app_key, secret_key):
+    # 生成随机数作为 nonce
+    nonce = str(random.randint(0, 2**31 - 1))
+    # 获取当前时间戳
+    timestamp = str(int(time.time()))
+    # 调用 get_sign 函数生成签名
+    sign = get_sign(nonce, timestamp, secret_key)
+    return {
+        "AppKey": app_key,
+        "Timestamp":timestamp,
+        "Nonce":nonce,
+        "Sign": sign}
+
+def get_sign(nonce, timestamp, secret_key):
+    keys = [nonce, secret_key, timestamp]
+    keys.sort()
+    key_str = ''.join(keys)
+    sha1_hash = hashlib.sha1()
+    sha1_hash.update(key_str.encode('utf-8'))
+    signature = sha1_hash.hexdigest()
+    return signature.lower()
+
 class SeedT2ICaller():
     def __init__(self, cfg, *args, **kwargs):
         self.cfg = cfg
@@ -34,6 +64,7 @@ class SeedT2ICaller():
                 # "width": 64,
                 # "height": 64
             })
+            authInfo = get_auth(APP_KEY, SECRET_KEY)
             logger.info(f"{req_json}")
             # 请求发送
             response = requests.post(
@@ -87,8 +118,8 @@ class SeedEditCaller():
                 data=json.dumps({
                     'AlgoType': 2,
                     'ReqJson': req_json,
-                    'BinaryData': [binary]
-                    # 'Base': base
+                    'BinaryData': [binary],
+                    'AuthInfo': get_auth(APP_KEY, SECRET_KEY)
                 })
             )
 

requirements.txt

@@ -1,5 +1,5 @@
-gradio
-numpy
-loguru
-Pillow
-requests
+gradio==4.44.1
+numpy==1.23.5
+loguru==0.7.0
+Pillow==9.4.0
+requests==2.32.3