sontungdo/sd-image-editor
1
0
Fork 0
mirror of https://github.com/sontungdo/sd-image-editor.git synced 2026-01-26 19:29:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor changes

Browse Source
This commit is contained in:
sontungdo
2024-04-07 17:40:03 -04:00
parent b5d834b436
commit 11fc9c4e22
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/sd-image-editor.py
View File

@@ -64,7 +64,7 @@ def draw_bbox(img, crop_enabled, bbox_w, bbox_h, bbox_center_x, bbox_center_y):
return img
def store_original_input(img):
def store_image(img):
return img
@@ -117,8 +117,6 @@ def save_image(img):
def open_folder():
# adopted from https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/20123d427b09901396133643be78f6b692393b0c/modules/util.py#L176-L208
"""Open a folder in the file manager of the respect OS."""
# import at function level to avoid potential issues
import gradio as gr
import platform
import sys
import subprocess
@@ -163,7 +161,8 @@ def on_ui_tabs():
tool=None,
image_mode="RGBA",
height=500)
init_img = gr.Image(label="Output image",
init_img = gr.Image(label="Unedited image",
elem_id="original_image",
height=500,
type="pil",
image_mode="RGBA",
@@ -292,8 +291,8 @@ def on_ui_tabs():
color_slider, contrast_slider, brightness_slider, sharpness_slider]
bbox_inputs = [crop_enabled, bbox_w, bbox_h, bbox_center_x, bbox_center_y]
# I/O
input_img.upload(store_original_input, inputs=[input_img], outputs=[init_img]) # Store persistent copy of the initial uploaded image in init_img
input_img.clear(store_original_input, inputs=[input_img], outputs=[init_img])
input_img.upload(store_image, inputs=[input_img], outputs=[init_img]) # Store persistent copy of the initial uploaded image in init_img
input_img.clear(store_image, inputs=[input_img], outputs=[init_img])
init_img.change(edit, inputs=[init_img, *control_inputs], outputs=[output_img])
render_button.click(edit, inputs=[init_img, *control_inputs], outputs=[output_img])