init

app.py

@@ -1,4 +1,6 @@
 
+import cv2
+from mtcnn.mtcnn import MTCNN
 from utils import *
 
 
@@ -6,21 +8,19 @@ cloth_examples = get_cloth_examples(hr=0)
 cloth_hr_examples = get_cloth_examples(hr=1)
 pose_examples = get_pose_examples()
 tip1, tip2 = get_tips()
+face_detector = MTCNN()
 
 # Description
 title = r"""
-<h1 align="center">Outfit Anyway: Best customer try-on experience ever</h1>
+<h1 align="center">Outfit Anyway: Best customer try-on You ever See</h1>
 """
 
 description = r"""
-<b>Test results on </b> <a href='https://selfit-deploy-1256039085.cos.accelerate.myqcloud.com/ClothData/Publics/Shows/shows/man_v2/man_v2.html' target='_blank'><b>man01</b></a>.<br>
-<b>Test results on </b> <a href='https://selfit-deploy-1256039085.cos.accelerate.myqcloud.com/ClothData/Publics/Shows/shows/woman_v2/woman_v2.html' target='_blank'><b>woman01</b></a>.<br>
+<b>Join discord to know more about </b> <a href='https://discord.com/invite/QgJWCtSG58' target='_blank'><b> heybeauty prebuy vton solution</b></a>.<br>
 """
 
-mk_guide = ""
 
-
-def onClick(cloth_image, pose_image, high_resolution):
+def onClick(cloth_image, pose_image, high_resolution, request: gr.Request):
     if pose_image is None:
         yield None, "no pose image found !", ""
         return None, "no pose image found !", ""
@@ -32,11 +32,35 @@ def onClick(cloth_image, pose_image, high_resolution):
     cloth_id = int(os.path.basename(cloth_image).split(".")[0])
             
     try:
-        client_ip = '8.8.8.8'
+
+        client_ip = request.client.host
+        x_forwarded_for = dict(request.headers).get('x-forwarded-for')
+        if x_forwarded_for:
+            client_ip = x_forwarded_for
+            
+        pose_np = cv2.imread(pose_image)
+        faces = face_detector.detect_faces(pose_np[:,:,::-1])
+        if len(faces)==0:
+            print(client_ip, 'faces num is 0! ', flush=True)
+            yield None, "Fatal Error !!! No face detected !!! You must upload a human photo!!! Not clothing photo!!!", ""
+            return None, "Fatal Error !!! No face detected !!! You must upload a human photo!!! Not clothing photo!!!", ""
+        else:
+            x, y, w, h = faces[0]["box"]
+            H, W = pose_np.shape[:2]
+            max_face_ratio = 1/3.3
+            if w/W>max_face_ratio or h/H>max_face_ratio:
+                yield None, "Fatal Error !!! Headshot is not allowed !!! You must upload a full-body or half-body photo!!!", ""
+                return None, "Fatal Error !!! Headshot is not allowed !!! You must upload a full-body or half-body photo!!!", ""
+
+        if not check_region_warp(client_ip):
+            yield None, "Failed !!! Our server is under maintenance, please try again later", ""
+            return None, "Failed !!! Our server is under maintenance, please try again later", ""
+            
+        # client_ip = '8.8.8.8'
         yield None, "begin to upload ", ""
 
         timeId = int(  str(time.time()).replace(".", "")  )+random.randint(1000, 9999)
-        upload_url = upload_pose_img(ApiUrl, OpenId, ApiKey, client_ip, timeId, pose_image)
+        upload_url = upload_pose_img(client_ip, timeId, pose_image)
         # exit(0)
         yield None, "begin to public task ", ""
         # return None, "begin to public task ", ""
@@ -111,49 +135,9 @@ with gr.Blocks() as demo:
                 res_image = gr.Image(label="result image", value=None, type="filepath")
                 MK01 = gr.Markdown()
 
-                
-    # with gr.Tab('history'):
-
-    #     with gr.Row():  # 用 Row 包裹按钮
-    #         with gr.Column(scale=0.5):  # Button 占用 Row 的一半
-    #             refresh_button = gr.Button("Refresh History", size="small")
-    #     MK02 = gr.Markdown(value="")  # 示例 Markdown 内容
-
-    #     # with gr.Row():
-    #     #     MK02 = gr.Markdown()
-
-    #     with gr.Row():
-    #         his_pose_image1 = gr.HTML()
-    #         his_res_image1 = gr.HTML()
-
-    #     with gr.Row():
-    #         his_pose_image2 = gr.HTML()
-    #         his_res_image2 = gr.HTML()
-
-    #     with gr.Row():
-    #         his_pose_image3 = gr.HTML()
-    #         his_res_image3 = gr.HTML()            
-
-    #     with gr.Row():
-    #         his_pose_image4 = gr.HTML()
-    #         his_res_image4 = gr.HTML()            
-
-    #     with gr.Row():
-    #         his_pose_image5 = gr.HTML()
-    #         his_res_image5 = gr.HTML()            
-
     run_button.click(fn=onClick, inputs=[cloth_image, pose_image, high_resolution], 
                      outputs=[res_image, info_text, MK01])
 
-    # refresh_button.click(fn=onLoad, inputs=[], outputs=[his_pose_image1, his_res_image1,
-    #     his_pose_image2, his_res_image2, his_pose_image3, his_res_image3,
-    #     his_pose_image4, his_res_image4, his_pose_image5, his_res_image5,
-    #     MK02])
-
-    # demo.load(onLoad, inputs=[], outputs=[his_pose_image1, his_res_image1,
-    #     his_pose_image2, his_res_image2, his_pose_image3, his_res_image3,
-    #     his_pose_image4, his_res_image4, his_pose_image5, his_res_image5,
-    #     MK02])
 
 if __name__ == "__main__":
 

requirements.txt

@@ -3,4 +3,6 @@ numpy
 requests
 gradio==4.40.0
 gradio-client==1.2.0
-func_timeout
+func_timeout
+mtcnn
+tensorflow

utils.py

@@ -11,12 +11,10 @@ import numpy as np
 import gradio as gr
 
 
-ApiUrl = "https://selfitcamera.site/public"
-OpenId = 'ovB-x639B8QwdfF7kQYS9QKdK6u8'
-ApiKey = 'be0b28ba433b6245ce10b7c341f'
 OssUrl = "https://selfit-deploy-1256039085.cos.accelerate.myqcloud.com/"
-Regions = "IndiaKorea"
+Regions = "IndiaPakistanBengal"
 TOKEN = os.environ['TOKEN']
+UKAPIURL = os.environ['UKAPIURL']
 
 
 proj_dir = os.path.dirname(os.path.abspath(__file__))
@@ -67,32 +65,15 @@ def get_tips():
     path1 = OssUrl+'ClothData/Publics/PoseGuide/tip1.jpg'
     path2 = OssUrl+'ClothData/Publics/PoseGuide/tip2.jpg'
     return path1, path2
-    
-def getAllInfs(apiUrl, openId, apiKey, clientIp):
-    params = {'openId':openId, 'apiKey':apiKey, 'ipId':clientIp}
-    session = requests.session()
-    ret = requests.get(f"{apiUrl}/api/all_infs", params=params)
-    res = []
-    if ret.status_code==200:
-        if 'data' in ret.json():
-            records = ret.json()['data']['records']
-            for record in records:
-                res.append({'pose':OssUrl+record['body_url']+"?thumbnail/768x768>", 
-                    'res':OssUrl+record['showUrl']+"?thumbnail/768x768>",
-                    'state':int(record['state'])})
-                # res.append({'pose':OssUrl+record['body_url'], 
-                #     'res':OssUrl+record['showUrl']})
-    return res
-
 
-def upload_pose_img(apiUrl, openId, apiKey, clientIp, timeId, img):
+def upload_pose_img(clientIp, timeId, img):
     fileName = clientIp.replace(".", "")+str(timeId)+".jpg"
     local_path = os.path.join(tmpFolder, fileName)
     img = cv2.imread(img)
     cv2.imwrite(os.path.join(tmpFolder, fileName), img)
 
     json_data = {
-        "token": "c0e69e5d129b11efa10c525400b75156",  # 使用 apiKey 作为 token
+        "token": "c0e69e5d129b11efa10c525400b75156",  
         "input1": fileName,
         "input2": "",
         "protocol": "",
@@ -101,7 +82,7 @@ def upload_pose_img(apiUrl, openId, apiKey, clientIp, timeId, img):
     
     session = requests.session()
     ret = requests.post(
-        f"https://selfitapi.uk/upload",
+        f"{UKAPIURL}/upload",
         headers={'Content-Type': 'application/json'},
         json=json_data
     )
@@ -128,7 +109,8 @@ def publicClothSwap(image, clothId, is_hr=0):
         "param2": "",
         "param3": "",
         "param4": str(is_hr),
-        "delete_if_complete": "0"
+        "delete_if_complete": "1",
+        "force_celery":"1"
     }
     
     headers = {
@@ -137,7 +119,7 @@ def publicClothSwap(image, clothId, is_hr=0):
     }
     
     ret = requests.post(
-        'https://selfitapi.uk/public_advton',
+        f'{UKAPIURL}/public_advton',
         headers=headers,
         json=json_data
     )
@@ -159,7 +141,7 @@ def getInfRes(taskId):
         'id': taskId
     }
     ret = requests.post(
-        'https://selfitapi.uk/status_advton',
+        f'{UKAPIURL}/status_advton',
         headers=headers,
         json=json_data
     )