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

Refactor UI code to add settings interface

Browse Source
This commit is contained in:
Paillat
2024-02-20 14:54:25 +01:00
parent 49857b7be3
commit f6e4fa6bd3
1 changed files with 31 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
ui/gradio_ui.py
View File

@@ -4,7 +4,6 @@ import gradio as gr
from src.engines import ENGINES, BaseEngine
from src.chore import GenerationContext
class GenerateUI:
def __init__(self):
self.css = """.generate_button {
@@ -37,13 +36,25 @@ class GenerateUI:
Returns:
tuple[list[gr.Blocks], list[str]]: A tuple containing a list of gr.Blocks interfaces and a list of interface names.
"""
return ([self.get_generate_interface()], ["Generate"])
return (
[self.get_generate_interface(), self.get_settings_interface()],
["Generate", "Settings"],
)
def get_settings_interface(self) -> gr.Blocks:
with gr.Blocks() as interface:
for engine_type, engines in ENGINES.items():
engines = engines["classes"]
with gr.Tab(engine_type) as engine_tab:
for engine in engines:
engine.get_settings()
return interface
def get_generate_interface(self) -> gr.Blocks:
with gr.Blocks() as interface:
with gr.Row(equal_height=False) as row:
inputs = []
with gr.Blocks() as col1:
with gr.Column(scale=2) as col1:
for engine_type, engines in ENGINES.items():
multiselect = engines["multiple"]
show_dropdown = engines.get("show_dropdown", True)
@@ -54,16 +65,20 @@ class GenerateUI:
choices=engine_names,
value=engine_names[0],
multiselect=multiselect,
label="Engine provider:" if not multiselect else "Engine providers:",
label="Engine provider:"
if not multiselect
else "Engine providers:",
visible=show_dropdown,
)
inputs.append(engine_dropdown)
engine_rows = []
for i, engine in enumerate(engines):
with gr.Group(
visible=(i == 0)
) as engine_row:
gr.Markdown(value = " ", label=f"{engine.name}", show_label=True)
with gr.Group(visible=(i == 0)) as engine_row:
gr.Markdown(
value=" ",
label=f"{engine.name}",
show_label=True,
)
engine_rows.append(engine_row)
options = engine.get_options()
inputs.extend(options)
@@ -72,23 +87,25 @@ class GenerateUI:
switcher, inputs=engine_dropdown, outputs=engine_rows
)
with gr.Blocks() as col2:
with gr.Column() as col2:
button = gr.Button(
"🚀",
scale=0.33,
size="lg",
variant="primary",
elem_classes="generate_button",
)
button.click(self.run_generate_interface, inputs=inputs)
output_gallery = gr.Markdown("aaa", render=False)
button.click(self.run_generate_interface, inputs=inputs, outputs=output_gallery)
output_gallery.render()
return interface
def run_generate_interface(self, *args) -> None:
def run_generate_interface(self, progress=gr.Progress(), *args) -> gr.update:
progress(0, desc="Loading engines... 🚀")
options = self.repack_options(*args)
arugments = {name.lower(): options[name] for name in ENGINES.keys()}
ctx = GenerationContext(**arugments)
ctx = GenerationContext(**arugments, progress=progress)
ctx.process() # Here we go ! 🚀
return gr.update(value=ctx.get_file_path("final.mp4"))
def repack_options(self, *args) -> dict[BaseEngine]:
"""
Repacks the options provided as arguments into a dictionary based on the selected engine.