init

Datas/UserRecord_9/Ips/127.0.0.1.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da8a6fd81a6423329221bade2789a60a1df18a1fd4d58c650bae257cd4104e06
+size 218

Datas/UserRecord_9/Tokens/f3beccd5-e00f-4af3-b442-efe7e8746899.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:35e0a937a52f61a660b50d2df8cd124d048507e6f08011cac3239733bde8ac8b
+size 561

Datas/task2/task2_showcase.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a78cd24de21a410869962ce968c79064faf2f669171d737ae6c7c3a827848998
+size 377845

Datas/task3/temps.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:406dbf7be2891d2c0d9d591f5b7825f7567fccf71f20d10540cdc68fc664e1a7
+size 2043353

app.py

@@ -43,13 +43,20 @@ def onClick(temp_image, user_image, caption_text, token_text,
     else:
         param4_text, param5_text = '', ''
 
+    if taskType in ['8', '9']:
+        if len(caption_text)==0:
+            yield None, "Please enter caption !!! "
+            return None, "Please enter caption !!! "
+        if taskType=='9':
+            temp_image, param4_text = '1536x1024.jpg', 'realistic_image'
+
     # print("======> temp_image ", type(temp_image))
     # print("======> user_image ", type(user_image))
     # print("======> caption_text ", type(caption_text))
     if temp_image is None: 
         yield None, "please choose a template!!!"
         return None, "please choose a template!!!"
-    if user_image is None: 
+    if user_image is None and taskType not in ['8', '9']: 
         yield None, "please upload a photo!!!"
         return None, "please upload a photo!!!"
 
@@ -68,7 +75,7 @@ def onClick(temp_image, user_image, caption_text, token_text,
 
         # 上传用户照片
         yield None, "start to upload, please wait..."
-        upload_url, uploadm_url = upload_user_img_mask(client_ip, user_image, user_mask)
+        upload_url, uploadm_url = upload_user_img_mask(client_ip, user_image, user_mask, taskType)
         if len(upload_url)==0:
             yield None, "fail to upload"
             return None, "fail to upload"
@@ -88,7 +95,7 @@ def onClick(temp_image, user_image, caption_text, token_text,
         # time.sleep(2)
         for i in range(max_try):
             time.sleep(wait_s)
-            taskStatus = getTaskRes(taskId)
+            taskStatus = getTaskRes(taskId, taskType)
             if taskStatus is None: continue
             user_recorder.save_record(taskStatus, ip=client_ip, token=token_text)
 
@@ -145,6 +152,8 @@ with gr.Blocks(css=css) as demo:
                     user_image = gr.ImageEditor(value=None, type="numpy", 
                         eraser=False, brush=brush ,layers=False, sources=['upload',],
                         transforms=[], label=UserLabel)
+                elif taskType in ['8', '9']:
+                    user_image = gr.Image(value=temp_examples[0][0], type="numpy", label=UserLabel, interactive=False)
                 else:
                     user_image = gr.Image(value=None, type="numpy", label=UserLabel)
                 param4_text = gr.Textbox(value="0.5", interactive=True, label=Param4Label, visible=Param4Visible)
@@ -177,7 +186,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             his_input3 = gr.HTML()
             his_output3 = gr.HTML()
-
+            
     # outputs_onload = [his_input1, his_output1, his_input2, his_output2, his_input3, his_output3,
     #     MK02, run_button]
 

config.py

@@ -41,7 +41,7 @@ if taskType=="3":
     Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br>
 
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br>
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
     # 组件信息
@@ -63,7 +63,7 @@ elif taskType=="4":
     Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br>
 
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br>
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
     # 组件信息
@@ -84,7 +84,7 @@ elif taskType=="6":
     Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br>
 
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br>
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
     # 组件信息
@@ -106,7 +106,7 @@ elif taskType=="5":
     Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br>
 
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br>
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
     # 组件信息
@@ -128,7 +128,7 @@ elif taskType=="1":
     Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br>
 
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br>
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
     # 组件信息
@@ -151,7 +151,7 @@ elif taskType=="2":
     Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br>
 
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br>
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
     # 组件信息
@@ -174,7 +174,7 @@ elif taskType=="7":
     Go to <a href='https://heybeauty.ai/extension' target='_blank'><b>HeyBeauty</b></a> for Faster and Free AI App! 🤗 .<br>
 
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/keys' target='_blank'><b>find your api key here</b></a> .<br>
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
     # 组件信息
@@ -184,10 +184,28 @@ elif taskType=="7":
     CaptionLabel = "English caption text (required)"
     CapVisible = True
 
-    Param4Label = "How much to expand width (0 - 2)"
-    Param4Visible = True
-    Param5Label = "How much to expand height (0 - 2)"
-    Param5Visible = True
+elif taskType=="9":
+    # 是否显示token输入框
+    is_show_token = True
+    TempInter = False
+    # 标题
+    title = r"""
+    <h1 align="center">Red Panda Image Generator by Recraft</h1>
+    """
+    # 标题下的描述
+    description = r"""
+    Go to <a href='https://heybeauty.ai/red-panda-recraft' target='_blank'><b>Recraft</b></a> for Faster and Free Generation! 🤗 .<br>
+
+    1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
+    2. After exhausted all trial attempts, you can <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>find your api key here</b></a> .<br>
+    3. Share app with your friends and enjoy! 😊.<br>
+    """
+    # 组件信息
+    TempLabel = "guide"
+    TempVisible = True
+    UserLabel = "upload and paint photo"
+    CaptionLabel = "caption text (required, any language)"
+    CapVisible = True
 
 else:
     # 是否显示token输入框
@@ -202,13 +220,13 @@ else:
 
     <b>Official 🤗 Gradio demo</b> for <a href='https://github.com/selfitcamera/Outfit-Anyone-in-the-Wild' target='_blank'><b>Product poster background generation</b></a>.<br>
     1. Please wait for a while after submitting the task, you will get the result within 30 seconds.
-    2. <a href='https://heybeauty.ai/keys' target='_blank'><b>You can find your api key here</b></a> .<br>
+    2. <a href='https://heybeauty.ai/home/apikeys' target='_blank'><b>You can find your api key here</b></a> .<br>
     3. Share app with your friends and enjoy! 😊.<br>
     """
 
 if is_show_token:
     # 当试用次数耗尽，显示的信息
-    no_more_attempts = "You've exhausted all trial attempts. [Enter Your API Key](https://heybeauty.ai/keys) to continue..."
+    no_more_attempts = "You've exhausted all trial attempts. [Enter Your API Key](https://heybeauty.ai/home/apikeys) to continue..."
 else:
     # 当试用次数耗尽，显示的信息
     no_more_attempts = "You've exhausted all trial attempts."

utils.py

@@ -39,15 +39,16 @@ def checkToken(token):
     '''
         输入token，检查token是否合法
     '''
-    return False
-    params = {'token':str(token)}
+    # return False
+    params = {'trans_type':'hf_space', 'cost_credits':1}
+    headers = {"Authorization": f"Bearer {token}"}
     session = requests.session()
-    ret = requests.post(f"{tokenUrl}", data=json.dumps(params))
+    ret = requests.post(f"{tokenUrl}", data=json.dumps(params), headers=headers)
     print(ret)
     res = False
     if ret.status_code==200:
-        if 'res' in ret.json():
-            res = (ret.json()['res'])>0
+        if 'left_credits' in ret.json():
+            res = (ret.json()['left_credits'])>0
     else:
         print(ret.json(), ret.status_code, 'call token failed')
     return res
@@ -86,7 +87,7 @@ class UserRecorder(object):
             if token_valid:
                 return True, ""
             else:
-                return False, "api key is invalid"
+                return False, "faild, api key is invalid"
         else:
             _, total_n, _ = self.get_record(ip=ip, token=token)
             if total_n>=LimitTask:
@@ -196,6 +197,12 @@ def get_showcase_examples(taskType):
         examples=[
             ["task7/temps/task7.webp", "task7/srcs/305.jpg", "task7/showcases/task7out.jpg"],
         ]
+    elif taskType=="9":
+        examples=[
+            ["task9/temps/caption.jpg", "task9/srcs/use1.jpg", "task9/showcases/show0.jpg"],
+            ["task9/temps/caption.jpg", "task9/srcs/use2.jpg", "task9/showcases/show1.webp"],
+            ["task9/temps/caption.jpg", "task9/srcs/use3.jpg", "task9/showcases/show2.webp"],
+        ]
 
     for i in range(len(examples)):
         for j in range(len(examples[i])):
@@ -208,7 +215,9 @@ def get_result_example(cloth_id, pose_id):
     res_path = os.path.join(result_dir, f"{cloth_id}_{pose_id}.jpg")
     return res_path
 
-def upload_user_img_mask(clientIp, img, mask=None):
+def upload_user_img_mask(clientIp, img, mask=None, taskType='1'):
+    if taskType in ['8', '9']:
+        return "img", ''
     timeId = int(  str(time.time()).replace(".", "")  )+random.randint(1000, 9999)
     fileName = clientIp.replace(".", "")+str(timeId)+".jpg"
     local_path = os.path.join(tmpFolder, fileName)
@@ -268,8 +277,8 @@ def publicSelfitTask(image, mask, temp_image, caption_text, param4_text, param5_
     else:
         print(ret.json(), ret.status_code, 'call public failed')
 
-def getTaskRes(taskId):
-    params = {'id':taskId}
+def getTaskRes(taskId, taskType):
+    params = {'id':taskId, 'task_type':taskType}
     session = requests.session()
     ret = requests.post(f"{apiUrl}/status", data=json.dumps(params))
     if ret.status_code==200: