File size: 3,854 Bytes
5d333f5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
import os
import sys
import cv2
import json
import random
import time
import requests
import func_timeout
import numpy as np
import gradio as gr
ApiUrl = os.environ['ApiUrl']
OpenId = os.environ['OpenId']
ApiKey = os.environ['ApiKey']
UploadToken = os.environ['UploadToken']
proj_dir = os.path.dirname(os.path.abspath(__file__))
data_dir = os.path.join(proj_dir, 'Datas')
# data_dir = "Datas"
tmpFolder = "tmp"
os.makedirs(tmpFolder, exist_ok=True)
def upload_imgs(apiUrl, UploadToken, cloth_image, pose_image):
folder = os.path.join(tmpFolder, str(random.randint(0, 100)))
os.makedirs(folder, exist_ok=True)
pose_path = os.path.join(folder, 'pose.jpg')
cloth_path = os.path.join(folder, 'cloth.jpg')
cv2.imwrite(pose_path, pose_image[:,:,::-1])
cv2.imwrite(cloth_path, cloth_image[:,:,::-1])
params = {'token':UploadToken,
'input1':'pose.jpg',
'input2':'cloth.jpg',
'protocol':'https'}
session = requests.session()
ret = requests.post(f"{apiUrl}/upload", data=json.dumps(params))
if ret.status_code==200:
if 'upload1' in ret.json():
data = ret.json()
with open(cloth_path, 'rb') as file:
headers = {"Content-Type": 'image/jpeg'}
response = requests.put(data['upload2'], data=file, headers=headers)
if response.status_code == 200:
print("上传成功")
else:
print(f"上传失败，状态码: {response.status_code}, 响应内容: {response.text}")
return
with open(pose_path, 'rb') as file:
response = requests.put(data['upload1'], data=file, headers=headers)
if response.status_code == 200:
print("上传成功")
else:
print(f"上传失败，状态码: {response.status_code}, 响应内容: {response.text}")
return
if os.path.exists(pose_path): os.remove(pose_path)
if os.path.exists(cloth_path): os.remove(cloth_path)
return {'pose':data['upload1'], 'cloth':data['upload2']}
def publicFastSwap(apiUrl, openId, apiKey, uploads, category, caption):
if category=="upper_cloth":
category = 1
elif category=="lower_cloth":
category = 2
elif category=="dresses":
category = 3
elif category=="full_body":
category = 4
params = {'openId':OpenId, 'apiKey':ApiKey,
'task_type':"10", 'image':str(uploads['pose']),
'mask':str(uploads['cloth']),
'param1':str(category), 'param2':str(caption),
'param3':'', 'param4':'', 'param5':'', }
session = requests.session()
ret = requests.post(f"{ApiUrl}/public", data=json.dumps(params))
if ret.status_code==200:
if 'id' in ret.json():
print('public task successfully!')
return ret.json()['id']
def getTaskRes(apiUrl, taskId):
params = {'id':taskId, 'task_type':"10"}
session = requests.session()
ret = requests.post(f"{apiUrl}/status", data=json.dumps(params))
if ret.status_code==200:
if 'status' in ret.json():
return ret.json()
else:
print(ret.json(), ret.status_code, 'call status failed')
return None
@func_timeout.func_set_timeout(10)
def check_func(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.lower() in Regions.lower():
print(nat, 'invalid')
return False
else:
print(nat, 'valid')
return True
def check_warp(ip):
try:
return check_func(ip)
except Exception as e:
print(e)
return True
|