Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update gradio_ui.py with engine selection and multiselect support

Browse Source
This commit is contained in:
Paillat
2024-02-16 20:21:22 +01:00
parent 1d32bc38d0
commit eedbc99121
1 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
ui/gradio_ui.py
View File

@@ -24,7 +24,7 @@ class GenerateUI:
def launch_ui(self): def launch_ui(self):
ui = gr.TabbedInterface( ui = gr.TabbedInterface(
*self.get_interfaces(), "Viral Factory", "NoCrypt/miku", css=self.css *self.get_interfaces(), "Viral Factory", gr.themes.Soft(), css=self.css
) )
ui.launch() ui.launch()
@@ -43,10 +43,15 @@ class GenerateUI:
inputs = [] inputs = []
with gr.Blocks() as col1: with gr.Blocks() as col1:
for engine_type, engines in ENGINES.items(): for engine_type, engines in ENGINES.items():
multiselect = engines["multiple"]
engines = engines["classes"]
with gr.Tab(engine_type) as engine_tab: with gr.Tab(engine_type) as engine_tab:
engine_names = [engine.name for engine in engines] engine_names = [engine.name for engine in engines]
engine_dropdown = gr.Dropdown( engine_dropdown = gr.Dropdown(
choices=engine_names, value=engine_names[0] choices=engine_names,
value=engine_names[0],
multiselect=multiselect,
label="Engine provider:"
) )
inputs.append(engine_dropdown) inputs.append(engine_dropdown)
engine_rows = [] engine_rows = []
@@ -92,10 +97,19 @@ class GenerateUI:
options = {} options = {}
args = list(args) args = list(args)
for engine_type, engines in ENGINES.items(): for engine_type, engines in ENGINES.items():
engine_name = args.pop(0) engines = engines["classes"]
selected_engines = args.pop(0)
if isinstance(selected_engines, str):
selected_engines = [selected_engines]
options[engine_type] = []
# for every selected engine
for engine in engines: for engine in engines:
if engine.name == engine_name: # if it correspods to the selected engine
options[engine_type] = engine(options=args[: engine.num_options]) if engine.name in selected_engines:
# we add it to the options
options[engine_type].append(
engine(options=args[: engine.num_options])
)
args = args[engine.num_options :] args = args[engine.num_options :]
else: else:
# we don't care about this, it's not the selected engine, we throw it away # we don't care about this, it's not the selected engine, we throw it away