Skip to content

Commit

Permalink
adds support for a notification.mp3 in the root directory that will p…
Browse files Browse the repository at this point in the history
…lay upon completion (fixes AUTOMATIC1111#1013)
  • Loading branch information
Connum committed Sep 26, 2022
1 parent 78d6aef commit 2846ca5
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 4 deletions.
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -19,4 +19,5 @@ __pycache__
/webui-user.sh
/interrogate
/user.css
/.idea
/.idea
notification.mp3
3 changes: 3 additions & 0 deletions javascript/notification.js
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,9 @@ onUiUpdate(function(){

lastHeadImg = headImg;

// play notification sound if available
gradioApp().querySelector('#audio_notification audio')?.play();

if (document.hasFocus()) return;

// Multiple copies of the images are in the DOM when one is selected. Dedup with a Set to get the real number generated.
Expand Down
9 changes: 6 additions & 3 deletions modules/ui.py
Original file line number Diff line number Diff line change
Expand Up @@ -564,13 +564,13 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo):
with gr.TabItem('Inpaint', id='inpaint'):
init_img_with_mask = gr.Image(label="Image for inpainting with mask", show_label=False, elem_id="img2maskimg", source="upload", interactive=True, type="pil", tool="sketch", image_mode="RGBA")

init_img_inpaint = gr.Image(label="Image for img2img", show_label=False, source="upload", interactive=True, type="pil", visible=False)
init_mask_inpaint = gr.Image(label="Mask", source="upload", interactive=True, type="pil", visible=False)
init_img_inpaint = gr.Image(label="Image for img2img", show_label=False, source="upload", interactive=True, type="pil", visible=False, elem_id="img_inpaint_base")
init_mask_inpaint = gr.Image(label="Mask", source="upload", interactive=True, type="pil", visible=False, elem_id="img_inpaint_mask")

mask_blur = gr.Slider(label='Mask blur', minimum=0, maximum=64, step=1, value=4)

with gr.Row():
mask_mode = gr.Radio(label="Mask mode", show_label=False, choices=["Draw mask", "Upload mask"], type="index", value="Draw mask")
mask_mode = gr.Radio(label="Mask mode", show_label=False, choices=["Draw mask", "Upload mask"], type="index", value="Draw mask", elem_id="mask_mode")
inpainting_mask_invert = gr.Radio(label='Masking mode', show_label=False, choices=['Inpaint masked', 'Inpaint not masked'], value='Inpaint masked', type="index")

inpainting_fill = gr.Radio(label='Masked content', choices=['fill', 'original', 'latent noise', 'latent nothing'], value='fill', type="index")
Expand Down Expand Up @@ -970,6 +970,9 @@ def run_settings(*args):
for interface, label, ifid in interfaces:
with gr.TabItem(label, id=ifid):
interface.render()

if os.path.exists(os.path.join(script_path, "notification.mp3")):
audio_notification = gr.Audio(interactive=False, value=os.path.join(script_path, "notification.mp3"), elem_id="audio_notification", visible=False)

text_settings = gr.Textbox(elem_id="settings_json", value=lambda: opts.dumpjson(), visible=False)
settings_submit.click(
Expand Down

0 comments on commit 2846ca5

Please sign in to comment.