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

Formatting & improving imports

Browse Source
This commit is contained in:
Paillat
2024-02-23 09:50:43 +01:00
parent aa5c03b17e
commit 32b6b5c42e
40 changed files with 148 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
ui/gradio_ui.py
View File

@@ -1,10 +1,12 @@
import os
import gradio as gr
import orjson
import sys
from src.engines import ENGINES, BaseEngine
import gradio as gr
import orjson
from src.chore import GenerationContext
from src.engines import ENGINES, BaseEngine
class GenerateUI:
def __init__(self):
@@ -12,6 +14,7 @@ class GenerateUI:
font-size: 5rem !important
}
"""
def get_presets(self):
with open("local/presets.json", "r") as f:
return orjson.loads(f.read())
@@ -53,6 +56,7 @@ class GenerateUI:
def get_settings_interface(self) -> gr.Blocks:
with gr.Blocks() as interface:
reload_ui = gr.Button("Reload UI", variant="primary")
def reload():
self.ui.close()
sys.exit("Reload")
@@ -115,12 +119,13 @@ class GenerateUI:
value=None
)
preset_button = gr.Button("Load")
def load_preset(preset_name, *inputs) -> list[gr.update]:
with open("local/presets.json", "r") as f:
presets = orjson.loads(f.read())
returnable = []
if preset_name in presets.keys():
# If the preset exists
# If the preset exists
preset = presets[preset_name]
for engine_type, engines in ENGINES.items():
engines = engines["classes"]
@@ -128,7 +133,8 @@ class GenerateUI:
for engine in engines:
if engine.name in preset.get(engine_type, {}).keys():
values[0].append(engine.name)
values.extend(gr.update(value=value) for value in preset[engine_type][engine.name])
values.extend(
gr.update(value=value) for value in preset[engine_type][engine.name])
else:
values.extend(gr.update() for _ in range(engine.num_options))
returnable.extend(values)
@@ -153,7 +159,8 @@ class GenerateUI:
presets[preset_name] = new_preset
f.write(orjson.dumps(presets))
return [gr.update(value=presets.keys()), *returnable]
preset_button.click(load_preset, inputs=[preset_dropdown, *inputs], outputs=[preset_dropdown,*inputs])
preset_button.click(load_preset, inputs=[preset_dropdown, *inputs],
outputs=[preset_dropdown, *inputs])
output_gallery = gr.Markdown("aaa", render=False)
button.click(
self.run_generate_interface,
@@ -197,8 +204,8 @@ class GenerateUI:
options[engine_type].append(
engine(options=args[: engine.num_options])
)
args = args[engine.num_options :]
args = args[engine.num_options:]
else:
# we don't care about this, it's not the selected engine, we throw it away
args = args[engine.num_options :]
return options
args = args[engine.num_options:]
return options