1580158
/
sd-webui-txt-img-to-3d-model
mirror of https://github.com/jtydhr88/sd-webui-txt-img-to-3d-model.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ui improve - only show prompt or init image when it is needed

master
jtydhr88 2023-07-06 23:42:17 -04:00
parent e36269e8f6
commit 33bde577a4
2 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
javascript/lazyload/txt-img-to-3d-mode.js
View File

@ -90,6 +90,22 @@ let _r = 0;
container.appendChild(text2);
}
let tiTo3DModelPromptDiv = document.getElementById('tiTo3DModelPrompt');
tiTo3DModelPromptDiv.style.display = 'none';
let tiTo3DModelInitImageDiv = document.getElementById('tiTo3DModelInitImage');
window.modeChangeTITo3DModel = function(mode) {
if (mode === 'img') {
tiTo3DModelPromptDiv.style.display = 'none';
tiTo3DModelInitImageDiv.style.display = 'block';
}
else {
tiTo3DModelPromptDiv.style.display = 'block';
tiTo3DModelInitImageDiv.style.display = 'none';
}
};
function to_gradio(v) {
return [v, _r++];

7
scripts/main.py
View File

@ -125,9 +125,9 @@ def on_ui_tabs():
with gr.Blocks(analytics_enabled=False) as shap_e:
with gr.Row():
with gr.Column():
mode = gr.Dropdown(["txt", "img"], label="Mode", value="img")
prompt_txt = gr.Textbox(label="Prompt", lines=2)
init_image = gr.Image(type="pil")
mode = gr.Dropdown(["txt", "img"], label="Mode", value="img", elem_id="tiTo3DModelMode")
prompt_txt = gr.Textbox(label="Prompt", lines=2, elem_id="tiTo3DModelPrompt")
init_image = gr.Image(type="pil", elem_id="tiTo3DModelInitImage")
batch_size_slider = gr.Slider(minimum=1, maximum=10, default=2, value=2, step=1, label="Batch Size",
interactive=True)
use_karras = gr.Checkbox(label="Use Karras", value=True)
@ -165,6 +165,7 @@ def on_ui_tabs():
sink = gr.HTML(value='', visible=False) # to suppress error in javascript
jscall('getGeneratedHistory', get_generated_history, id, js, sink)
mode.change(None, mode, None, _js="modeChangeTITo3DModel")
btn.click(fn=generate,
inputs=[mode, batch_size_slider, prompt_txt, use_karras, karras_steps_slider, init_image,