P01yH3dr0n commited on
Commit
719bb3d
1 Parent(s): 151b498

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -73,16 +73,15 @@ def control_ui():
73
  use_inp = gr.Checkbox(label='启用', value=False)
74
  overlay = gr.Checkbox(label='覆盖原图', value=True)
75
  inp_img = gr.ImageEditor(label="上传图片", value=None, sources=["upload"], interactive=True, type="pil", eraser=False, transforms=None, brush="imagemask")
76
- reuse_img_inp = gr.Button(value='使用上一次生成的图片')
77
  with gr.Column():
78
  gr.Textbox(value=get_count, label='Usage count', every=10)
79
- save = gr.Checkbox(value=True, label='Always save all generated images')
80
 
81
  gen_btn = gr.Button(value="Generate", variant="primary")
82
  rand_seed.click(fn=lambda: -1, inputs=None, outputs=seed)
83
  width.change(lambda w, h: h if w*h<=1024*1024 else (1024*1024//w//64)*64, [width, height], height)
84
  height.change(lambda w, h: w if w*h<=1024*1024 else (1024*1024//h//64)*64, [width, height], width)
85
- return gen_btn,[prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str, inp_img, overlay], [save, rand_seed, reuse_seed, reuse_img_vibe, reuse_img_inp, use_inp]
86
 
87
 
88
  def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str, inp_img, overlay, save, use_inp):
@@ -116,10 +115,9 @@ def main_ui():
116
  gen_btn, paras, others = control_ui()
117
  with gr.Column():
118
  image, info = preview_ui()
119
- gen_btn.click(generate, paras + [others[0], others[5]], [image, info])
120
  others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
121
  others[3].click(lambda i: i, inputs=image, outputs=paras[14])
122
- others[4].click(lambda i, d: {'background': i, 'layers':d['layers'], 'composite': i}, inputs=[image, paras[17]], outputs=paras[17])
123
  return page, paras[:14]
124
 
125
 
 
73
  use_inp = gr.Checkbox(label='启用', value=False)
74
  overlay = gr.Checkbox(label='覆盖原图', value=True)
75
  inp_img = gr.ImageEditor(label="上传图片", value=None, sources=["upload"], interactive=True, type="pil", eraser=False, transforms=None, brush="imagemask")
 
76
  with gr.Column():
77
  gr.Textbox(value=get_count, label='Usage count', every=10)
78
+ save = gr.Checkbox(value=False, label='Always save all generated images', visible=False)
79
 
80
  gen_btn = gr.Button(value="Generate", variant="primary")
81
  rand_seed.click(fn=lambda: -1, inputs=None, outputs=seed)
82
  width.change(lambda w, h: h if w*h<=1024*1024 else (1024*1024//w//64)*64, [width, height], height)
83
  height.change(lambda w, h: w if w*h<=1024*1024 else (1024*1024//h//64)*64, [width, height], width)
84
+ return gen_btn,[prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str, inp_img, overlay], [save, rand_seed, reuse_seed, reuse_img_vibe, use_inp]
85
 
86
 
87
  def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_image, info_extract, ref_str, inp_img, overlay, save, use_inp):
 
115
  gen_btn, paras, others = control_ui()
116
  with gr.Column():
117
  image, info = preview_ui()
118
+ gen_btn.click(generate, paras + [others[0], others[4]], [image, info])
119
  others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
120
  others[3].click(lambda i: i, inputs=image, outputs=paras[14])
 
121
  return page, paras[:14]
122
 
123