happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

webui.py

This commit is contained in:
Alen-joy47 2025-04-07 12:25:50 +05:30 committed by GitHub
parent b42590dd61
commit f140a4c552
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 29 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
webui.py
View File

@ -1,12 +1,16 @@
import gradio as gr
import os
import gradio as gr
import modules.async_worker as worker
from modules.util import load_file
from modules.config import cfg
from modules.default_values import default_prompt, default_negative_prompt
SAVE_PATH = "saved_character.txt"
def save_character(character_description):
with open(SAVE_PATH, "w", encoding="utf-8") as f:
f.write(character_description)
return f"✅ Character saved!"
return "✅ Character saved!"
def load_character():
if os.path.exists(SAVE_PATH):
@ -14,13 +18,29 @@ def load_character():
return f.read()
return ""
with gr.Blocks() as demo:
with gr.Accordion("Advanced", open=False):
character_input = gr.Textbox(label="Character Description", value=load_character())
save_button = gr.Button("Save Character")
save_output = gr.Markdown()
def start_processing(prompt, negative_prompt):
return worker.run_task(prompt, negative_prompt)
save_button.click(fn=save_character, inputs=character_input, outputs=save_output)
with gr.Blocks(css="style.css") as ui:
gr.Markdown("## 🖼️ Fooocus - Advanced UI")
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label="Prompt", value=default_prompt)
negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
generate_btn = gr.Button("Generate")
output = gr.Image(label="Result")
with gr.Column():
with gr.Accordion("Advanced", open=False):
character_input = gr.Textbox(label="Character Description", value=load_character())
save_button = gr.Button("💾 Save Character")
save_output = gr.Markdown()
save_button.click(fn=save_character, inputs=character_input, outputs=save_output)
generate_btn.click(fn=start_processing, inputs=[prompt, negative_prompt], outputs=output)
ui.launch(share=True)
demo.launch(share=True)