add mock controls for experimental features

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,
         }
     )

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
 

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() {
       </Stack>
       <HighresControl selectHighres={selectHighres} setHighres={setHighres} />
       <UpscaleControl selectUpscale={selectUpscale} setUpscale={setUpscale} />
+      <ExperimentalControl setExperimental={(props) => {
+        // eslint-disable-next-line no-console
+        console.log('setting experimental props', props);
+      }} />
       <Button
         disabled={doesExist(source) === false}
         variant='contained'

gui/src/components/tab/Inpaint.tsx

@@ -22,6 +22,7 @@ import { ImageInput } from '../input/ImageInput.js';
 import { MaskCanvas } from '../input/MaskCanvas.js';
 import { NumericField } from '../input/NumericField.js';
 import { QueryList } from '../input/QueryList.js';
+import { ExperimentalControl } from '../control/ExperimentalControl.js';
 
 export function Inpaint() {
   const { params } = mustExist(useContext(ConfigContext));
@@ -222,6 +223,10 @@ export function Inpaint() {
       <OutpaintControl />
       <HighresControl selectHighres={selectHighres} setHighres={setHighres} />
       <UpscaleControl selectUpscale={selectUpscale} setUpscale={setUpscale} />
+      <ExperimentalControl setExperimental={(props) => {
+        // eslint-disable-next-line no-console
+        console.log('setting experimental props', props);
+      }} />
       <Button
         disabled={preventInpaint()}
         variant='contained'

gui/src/components/tab/Txt2Img.tsx

@@ -20,6 +20,7 @@ import { ModelControl } from '../control/ModelControl.js';
 import { UpscaleControl } from '../control/UpscaleControl.js';
 import { VariableControl } from '../control/VariableControl.js';
 import { NumericField } from '../input/NumericField.js';
+import { ExperimentalControl } from '../control/ExperimentalControl.js';
 
 export function SizeControl() {
   const { params } = mustExist(useContext(ConfigContext));
@@ -106,6 +107,10 @@ export function Txt2Img() {
       <HighresControl selectHighres={selectHighres} setHighres={setHighres} />
       <UpscaleControl selectUpscale={selectUpscale} setUpscale={setUpscale} />
       <VariableControl selectGrid={selectVariable} setGrid={setVariable} />
+      <ExperimentalControl setExperimental={(props) => {
+        // eslint-disable-next-line no-console
+        console.log('setting experimental props', props);
+      }} />
       <Button
         variant='contained'
         onClick={() => generate.mutate()}

gui/src/types/api.ts

@@ -44,6 +44,7 @@ export type NetworkModel = EmbeddingNetwork | LoraNetwork | ControlNetwork;
 
 export interface FilterResponse {
   mask: Array<string>;
+  prompt: Array<string>;
   source: Array<string>;
 }