save code
This commit is contained in:
colomi
@@ -0,0 +1,323 @@
|
||||
import os
|
||||
import uuid
|
||||
import requests
|
||||
from flask import Flask, request, jsonify
|
||||
from urllib.parse import urlparse
|
||||
from volcenginesdkarkruntime import Ark
|
||||
import json
|
||||
import base64
|
||||
import time
|
||||
from PIL import Image
|
||||
from datetime import datetime
|
||||
|
||||
APP_ROOT = os.path.dirname(os.path.abspath(__file__))
|
||||
STATIC_FOLDER = os.path.join(APP_ROOT, 'static')
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
# 请确保您已将 API Key 存储在环境变量 ARK_API_KEY 中
|
||||
# 初始化Ark客户端，从环境变量中读取您的API Key
|
||||
client = Ark(
|
||||
# 此为默认路径，您可根据业务所在地域进行配置
|
||||
base_url="https://ark.cn-beijing.volces.com/api/v3",
|
||||
# 从环境变量中获取您的 API Key。此为默认方式，您可根据需要进行修改
|
||||
# api_key=os.environ.get("ARK_API_KEY"),
|
||||
api_key='14fc0280-fc65-462d-ac2d-50178c0212e3'
|
||||
)
|
||||
|
||||
def parse_facial_data(data_str):
|
||||
print(f'原始字符串 {data_str}')
|
||||
# 去除多余的标记和换行符
|
||||
json_str_clean = data_str.strip('```json\n').strip('```').strip()
|
||||
|
||||
# 转换为Python字典
|
||||
data = json.loads(json_str_clean)
|
||||
s = data['面部年龄']
|
||||
s = s.replace('岁', '')
|
||||
parts = s.split('-')
|
||||
start_age = int(parts[0])
|
||||
end_age = int(parts[1])
|
||||
data['start_age'] = start_age
|
||||
data['end_age'] = end_age
|
||||
return data
|
||||
|
||||
def GetPicDesc(img_url):
|
||||
response = client.chat.completions.create(
|
||||
# 指定您创建的方舟推理接入点 ID，此处已帮您修改为您的推理接入点 ID
|
||||
model="doubao-1.5-vision-pro-250328",
|
||||
messages=[
|
||||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
{
|
||||
"type": "image_url",
|
||||
"image_url": {
|
||||
"url": img_url
|
||||
},
|
||||
},
|
||||
{"type": "text", "text": "告诉我图片中的服装长度, 只要答案, 选项有(腰部以上、腰、跨、大腿上部1/3、大腿中部、膝盖、小腿、小腿上部1/3、脚踝、拖地、难以辨认)"},
|
||||
],
|
||||
}
|
||||
],
|
||||
|
||||
)
|
||||
|
||||
text = response.choices[0].message.content
|
||||
result = {}
|
||||
|
||||
try:
|
||||
result = parse_facial_data(text)
|
||||
except:
|
||||
print(f'Error {text}')
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def change(human_name, cloth_name, c_width, c_height):
|
||||
with open('/home/szlc/code/ComfyUI/change_cloth/change_new.json', 'r', encoding='utf-8') as file:
|
||||
prompt_text = file.read()
|
||||
|
||||
prompt = json.loads(prompt_text)
|
||||
|
||||
prompt["99"]["inputs"]["width"] = int((c_width/c_height) * 1024)
|
||||
|
||||
#input cloth img
|
||||
prompt["22"]["inputs"]["image"] = cloth_name
|
||||
|
||||
#input human img
|
||||
prompt["61"]["inputs"]["image"] = human_name
|
||||
|
||||
#out put name
|
||||
out_img_name = str(uuid.uuid4())[:8]
|
||||
prompt["102"]["inputs"]["filename_prefix"] = out_img_name
|
||||
|
||||
p = {"prompt": prompt}
|
||||
|
||||
data = json.dumps(p).encode('utf-8')
|
||||
|
||||
response = requests.post("http://localhost:8188/prompt", data=data)
|
||||
prompt_id = response.json()["prompt_id"]
|
||||
|
||||
|
||||
# 2. 轮询队列，直到任务完成
|
||||
while True:
|
||||
queue = requests.get("http://localhost:8188/queue").json()
|
||||
# print(queue)
|
||||
if not queue["queue_running"] and not queue["queue_pending"]:
|
||||
break # 队列为空，任务已完成
|
||||
time.sleep(0.5) # 避免频繁请求
|
||||
|
||||
# 3. 从历史记录中获取结果
|
||||
history = requests.get("http://localhost:8188/history").json()
|
||||
print("History:", history)
|
||||
outputs = history[prompt_id]["outputs"]
|
||||
for out in outputs:
|
||||
if 'images' in outputs[out]:
|
||||
for out_img_name in outputs[out]['images']:
|
||||
if 'filename' in out_img_name:
|
||||
return out_img_name['filename']
|
||||
return None
|
||||
|
||||
def save_base64_image(base64_str, prefix):
|
||||
"""
|
||||
将base64字符串保存为图片文件
|
||||
:param base64_str: 带有类型标识的base64字符串
|
||||
:param prefix: 文件名前缀
|
||||
:return: 保存的文件路径
|
||||
"""
|
||||
try:
|
||||
# 分离base64头部和实际数据
|
||||
header, data = base64_str.split(',', 1)
|
||||
|
||||
# 从头部获取文件扩展名
|
||||
file_ext = ''
|
||||
if 'image/png' in header:
|
||||
file_ext = '.png'
|
||||
elif 'image/jpeg' in header:
|
||||
file_ext = '.jpg'
|
||||
elif 'image/jpg' in header:
|
||||
file_ext = '.jpg'
|
||||
elif 'image/gif' in header:
|
||||
file_ext = '.gif'
|
||||
else:
|
||||
file_ext = '.png' # 默认使用png
|
||||
|
||||
# 生成唯一文件名
|
||||
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
unique_id = str(uuid.uuid4())[:8]
|
||||
filename = f"{prefix}_{timestamp}_{unique_id}{file_ext}"
|
||||
filepath = os.path.join('/home/szlc/code/ComfyUI/change_cloth/static', filename)
|
||||
|
||||
# 解码并保存图片
|
||||
with open(filepath, 'wb') as f:
|
||||
f.write(base64.b64decode(data))
|
||||
|
||||
input_filepath = os.path.join('/home/szlc/code/ComfyUI/input', filename)
|
||||
# 解码并保存图片
|
||||
with open(input_filepath, 'wb') as f:
|
||||
f.write(base64.b64decode(data))
|
||||
|
||||
return filename
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error saving image: {str(e)}")
|
||||
return None
|
||||
|
||||
def get_image_dimensions(image_path):
|
||||
try:
|
||||
with Image.open(image_path) as img:
|
||||
width, height = img.size
|
||||
return width, height
|
||||
except Exception as e:
|
||||
print(f"Error: {e}")
|
||||
return None
|
||||
|
||||
def image_to_base64(file_path, mime_type=None):
|
||||
if mime_type is None:
|
||||
extension = file_path.split('.')[-1].lower()
|
||||
mime_types = {
|
||||
'jpg': 'image/jpeg',
|
||||
'jpeg': 'image/jpeg',
|
||||
'png': 'image/png',
|
||||
'gif': 'image/gif',
|
||||
'webp': 'image/webp',
|
||||
'bmp': 'image/bmp'
|
||||
}
|
||||
mime_type = mime_types.get(extension, 'application/octet-stream')
|
||||
|
||||
# 读取文件内容并编码为Base64
|
||||
with open(file_path, 'rb') as image_file:
|
||||
encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
|
||||
|
||||
# 组合成Data URI格式
|
||||
return f"data:{mime_type};base64,{encoded_string}"
|
||||
|
||||
def process_change_cloth(human_filename, cloth_filename, output_format):
|
||||
w,h = get_image_dimensions(f'/home/szlc/code/ComfyUI/input/{human_filename}')
|
||||
|
||||
out_put_name = change(human_filename, cloth_filename, w, h)
|
||||
if out_put_name == None:
|
||||
return jsonify({"ret":-1, 'msg': 'Failed to change cloth'}), 500
|
||||
|
||||
image = Image.open(f'/home/szlc/code/ComfyUI/output/{out_put_name}')
|
||||
jpg_name = out_put_name.replace(".png", ".jpg")
|
||||
jpg_path_name = f'/home/szlc/code/ComfyUI/change_cloth/static/{jpg_name}'
|
||||
image.save(jpg_path_name, quality=95)
|
||||
|
||||
if 'base64' in output_format:
|
||||
return jsonify({
|
||||
"ret":0,
|
||||
"msg":"success",
|
||||
"data":image_to_base64(jpg_path_name)
|
||||
})
|
||||
else:
|
||||
return jsonify({
|
||||
"ret":0,
|
||||
"msg":"success",
|
||||
"url":f"http://112.126.94.241:18018/static/{jpg_name}"
|
||||
})
|
||||
|
||||
@app.route('/change_cloth_base64', methods=['POST'])
|
||||
def change_cloth_base64():
|
||||
# 获取参数
|
||||
data = request.get_json()
|
||||
if not data:
|
||||
return jsonify({"ret":-1,'msg': 'No JSON data provided'}), 400
|
||||
|
||||
human_img = data.get('human_img')
|
||||
cloth_img = data.get('cloth_img')
|
||||
output_format = data.get('output_format')
|
||||
|
||||
if not human_img or not cloth_img:
|
||||
return jsonify({"ret":-1, 'msg': 'Both human_img and cloth_img are required'}), 400
|
||||
|
||||
try:
|
||||
|
||||
if 'base64' in output_format:
|
||||
return jsonify({
|
||||
"ret":0,
|
||||
"msg":"success",
|
||||
"data":image_to_base64("/home/szlc/code/ComfyUI/input/girl_full.jpg")
|
||||
})
|
||||
else:
|
||||
return jsonify({
|
||||
"ret":0,
|
||||
"msg":"success",
|
||||
"url":f"http://43.143.205.217/imgs/processed_f60d0a9862834e1c92bc1c64ecf8f3ba.jpg"
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({"ret":-1, 'error': str(e)}), 500
|
||||
|
||||
def get_file_extension(url_or_filename):
|
||||
"""从 URL 或文件名中提取扩展名（如 .jpg、.png）"""
|
||||
# 处理 URL 情况（如 https://example.com/image.jpg?width=200）
|
||||
if url_or_filename.startswith(('http://', 'https://')):
|
||||
parsed = urlparse(url_or_filename)
|
||||
path = parsed.path
|
||||
else:
|
||||
path = url_or_filename
|
||||
|
||||
# 提取扩展名（转换为小写，去掉问号后的参数）
|
||||
ext = os.path.splitext(path)[1].lower().split('?')[0]
|
||||
return ext if ext else '.png' # 默认 PNG（如果无法提取）
|
||||
|
||||
def save_image_from_url(image_url):
|
||||
"""从 URL 下载图片并保留原始格式"""
|
||||
try:
|
||||
response = requests.get(image_url, stream=True)
|
||||
response.raise_for_status()
|
||||
|
||||
# 获取原始图片格式
|
||||
ext = get_file_extension(image_url)
|
||||
filename = f"{uuid.uuid4()}{ext}"
|
||||
filepath = os.path.join('/home/szlc/code/ComfyUI/change_cloth/static', filename)
|
||||
|
||||
with open(filepath, "wb") as f:
|
||||
for chunk in response.iter_content(1024):
|
||||
f.write(chunk)
|
||||
|
||||
input_filepath = os.path.join('/home/szlc/code/ComfyUI/input', filename)
|
||||
with open(input_filepath, "wb") as f:
|
||||
for chunk in response.iter_content(1024):
|
||||
f.write(chunk)
|
||||
|
||||
return filename
|
||||
except Exception as e:
|
||||
print(f"Error saving image from URL: {e}")
|
||||
return None
|
||||
|
||||
@app.route('/change_cloth', methods=['POST'])
|
||||
def change_cloth():
|
||||
"""从 URL 下载图片"""
|
||||
data = request.json
|
||||
human_url = data.get("human_url")
|
||||
|
||||
if not human_url:
|
||||
return jsonify({"error": "Missing 'human_url' parameter"}), 400
|
||||
|
||||
|
||||
cloth_url = data.get("cloth_url")
|
||||
|
||||
if not cloth_url:
|
||||
return jsonify({"error": "Missing 'cloth_url' parameter"}), 400
|
||||
|
||||
output_format = data.get('output_format')
|
||||
if not output_format:
|
||||
return jsonify({"error": "Missing 'output_format' parameter"}), 500
|
||||
|
||||
if 'base64' in output_format:
|
||||
return jsonify({
|
||||
"ret":0,
|
||||
"msg":"success",
|
||||
"data":image_to_base64("/home/szlc/code/ComfyUI/input/girl_full.jpg")
|
||||
})
|
||||
else:
|
||||
return jsonify({
|
||||
"ret":0,
|
||||
"msg":"success",
|
||||
"url":f"http://43.143.205.217/imgs/processed_f60d0a9862834e1c92bc1c64ecf8f3ba.jpg"
|
||||
})
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(host="0.0.0.0", port=8888, debug=False)
|
||||
+1
-1
@@ -164,7 +164,7 @@
|
||||
console.log("准备发送的数据:", data); // 调试用
|
||||
|
||||
// 调用API
|
||||
fetch('http://112.126.94.241:18018/change_cloth_base64', {
|
||||
fetch('http://112.126.94.241:18888/change_cloth_base64', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json'
|
||||
|
||||
@@ -0,0 +1,227 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="zh-CN">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>服装更换工具</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: Arial, sans-serif;
|
||||
max-width: 800px;
|
||||
margin: 0 auto;
|
||||
padding: 20px;
|
||||
line-height: 1.6;
|
||||
}
|
||||
h1 {
|
||||
text-align: center;
|
||||
color: #333;
|
||||
}
|
||||
.image-upload-container {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
.image-upload-box {
|
||||
width: 48%;
|
||||
border: 2px dashed #ccc;
|
||||
padding: 15px;
|
||||
text-align: center;
|
||||
border-radius: 5px;
|
||||
}
|
||||
.image-upload-box h3 {
|
||||
margin-top: 0;
|
||||
}
|
||||
.image-preview {
|
||||
max-width: 100%;
|
||||
max-height: 300px;
|
||||
margin-top: 10px;
|
||||
display: none;
|
||||
}
|
||||
button {
|
||||
background-color: #4CAF50;
|
||||
color: white;
|
||||
padding: 10px 15px;
|
||||
border: none;
|
||||
border-radius: 4px;
|
||||
cursor: pointer;
|
||||
font-size: 16px;
|
||||
display: block;
|
||||
margin: 20px auto;
|
||||
}
|
||||
button:hover {
|
||||
background-color: #45a049;
|
||||
}
|
||||
#result {
|
||||
margin-top: 20px;
|
||||
padding: 15px;
|
||||
border: 1px solid #ddd;
|
||||
border-radius: 4px;
|
||||
display: none;
|
||||
}
|
||||
#loading {
|
||||
text-align: center;
|
||||
display: none;
|
||||
margin: 20px 0;
|
||||
}
|
||||
.spinner {
|
||||
border: 5px solid #f3f3f3;
|
||||
border-top: 5px solid #3498db;
|
||||
border-radius: 50%;
|
||||
width: 50px;
|
||||
height: 50px;
|
||||
animation: spin 2s linear infinite;
|
||||
margin: 0 auto;
|
||||
}
|
||||
@keyframes spin {
|
||||
0% { transform: rotate(0deg); }
|
||||
100% { transform: rotate(360deg); }
|
||||
}
|
||||
.error {
|
||||
color: red;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<h1>服装更换工具</h1>
|
||||
|
||||
<div class="image-upload-container">
|
||||
<div class="image-upload-box">
|
||||
<h3>人体图片</h3>
|
||||
<input type="file" id="humanInput" accept="image/*">
|
||||
<img id="humanPreview" class="image-preview" alt="人体图片预览">
|
||||
</div>
|
||||
|
||||
<div class="image-upload-box">
|
||||
<h3>衣服图片</h3>
|
||||
<input type="file" id="clothInput" accept="image/*">
|
||||
<img id="clothPreview" class="image-preview" alt="衣服图片预览">
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<button id="submitBtn">提交处理</button>
|
||||
|
||||
<div id="loading">
|
||||
<div class="spinner"></div>
|
||||
<p>正在处理中，请稍候...</p>
|
||||
</div>
|
||||
|
||||
<div id="result"></div>
|
||||
|
||||
<script>
|
||||
// 图片预览功能
|
||||
document.getElementById('humanInput').addEventListener('change', function(e) {
|
||||
const file = e.target.files[0];
|
||||
if (file) {
|
||||
const reader = new FileReader();
|
||||
reader.onload = function(event) {
|
||||
const img = document.getElementById('humanPreview');
|
||||
img.src = event.target.result;
|
||||
img.style.display = 'block';
|
||||
};
|
||||
reader.readAsDataURL(file);
|
||||
}
|
||||
});
|
||||
|
||||
document.getElementById('clothInput').addEventListener('change', function(e) {
|
||||
const file = e.target.files[0];
|
||||
if (file) {
|
||||
const reader = new FileReader();
|
||||
reader.onload = function(event) {
|
||||
const img = document.getElementById('clothPreview');
|
||||
img.src = event.target.result;
|
||||
img.style.display = 'block';
|
||||
};
|
||||
reader.readAsDataURL(file);
|
||||
}
|
||||
});
|
||||
|
||||
// 提交处理
|
||||
document.getElementById('submitBtn').addEventListener('click', function() {
|
||||
const humanFile = document.getElementById('humanInput').files[0];
|
||||
const clothFile = document.getElementById('clothInput').files[0];
|
||||
|
||||
if (!humanFile || !clothFile) {
|
||||
alert('请同时选择人体图片和衣服图片！');
|
||||
return;
|
||||
}
|
||||
|
||||
// 显示加载动画
|
||||
document.getElementById('loading').style.display = 'block';
|
||||
document.getElementById('result').style.display = 'none';
|
||||
|
||||
// 读取图片并转换为Base64（包含完整前缀）
|
||||
Promise.all([
|
||||
readFileAsDataURL(humanFile),
|
||||
readFileAsDataURL(clothFile)
|
||||
]).then(([humanDataURL, clothDataURL]) => {
|
||||
// 准备请求数据
|
||||
const data = {
|
||||
human_img: humanDataURL, // 包含完整前缀的Base64
|
||||
cloth_img: clothDataURL, // 包含完整前缀的Base64
|
||||
output_format: "url"
|
||||
};
|
||||
|
||||
console.log("准备发送的数据:", data); // 调试用
|
||||
|
||||
// 调用API
|
||||
fetch('http://112.126.94.241:18888/change_cloth_base64', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json'
|
||||
},
|
||||
body: JSON.stringify(data)
|
||||
})
|
||||
.then(response => {
|
||||
if (!response.ok) {
|
||||
throw new Error(`HTTP错误! 状态: ${response.status}`);
|
||||
}
|
||||
return response.json();
|
||||
})
|
||||
.then(data => {
|
||||
// 隐藏加载动画
|
||||
document.getElementById('loading').style.display = 'none';
|
||||
|
||||
// 显示结果
|
||||
const resultDiv = document.getElementById('result');
|
||||
if (data.url) {
|
||||
resultDiv.innerHTML = `
|
||||
<h3>处理结果</h3>
|
||||
<p>处理成功！点击查看结果：</p>
|
||||
<a href="${data.url}" target="_blank">${data.url}</a>
|
||||
<p><img src="${data.url}" style="max-width: 100%; margin-top: 10px;"></p>
|
||||
`;
|
||||
} else {
|
||||
resultDiv.innerHTML = `
|
||||
<h3>处理结果</h3>
|
||||
<p class="error">处理完成，但未返回预期的URL</p>
|
||||
<p>返回数据：</p>
|
||||
<pre>${JSON.stringify(data, null, 2)}</pre>
|
||||
`;
|
||||
}
|
||||
resultDiv.style.display = 'block';
|
||||
})
|
||||
.catch(error => {
|
||||
console.error('Error:', error);
|
||||
document.getElementById('loading').style.display = 'none';
|
||||
document.getElementById('result').innerHTML = `
|
||||
<h3 class="error">错误</h3>
|
||||
<p>处理过程中出现错误：${error.message}</p>
|
||||
<p>请检查控制台获取更多信息</p>
|
||||
`;
|
||||
document.getElementById('result').style.display = 'block';
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
// 辅助函数：读取文件为DataURL（包含完整前缀的Base64）
|
||||
function readFileAsDataURL(file) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const reader = new FileReader();
|
||||
reader.onload = () => resolve(reader.result);
|
||||
reader.onerror = (error) => reject(error);
|
||||
reader.readAsDataURL(file);
|
||||
});
|
||||
}
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
Reference in New Issue
Block a user