Spaces:
Running
Running
Add more edit models and fix img2video mode display bug
Browse files- Added FLUX Pro/Dev Edit, Dreamina Edit, Qwen Edit to img2img models
- Fixed bug where img2video section didn't show (backend-section wrapper)
- Organized edit model dropdown with optgroups
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
src/content_engine/api/ui.html
CHANGED
|
@@ -828,11 +828,13 @@ select { cursor: pointer; }
|
|
| 828 |
<div class="chip" onclick="selectMode(this, 'img2video')">Image to Video</div>
|
| 829 |
</div>
|
| 830 |
|
| 831 |
-
<div
|
| 832 |
-
|
| 833 |
-
<div class="
|
| 834 |
-
|
| 835 |
-
|
|
|
|
|
|
|
| 836 |
</div>
|
| 837 |
|
| 838 |
<div id="cloud-model-select" style="display:none">
|
|
@@ -866,10 +868,22 @@ select { cursor: pointer; }
|
|
| 866 |
<div id="cloud-edit-model-select" style="display:none">
|
| 867 |
<label>Cloud Edit Model (Image-to-Image)</label>
|
| 868 |
<select id="gen-cloud-edit-model">
|
| 869 |
-
<
|
| 870 |
-
|
| 871 |
-
|
| 872 |
-
<
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 873 |
</select>
|
| 874 |
<div style="font-size:11px;color:var(--text-secondary);margin-top:4px">
|
| 875 |
Upload a reference photo and describe what you want. The model preserves identity, face, and pose while applying your prompt.
|
|
@@ -1687,8 +1701,8 @@ function selectMode(chip, mode) {
|
|
| 1687 |
selectedMode = mode;
|
| 1688 |
document.getElementById('img2img-section').style.display = mode === 'img2img' ? '' : 'none';
|
| 1689 |
document.getElementById('img2video-section').style.display = mode === 'img2video' ? '' : 'none';
|
| 1690 |
-
// Hide regular backend
|
| 1691 |
-
const backendSection = document.getElementById('backend-
|
| 1692 |
if (backendSection) {
|
| 1693 |
backendSection.style.display = mode === 'img2video' ? 'none' : '';
|
| 1694 |
}
|
|
|
|
| 828 |
<div class="chip" onclick="selectMode(this, 'img2video')">Image to Video</div>
|
| 829 |
</div>
|
| 830 |
|
| 831 |
+
<div id="backend-section">
|
| 832 |
+
<div class="section-title">Backend</div>
|
| 833 |
+
<div class="chips" id="backend-chips">
|
| 834 |
+
<div class="chip" onclick="selectBackend(this, 'local')">Local GPU</div>
|
| 835 |
+
<div class="chip selected" onclick="selectBackend(this, 'pod')">RunPod GPU</div>
|
| 836 |
+
<div class="chip" onclick="selectBackend(this, 'cloud')">Cloud API</div>
|
| 837 |
+
</div>
|
| 838 |
</div>
|
| 839 |
|
| 840 |
<div id="cloud-model-select" style="display:none">
|
|
|
|
| 868 |
<div id="cloud-edit-model-select" style="display:none">
|
| 869 |
<label>Cloud Edit Model (Image-to-Image)</label>
|
| 870 |
<select id="gen-cloud-edit-model">
|
| 871 |
+
<optgroup label="SeeDream Edit (ByteDance)">
|
| 872 |
+
<option value="seedream-4.5-edit">SeeDream v4.5 Edit (Best Quality)</option>
|
| 873 |
+
<option value="seedream-4-edit">SeeDream v4 Edit</option>
|
| 874 |
+
</optgroup>
|
| 875 |
+
<optgroup label="NanoBanana Edit (Google)">
|
| 876 |
+
<option value="nano-banana-pro-edit">NanoBanana Pro Edit (Best Consistency)</option>
|
| 877 |
+
<option value="nano-banana-edit">NanoBanana Edit</option>
|
| 878 |
+
</optgroup>
|
| 879 |
+
<optgroup label="FLUX Edit (Black Forest Labs)">
|
| 880 |
+
<option value="flux-pro-edit">FLUX Pro Edit (Highest Quality)</option>
|
| 881 |
+
<option value="flux-dev-edit">FLUX Dev Edit</option>
|
| 882 |
+
</optgroup>
|
| 883 |
+
<optgroup label="Other">
|
| 884 |
+
<option value="dreamina-3-edit">Dreamina v3 Edit</option>
|
| 885 |
+
<option value="qwen-edit">Qwen Edit</option>
|
| 886 |
+
</optgroup>
|
| 887 |
</select>
|
| 888 |
<div style="font-size:11px;color:var(--text-secondary);margin-top:4px">
|
| 889 |
Upload a reference photo and describe what you want. The model preserves identity, face, and pose while applying your prompt.
|
|
|
|
| 1701 |
selectedMode = mode;
|
| 1702 |
document.getElementById('img2img-section').style.display = mode === 'img2img' ? '' : 'none';
|
| 1703 |
document.getElementById('img2video-section').style.display = mode === 'img2video' ? '' : 'none';
|
| 1704 |
+
// Hide regular backend section for video mode (video has its own backend selector)
|
| 1705 |
+
const backendSection = document.getElementById('backend-section');
|
| 1706 |
if (backendSection) {
|
| 1707 |
backendSection.style.display = mode === 'img2video' ? 'none' : '';
|
| 1708 |
}
|
src/content_engine/services/cloud_providers/wavespeed_provider.py
CHANGED
|
@@ -82,10 +82,19 @@ VIDEO_MODEL_MAP = {
|
|
| 82 |
|
| 83 |
# Map friendly names to WaveSpeed edit model API paths
|
| 84 |
EDIT_MODEL_MAP = {
|
|
|
|
| 85 |
"seedream-4.5-edit": "bytedance/seedream-v4.5/edit",
|
| 86 |
"seedream-4-edit": "bytedance/seedream-v4/edit",
|
|
|
|
| 87 |
"nano-banana-edit": "google/nano-banana/edit",
|
| 88 |
"nano-banana-pro-edit": "google/nano-banana-pro/edit",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
# Default edit model
|
| 90 |
"default": "bytedance/seedream-v4.5/edit",
|
| 91 |
}
|
|
|
|
| 82 |
|
| 83 |
# Map friendly names to WaveSpeed edit model API paths
|
| 84 |
EDIT_MODEL_MAP = {
|
| 85 |
+
# SeeDream Edit (ByteDance)
|
| 86 |
"seedream-4.5-edit": "bytedance/seedream-v4.5/edit",
|
| 87 |
"seedream-4-edit": "bytedance/seedream-v4/edit",
|
| 88 |
+
# NanoBanana Edit (Google)
|
| 89 |
"nano-banana-edit": "google/nano-banana/edit",
|
| 90 |
"nano-banana-pro-edit": "google/nano-banana-pro/edit",
|
| 91 |
+
# FLUX Edit (Black Forest Labs)
|
| 92 |
+
"flux-dev-edit": "black-forest-labs/flux-dev/edit",
|
| 93 |
+
"flux-pro-edit": "black-forest-labs/flux-pro/edit",
|
| 94 |
+
# Dreamina Edit (ByteDance)
|
| 95 |
+
"dreamina-3-edit": "bytedance/dreamina-v3/edit",
|
| 96 |
+
# Qwen Edit (WaveSpeed)
|
| 97 |
+
"qwen-edit": "wavespeedai/qwen-image/edit",
|
| 98 |
# Default edit model
|
| 99 |
"default": "bytedance/seedream-v4.5/edit",
|
| 100 |
}
|