From 444369e95090a3eb0927ec8f8299501148512346 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Fri, 16 Feb 2024 22:32:29 -0600
Subject: [PATCH] add mock controls for experimental features

---
 api/onnx_web/server/api.py         |  3 +++
 api/onnx_web/server/load.py        | 10 ++++++++++
 gui/src/components/tab/Img2Img.tsx |  5 +++++
 gui/src/components/tab/Inpaint.tsx |  5 +++++
 gui/src/components/tab/Txt2Img.tsx |  5 +++++
 gui/src/types/api.ts               |  1 +
 6 files changed, 29 insertions(+)

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() {
       </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'
diff --git a/gui/src/components/tab/Inpaint.tsx b/gui/src/components/tab/Inpaint.tsx
index af022b1f..0ee2a8a6 100644
--- a/gui/src/components/tab/Inpaint.tsx
+++ b/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'
diff --git a/gui/src/components/tab/Txt2Img.tsx b/gui/src/components/tab/Txt2Img.tsx
index b0b04c7f..0936efa4 100644
--- a/gui/src/components/tab/Txt2Img.tsx
+++ b/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()}
diff --git a/gui/src/types/api.ts b/gui/src/types/api.ts
index ebf9e76d..f16cb6fd 100644
--- a/gui/src/types/api.ts
+++ b/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>;
 }