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

Add AssetsEngine and related files

Browse Source
This commit is contained in:
Paillat
2024-02-18 00:56:49 +01:00
parent e3229518d4
commit 6c48e19af0
9 changed files with 219 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
ui/gradio_ui.py
View File

@@ -13,10 +13,12 @@ class GenerateUI:
"""
def get_switcher_func(self, engine_names: list[str]) -> list[gr.update]:
def switch(selected: str):
def switch(selected: str | list[str]):
if isinstance(selected, str):
selected = [selected]
returnable = []
for i, name in enumerate(engine_names):
returnable.append(gr.update(visible=name == selected))
returnable.append(gr.update(visible=name in selected))
return returnable
@@ -51,14 +53,15 @@ class GenerateUI:
choices=engine_names,
value=engine_names[0],
multiselect=multiselect,
label="Engine provider:"
label="Engine provider:" if not multiselect else "Engine providers:",
)
inputs.append(engine_dropdown)
engine_rows = []
for i, engine in enumerate(engines):
with gr.Row(
equal_height=True, visible=(i == 0)
with gr.Group(
visible=(i == 0)
) as engine_row:
gr.Label(engine.name)
engine_rows.append(engine_row)
options = engine.get_options()
inputs.extend(options)