Files
change_hair_3090/hair_service_sd/common/callback.py
T
colomi 0eb61f3e60 初始化换发型项目:3个微服务代码 + 部署脚本
包含:
- hair_service_sd: 换发型/换发色算法服务 (端口 8801)
- photo_service: LoRA 训练调度服务 (端口 32678)
- stable-diffusion-webui: SD WebUI 推理服务 (端口 57860)
- kohya_ss_home: 训练环境代码
- meidaojia: 监控测试脚本
- setup.sh: 一键部署脚本 (conda环境恢复 + 配置生成 + 完整性检查)
- start_all_services.sh: 启动3个服务
- configure.ini.template: 路径模板化 (BASE_DIR自动推导)
- conda_envs/py310.yml: py310 环境定义

大文件 (weights/, models/, data/, conda_envs/*.tar.gz 等) 通过 .gitignore 排除,
由网盘单独上传。
2026-07-11 18:11:49 +08:00

47 lines
1.2 KiB
Python

import requests
import json
callback_hairstyle_url = "https://puton.meidaojia.com/api/cloth/callBack"
def callback_color(color_id, cover_img, success):
url = "http://172.21.0.3:8080/ydapp/system/platform/complete_color_model"
payload = json.dumps({
"success": success,
"colorId": color_id,
"coverImg": cover_img
})
headers = {
'X-MZ-API-TOKEN': '98bcf37c942a2240ba2d907c96ed1137',
'Content-Type': 'application/json'
}
response = requests.request("POST", url, headers=headers, data=payload)
return response
# print(response.text)
def callback_hairstyle(req_id, state, message, clothId):
url = callback_hairstyle_url
payload = json.dumps({
"taskId": req_id,
"status": state,
"clothId": clothId,
"msg": message
})
status = -1
print(f"url:{url},complete_hair_model payload:{payload}")
headers = {
'X-MZ-API-TOKEN': '98bcf37c942a2240ba2d907c96ed1137',
'Content-Type': 'application/json'
}
try:
response = requests.request("POST", url, headers=headers, data=payload)
print("response:", response.text)
status = 0
except Exception as e:
print(e)
return response.text, status