nodes/risunobushi-ComfyUI_sm4ll-Wrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update vton_api_node.py

Browse Source
This commit is contained in:
risunobushi
2025-07-02 16:31:53 +02:00
committed by GitHub
parent 016f22ef44
commit 82da008260
1 changed files with 14 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vton_api_node.py
+14 -32
View File
@@ -254,41 +254,23 @@ def call_vton_api(base_file_path, product_file_path, model_choice, base_url, ses
api_model_choice = model_mapping.get(model_choice, model_choice)
print(f"\n🎨 Calling VTON API with model: {model_choice} → {api_model_choice}")
# All models support masks - determine API structure based on mask availability
# All models use 4-parameter structure: [base, product, model, mask]
# Use user-provided mask if available, otherwise use base image as fallback
mask_to_use = mask_file_path if mask_file_path else base_file_path
api_data = {
"data": [
{"path": base_file_path, "meta": {"_type": "gradio.FileData"}},
{"path": product_file_path, "meta": {"_type": "gradio.FileData"}},
api_model_choice,
{"path": mask_to_use, "meta": {"_type": "gradio.FileData"}}
]
}
if mask_file_path:
# 4-parameter structure with mask: [base, product, model, mask]
api_data = {
"data": [
{"path": base_file_path, "meta": {"_type": "gradio.FileData"}},
{"path": product_file_path, "meta": {"_type": "gradio.FileData"}},
api_model_choice,
{"path": mask_file_path, "meta": {"_type": "gradio.FileData"}}
]
}
print(f" 🎭 Including user-provided mask in API call: {mask_file_path}")
elif api_model_choice == "dress":
# Dress model requires a mask - use base image as fallback when no mask provided
api_data = {
"data": [
{"path": base_file_path, "meta": {"_type": "gradio.FileData"}},
{"path": product_file_path, "meta": {"_type": "gradio.FileData"}},
api_model_choice,
{"path": base_file_path, "meta": {"_type": "gradio.FileData"}}
]
}
print(f" 🎭 No mask provided - using base image as fallback mask for dress model")
else:
# Eyewear/Footwear models without mask: [base, product, model]
api_data = {
"data": [
{"path": base_file_path, "meta": {"_type": "gradio.FileData"}},
{"path": product_file_path, "meta": {"_type": "gradio.FileData"}},
api_model_choice
]
}
print(f" 🎭 {api_model_choice} model - no mask provided, using 3-parameter structure")
print(f" 🎭 No mask provided - using base image as fallback mask for {api_model_choice} model")
print(f" 📤 API request data: {api_data}")