soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Add button to send generated image to svd tab (#38)

Browse Source
This commit is contained in:
Chenlei Hu 2024-02-04 23:22:56 +00:00 committed by GitHub
parent 2954a36c18
commit 8117d7fcee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
extensions-builtin/sd_forge_svd/scripts/forge_svd.py
View File

@ -3,6 +3,7 @@ import gradio as gr
import os
import pathlib
import modules.infotext_utils as parameters_copypaste
from modules import script_callbacks
from modules.paths import models_path
from modules.ui_common import ToolButton, refresh_symbol
@ -106,6 +107,12 @@ def on_ui_tabs():
visible=True, height=1024, columns=4)
generate_button.click(predict, inputs=ctrls, outputs=[output_gallery, output_video])
PasteField = parameters_copypaste.PasteField
paste_fields = [
PasteField(width, "Size-1", api="width"),
PasteField(height, "Size-2", api="height"),
]
parameters_copypaste.add_paste_fields("svd", init_img=input_image, fields=paste_fields)
return [(svd_block, "SVD", "svd")]

5
javascript/ui.js
View File

@ -82,6 +82,11 @@ function switch_to_extras() {
return Array.from(arguments);
}
function switch_to_svd() {
gradioApp().querySelector('#tabs').querySelectorAll('button')[6].click();
return Array.from(arguments);
}
function get_tab_index(tabId) {
let buttons = gradioApp().getElementById(tabId).querySelector('div').querySelectorAll('button');
for (let i = 0; i < buttons.length; i++) {

3
modules/ui_common.py
View File

@ -206,7 +206,8 @@ Requested path was: {f}
buttons = {
'img2img': ToolButton('🖼️', elem_id=f'{tabname}_send_to_img2img', tooltip="Send image and generation parameters to img2img tab."),
'inpaint': ToolButton('🎨️', elem_id=f'{tabname}_send_to_inpaint', tooltip="Send image and generation parameters to img2img inpaint tab."),
'extras': ToolButton('📐', elem_id=f'{tabname}_send_to_extras', tooltip="Send image and generation parameters to extras tab.")
'extras': ToolButton('📐', elem_id=f'{tabname}_send_to_extras', tooltip="Send image and generation parameters to extras tab."),
'svd': ToolButton('🎬', elem_id=f'{tabname}_send_to_svd', tooltip="Send image and generation parameters to SVD tab."),
}
if tabname == 'txt2img':