diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py
index 251730ec..37485dfd 100644
--- a/api/onnx_web/server/api.py
+++ b/api/onnx_web/server/api.py
@@ -47,6 +47,7 @@ from .load import (
get_mask_filters,
get_network_models,
get_noise_sources,
+ get_prompt_filters,
get_source_filters,
get_upscaling_models,
get_wildcard_data,
@@ -203,10 +204,12 @@ def list_extra_strings(server: ServerContext):
def list_filters(server: ServerContext):
mask_filters = list(get_mask_filters().keys())
+ prompt_filters = list(get_prompt_filters().keys())
source_filters = list(get_source_filters().keys())
return jsonify(
{
"mask": mask_filters,
+ "prompt": prompt_filters,
"source": source_filters,
}
)
diff --git a/api/onnx_web/server/load.py b/api/onnx_web/server/load.py
index 3c76f9cc..78f875ae 100644
--- a/api/onnx_web/server/load.py
+++ b/api/onnx_web/server/load.py
@@ -83,6 +83,12 @@ source_filters = {
"segment": source_filter_segment,
"scribble": source_filter_scribble,
}
+prompt_filters = [
+ "AUTOMATIC/promptgen-lexart",
+ "AUTOMATIC/promptgen-majinai-safe",
+ "AUTOMATIC/promptgen-majinai-unsafe",
+ "Gustavosta/MagicPrompt-Stable-Diffusion",
+]
# Available ORT providers
available_platforms: List[DeviceParams] = []
@@ -148,6 +154,10 @@ def get_noise_sources():
return noise_sources
+def get_prompt_filters():
+ return prompt_filters
+
+
def get_source_filters():
return source_filters
diff --git a/gui/src/components/tab/Img2Img.tsx b/gui/src/components/tab/Img2Img.tsx
index 86472090..0a9488cb 100644
--- a/gui/src/components/tab/Img2Img.tsx
+++ b/gui/src/components/tab/Img2Img.tsx
@@ -20,6 +20,7 @@ import { UpscaleControl } from '../control/UpscaleControl.js';
import { ImageInput } from '../input/ImageInput.js';
import { NumericField } from '../input/NumericField.js';
import { QueryList } from '../input/QueryList.js';
+import { ExperimentalControl } from '../control/ExperimentalControl.js';
export function Img2Img() {
const { params } = mustExist(useContext(ConfigContext));
@@ -140,6 +141,10 @@ export function Img2Img() {
+ {
+ // eslint-disable-next-line no-console
+ console.log('setting experimental props', props);
+ }} />
+ {
+ // eslint-disable-next-line no-console
+ console.log('setting experimental props', props);
+ }} />
+ {
+ // eslint-disable-next-line no-console
+ console.log('setting experimental props', props);
+ }} />