Skip to content

Commit

Permalink
✨ 支持保存上一次提示词
Browse files Browse the repository at this point in the history
  • Loading branch information
zhulinyv committed Oct 10, 2024
1 parent d841135 commit bb3a478
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 10 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -23,3 +23,4 @@ rife-v4.13-lite
rife-v4.14
plugins
run_stand_alone_scripts.bat
start.json
32 changes: 22 additions & 10 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@


def main():
import os
from pathlib import Path

from src.batch_inpaint import for_webui as inpaint
Expand Down Expand Up @@ -57,6 +58,17 @@ def main():

webui_language = read_json(f"./files/languages/{env.webui_lang}/webui.json")

default_positive_input = (
webui_language["example"]["positive"]
if not os.path.exists("start.json")
else read_json("start.json")["positive"]
)
default_negative_input = (
webui_language["example"]["negative"]
if not os.path.exists("start.json")
else read_json("start.json")["negative"]
)

# ------------------------------ #

def open_output_folder_block(output_folder):
Expand Down Expand Up @@ -85,13 +97,13 @@ def open_output_folder_block(output_folder):
with gr.Column():
with gr.Column(scale=3):
text2image_positive_input = gr.Textbox(
value=webui_language["example"]["positive"],
value=default_positive_input,
lines=2,
label=webui_language["t2i"]["positive"],
)
with gr.Row():
text2image_negative_input = gr.Textbox(
value=webui_language["example"]["negative"],
value=default_negative_input,
lines=2,
label=webui_language["t2i"]["negative"],
scale=3,
Expand Down Expand Up @@ -416,13 +428,13 @@ def open_output_folder_block(output_folder):
with gr.Column():
with gr.Column(scale=3):
vibe_transfer_positive_input = gr.Textbox(
value=webui_language["example"]["positive"],
value=default_positive_input,
lines=2,
label=webui_language["t2i"]["positive"],
)
with gr.Row():
vibe_transfer_negative_input = gr.Textbox(
value=webui_language["example"]["negative"],
value=default_negative_input,
lines=2,
label=webui_language["t2i"]["negative"],
scale=3,
Expand Down Expand Up @@ -520,13 +532,13 @@ def open_output_folder_block(output_folder):
with gr.Column():
with gr.Column():
image2image_positive_input = gr.Textbox(
value=webui_language["example"]["positive"],
value=default_positive_input,
lines=2,
label=webui_language["t2i"]["positive"],
)
with gr.Row():
image2image_negative_input = gr.Textbox(
value=webui_language["example"]["negative"],
value=default_negative_input,
lines=3,
label=webui_language["t2i"]["negative"],
scale=3,
Expand Down Expand Up @@ -686,7 +698,7 @@ def open_output_folder_block(output_folder):
scale=1,
)
movie2movie_negative = gr.Textbox(
webui_language["example"]["negative"],
default_negative_input,
label=webui_language["t2i"]["negative"],
lines=2,
)
Expand Down Expand Up @@ -780,7 +792,7 @@ def open_output_folder_block(output_folder):
tiled_upscale_img_path = gr.Textbox(value=None, label=webui_language["tile"]["img_path"])
tiled_upscale_positive_input = gr.Textbox("", lines=2, label=webui_language["tile"]["positive"])
tiled_upscale_negative_input = gr.Textbox(
webui_language["example"]["negative"],
default_negative_input,
label="负面提示词",
lines=3,
)
Expand Down Expand Up @@ -837,13 +849,13 @@ def open_output_folder_block(output_folder):
with gr.Column():
with gr.Column():
inpaint_positive_input = gr.Textbox(
value=webui_language["example"]["positive"],
value=default_positive_input,
lines=2,
label=webui_language["t2i"]["positive"],
)
with gr.Row():
inpaint_negative_input = gr.Textbox(
value=webui_language["example"]["negative"],
value=default_negative_input,
lines=3,
label=webui_language["t2i"]["negative"],
scale=3,
Expand Down
3 changes: 3 additions & 0 deletions utils/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,9 @@ def generate_image(json_data):
Returns:
(bytes): 二进制图片
"""
with open("start.json", "w") as f:
json.dump({"positive": json_data["input"], "negative": json_data["parameters"]["negative_prompt"]}, f)

try:
rep = requests.post(
"https://image.novelai.net/ai/generate-image", json=json_data, headers=headers, proxies=proxies
Expand Down

0 comments on commit bb3a478

Please sign in to comment.