transparant background

app.py

@@ -100,15 +100,18 @@ def make_prompt_row():
 def make_sidebar():
     with st.sidebar:
         input_image = st.file_uploader("", type=["png", "jpg"], key='input_image', on_change=on_upload)
-        generation_mode = st.selectbox("Generation mode", ["Segmentation conditioning", "Inpainting"], on_change=on_change_radio)
+        generation_mode = st.selectbox("Generation mode", ["Re-generate objects",
+                                                           "Segmentation conditioning",
+                                                           "Inpainting"], on_change=on_change_radio)
 
-        paint_mode = st.sidebar.selectbox("Painting mode", ("freedraw", "polygon"))
-        if paint_mode == "freedraw":
-            brush = st.slider("Stroke width", 5, 140, 100, key='slider_seg')
-        else:
-            brush = 5
 
         if generation_mode == "Segmentation conditioning":
+            paint_mode = st.sidebar.selectbox("Painting mode", ("freedraw", "polygon"))
+            if paint_mode == "freedraw":
+                brush = st.slider("Stroke width", 5, 140, 100, key='slider_seg')
+            else:
+                brush = 5
+    
             category_chooser = st.sidebar.selectbox("Filter on category", list(
                 COLOR_MAPPING_CATEGORY.keys()), index=0, key='category_chooser')
 
@@ -117,7 +120,19 @@ def make_sidebar():
             color_chooser = st.sidebar.selectbox(
                 "Choose a color", chosen_colors, index=0, format_func=map_colors, key='color_chooser'
             )
+
+        elif generation_mode == "Re-generate objects":
+            color_chooser = "rgba(0, 0, 0, 0.0)"
+            paint_mode = 'freedraw'
+            brush = 0
+
         else:
+            paint_mode = st.sidebar.selectbox("Painting mode", ("freedraw", "polygon"))
+            if paint_mode == "freedraw":
+                brush = st.slider("Stroke width", 5, 140, 100, key='slider_seg')
+            else:
+                brush = 5
+
             color_chooser = "#000000"
     return input_image, generation_mode, brush, color_chooser, paint_mode
 
@@ -147,7 +162,7 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
         _reset_state=_reset_state
     )
 
-    if generation_mode == "Segmentation conditioning":
+    if generation_mode == "Re-generate objects":
         canvas = st_canvas(
             **canvas_dict,
         )
@@ -168,9 +183,8 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
             options=st.session_state['unique_colors'],
             key='chosen_colors',
             default=st.session_state['unique_colors'],
-            format_func=map_colors_rgb
+            format_func=map_colors_rgb,
         )
-        print(st.session_state['unique_colors'])
 
         if st.button("generate image", key='generate_button'):
             image = get_image()
@@ -181,7 +195,6 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
             for color in chosen_colors:
                 # if the color is in the segmentation, set mask to 1
                 mask[np.where((segmentation == color).all(axis=2))] = 1
-            print(mask)
 
             with st.spinner(text="Generating image"):
                 result_image = make_image_controlnet(image=image,