jonigata commited on
Commit
32eb5ca
1 Parent(s): 0700b92

add remove background

Browse files
Files changed (2) hide show
  1. js/poseMaker.js +6 -0
  2. main.py +8 -2
js/poseMaker.js CHANGED
@@ -807,6 +807,12 @@ function savePose() {
807
  }
808
 
809
  function importBackground(image) {
 
 
 
 
 
 
810
  let m = new Image();
811
  m.src = image;
812
  m.onload = function() {
 
807
  }
808
 
809
  function importBackground(image) {
810
+ if (image == null) {
811
+ canvasBg = null;
812
+ Redraw();
813
+ return;
814
+ }
815
+
816
  let m = new Image();
817
  m.src = image;
818
  m.onload = function() {
main.py CHANGED
@@ -57,6 +57,8 @@ with gr.Blocks(css="""button { min-width: 80px; }""") as demo:
57
  importBtn = gr.Button(value="Import")
58
  with gr.Column(min_width=80):
59
  bgBtn = gr.Button(value="Background")
 
 
60
  with gr.Accordion(label="Json", open=False):
61
  with gr.Row():
62
  with gr.Column(min_width=80):
@@ -137,8 +139,12 @@ When using Q, X, C, R, pressing and dont release until the operation is complete
137
  fn = None,
138
  inputs = [source],
139
  outputs = [],
140
- _js="(image) => { importBackground(image); return []; }"
141
- )
 
 
 
 
142
 
143
  saveBtn.click(
144
  fn = None,
 
57
  importBtn = gr.Button(value="Import")
58
  with gr.Column(min_width=80):
59
  bgBtn = gr.Button(value="Background")
60
+ with gr.Column(min_width=80):
61
+ removeBgBtn = gr.Button(value="RemoveBG")
62
  with gr.Accordion(label="Json", open=False):
63
  with gr.Row():
64
  with gr.Column(min_width=80):
 
139
  fn = None,
140
  inputs = [source],
141
  outputs = [],
142
+ _js="(image) => { importBackground(image); return []; }")
143
+ removeBgBtn.click(
144
+ fn = None,
145
+ inputs = [],
146
+ outputs = [],
147
+ _js="() => { importBackground(null); return []; }")
148
 
149
  saveBtn.click(
150
  fn = None,