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 Scientific facts script & open folder button

Browse Source
This commit is contained in:
Paillat
2024-02-24 23:32:08 +01:00
parent ccfad9db06
commit 57e3225456
5 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
ui/gradio_ui.py
View File

@@ -1,3 +1,5 @@
import os
import gradio as gr
import orjson
@@ -169,22 +171,34 @@ class GenerateUI:
load_preset = self.get_preset_func()
preset_button.click(load_preset, inputs=[preset_dropdown, *inputs],
outputs=[preset_dropdown, *inputs])
output_gallery = gr.Markdown("aaa", render=False)
output_title = gr.Markdown(visible=True, render=False)
output_description = gr.Markdown(visible=True, render=False)
output_video = gr.Video(visible=True, render=False)
open_folder = gr.Button("📁", size="sm", variant="secondary", render=False)
output_path = gr.State(value=None)
button.click(
self.run_generate_interface,
inputs=inputs,
outputs=output_gallery,
outputs=[output_video, output_title, output_description, output_path, open_folder],
)
output_gallery.render()
with gr.Row():
with gr.Column():
output_title.render()
output_description.render()
open_folder.render()
open_folder.click(lambda x: os.system(f"open {os.path.abspath(x)}") if os.name == "posix" else os.system(f"explorer {os.abspath(x)}"), inputs=output_path)
with gr.Column():
output_video.render()
return interface
def run_generate_interface(self, progress=gr.Progress(), *args) -> gr.update:
def run_generate_interface(self, progress=gr.Progress(), *args) -> list[gr.update]:
progress(0, desc="Loading engines... 🚀")
options = self.repack_options(*args)
arguments = {name.lower(): options[name] for name in ENGINES.keys()}
ctx = GenerationContext(**arguments, progress=progress)
ctx.process() # Here we go ! 🚀
return gr.update(value=ctx.get_file_path("final.mp4"))
return [gr.update(value=ctx.get_file_path("final.mp4"), visible=True), gr.update(value=ctx.title, visible=True), gr.update(value=ctx.description, visible=True), gr.update(value=ctx.dir), gr.update(visible=True)]
def repack_options(self, *args) -> dict[str, list[BaseEngine]]:
"""