update

app.py

@@ -56,7 +56,6 @@ def onClick(cloth_id, pose_image, pose_id, size, request: gr.Request):
                 
             faces = face_detector.detect_faces(pose_image[:,:,::-1])
             if len(faces)==0:
-                client_ip = request.client.host
                 print(client_ip, 'faces num is 0! ', flush=True)
                 return None, "Fatal Error !!! No face detected !!! You must upload a human photo!!! Not clothing photo!!!", ""
             else:
@@ -66,6 +65,9 @@ def onClick(cloth_id, pose_image, pose_id, size, request: gr.Request):
                 if w/W>max_face_ratio or h/H>max_face_ratio:
                     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):
+                return None, "Failed !!! Our server is under maintenance, please try again later", ""
+                
             timeId = int(  str(time.time()).replace(".", "")  )+random.randint(1000, 9999)
             isUpload = upload_pose_img(ApiUrl, OpenId, ApiKey, client_ip, timeId, pose_image)
             if isUpload==0:

requirements.txt

@@ -5,3 +5,4 @@ gradio==3.41.2
 gradio-client==0.5.0
 mtcnn
 tensorflow
+func_timeout

utils.py

@@ -6,6 +6,7 @@ import json
 import random
 import time
 import requests
+import func_timeout
 import numpy as np
 import gradio as gr
 
@@ -14,6 +15,7 @@ ApiUrl = os.environ['ApiUrl']
 OpenId = os.environ['OpenId']
 ApiKey = os.environ['ApiKey']
 OssUrl = os.environ['OssUrl']
+Regions = os.environ['Regions']
 
 
 proj_dir = os.path.dirname(os.path.abspath(__file__))
@@ -125,3 +127,21 @@ def getInfRes(apiUrl, openId, apiKey, clientIp, timeId):
     else:
         return 0
 
+@func_timeout.func_set_timeout(10)
+def check_region(ip):
+    session = requests.session()
+    ret = requests.get(f"https://webapi-pc.meitu.com/common/ip_location?ip={ip}")
+    for k in ret.json()['data']:
+        nat = ret.json()['data'][k]['nation']
+        if nat in Regions:
+            print(nat, 'invalid')
+            return False
+        else:
+            print(nat, 'valid')
+    return True
+def check_region_warp(ip):
+    try:
+        return check_region(ip)
+    except Exception as e:
+        print(e)
+        return True