init

Datas/UserRecord_6/Ips/127.0.0.1.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:94d8b6463d24175ce07c7e3c63920c1028089b49010e33edd8c2ffaef17f0bac
-size 830
+oid sha256:9d5a1a961259242c785bddccab82cd792a956e381fa778c0554b9f61d8812ba9
+size 1091

Datas/UserRecord_7/Ips/127.0.0.1.pkl

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

app.py

@@ -15,7 +15,8 @@ css = """
 """
 
 
-def onClick(temp_image, user_image, caption_text, token_text, request: gr.Request):
+def onClick(temp_image, user_image, caption_text, token_text, 
+        param4_text, param5_text, request: gr.Request):
 
     user_mask = None
     if taskType=='2':
@@ -30,12 +31,24 @@ def onClick(temp_image, user_image, caption_text, token_text, request: gr.Reques
             yield None, "please draw a area!!!"
             return None, "please draw a area!!!" 
 
+    if taskType=='7':
+        try:
+            param4_text, param5_text = str(float(param4_text)), str(float(param5_text))
+        except ValueError:
+            yield None, "Invalid width/height: Please enter a valid float"
+            return None, "Invalid width/height: Please enter a valid float"
+        if len(caption_text)==0:
+            yield None, "Please enter English caption text !!! "
+            return None, "Please enter English caption text !!! "
+    else:
+        param4_text, param5_text = '', ''
+
     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 background!!!"
-        return None, "please choose a template background!!!"
+        yield None, "please choose a template!!!"
+        return None, "please choose a template!!!"
     if user_image is None: 
         yield None, "please upload a photo!!!"
         return None, "please upload a photo!!!"
@@ -61,7 +74,8 @@ def onClick(temp_image, user_image, caption_text, token_text, request: gr.Reques
 
         # return 
         # 发布任务
-        taskId = publicSelfitTask(upload_url, uploadm_url, temp_image, caption_text)
+        taskId = publicSelfitTask(upload_url, uploadm_url, temp_image, 
+            caption_text, param4_text, param5_text)
         if not taskId:
             yield None, "fail to public task..."
             return None, "fail to public task..."
@@ -112,7 +126,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             with gr.Column():
                 temp_image = gr.Image(sources='clipboard', type="filepath", label=TempLabel, 
-                    value=temp_examples[0][0], visible=TempVisible)
+                    value=temp_examples[0][0], visible=TempVisible, interactive=TempInter)
                 temp_example = gr.Examples(inputs=[temp_image], examples_per_page=9, 
                     examples=temp_examples, visible=TempVisible)
         with gr.Column():
@@ -124,7 +138,10 @@ with gr.Blocks(css=css) as demo:
                         transforms=[], label=UserLabel)
                 else:
                     user_image = gr.Image(value=None, type="numpy", label=UserLabel)
+                param4_text = gr.Textbox(value="0.5", interactive=True, label=Param4Label, visible=Param4Visible)
+                param5_text = gr.Textbox(value="0.5", interactive=True, label=Param5Label, visible=Param5Visible)
                 caption_text = gr.Textbox(value="", interactive=True, label=CaptionLabel, visible=CapVisible)
+
         with gr.Column():
             with gr.Column():
                 res_image = gr.Image(label="generate image", value=None, type="filepath")
@@ -155,8 +172,8 @@ with gr.Blocks(css=css) as demo:
     outputs_onload = [his_input1, his_output1, his_input2, his_output2, his_input3, his_output3,
         MK02, run_button]
 
-    run_button.click(fn=onClick, inputs=[temp_image, user_image, caption_text, token_text], 
-                     outputs=[res_image, info_text])
+    run_button.click(fn=onClick, inputs=[temp_image, user_image, caption_text, 
+        token_text, param4_text, param5_text], outputs=[res_image, info_text])
 
     refresh_button.click(fn=onLoad, inputs=[token_text], outputs=outputs_onload)
 

config.py

@@ -23,6 +23,11 @@ except Exception as e:
 
 
 CapVisible = True
+Param4Label = ""
+Param4Visible = False
+Param5Label = ""
+Param5Visible = False
+TempInter = True
 
 if taskType=="3":
     # 是否显示token输入框
@@ -47,6 +52,7 @@ if taskType=="3":
 
 elif taskType=="4":
     # 是否显示token输入框
+    TempInter = False
     is_show_token = True
     # 标题
     title = r"""
@@ -90,6 +96,7 @@ elif taskType=="6":
 elif taskType=="5":
     # 是否显示token输入框
     is_show_token = True
+    TempInter = False
     # 标题
     title = r"""
     <h1 align="center">Selfit AI App: Poster(human) Background Swap</h1>
@@ -111,6 +118,7 @@ elif taskType=="5":
 elif taskType=="1":
     # 是否显示token输入框
     is_show_token = True
+    TempInter = False
     # 标题
     title = r"""
     <h1 align="center">Selfit AI App: Object Segment</h1>
@@ -133,6 +141,7 @@ elif taskType=="1":
 elif taskType=="2":
     # 是否显示token输入框
     is_show_token = True
+    TempInter = False
     # 标题
     title = r"""
     <h1 align="center">Selfit AI App: Object Remove</h1>
@@ -152,6 +161,34 @@ elif taskType=="2":
     CaptionLabel = "English caption text (optional)"
     CapVisible = False
 
+elif taskType=="7":
+    # 是否显示token输入框
+    is_show_token = True
+    TempInter = False
+    # 标题
+    title = r"""
+    <h1 align="center">Selfit AI App: Image Outpaint</h1>
+    """
+    # 标题下的描述
+    description = r"""
+    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>
+    3. Share app with your friends and enjoy! 😊.<br>
+    """
+    # 组件信息
+    TempLabel = "guide"
+    TempVisible = True
+    UserLabel = "upload and paint photo"
+    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
+
 else:
     # 是否显示token输入框
     is_show_token = False

utils.py

@@ -192,6 +192,10 @@ def get_showcase_examples(taskType):
         examples=[
             ["task2/temps/caption.jpg", "task2/srcs/street.webp", "task2/showcases/out1.jpg"],
         ]
+    elif taskType=="7":
+        examples=[
+            ["task2/temps/caption.jpg", "task2/srcs/street.webp", "task2/showcases/out1.jpg"],
+        ]
 
     for i in range(len(examples)):
         for j in range(len(examples[i])):
@@ -249,14 +253,11 @@ def upload_user_img_mask(clientIp, img, mask=None):
     return res, uploadm_url
 
 
-def publicSelfitTask(image, mask, temp_image, caption_text):
-    if taskType in ['5']: # 无模板
-        temp_name = ''
-    else:
-        temp_name = os.path.basename(temp_image).split('.')[0]
+def publicSelfitTask(image, mask, temp_image, caption_text, param4_text, param5_text):
+    temp_name = os.path.basename(temp_image).split('.')[0]
     params = {'openId':openId, 'apiKey':apiKey, 'image':image, 'mask':mask,
         "image_type":"2", "task_type":taskType, 'param1':temp_name, 
-        'param2':str(caption_text), 'param3':"1", 'param4':"", 'param5':""}
+        'param2':str(caption_text), 'param3':"1", 'param4':param4_text, 'param5':param5_text}
     session = requests.session()
     ret = requests.post(f"{apiUrl}/public", data=json.dumps(params))
     print(ret)