Update app.py

app.py

@@ -21,12 +21,11 @@ ITEMS_NAMES = [ "💡 Lamp","👜 Bag","🛋️ Sofa","🪑 Chair","🏎️ Car"
 MAX_SEED = np.iinfo(np.int32).max
 DEFAULT_ITEM_NAME = "💡 Lamp"
 COLORS = ["blue", "red", "black", "yellow", "green", "pink", "grey"]
-SELECTED_COLOR = random.choice(COLORS)
+DEFAULT_COLOR = "yellow"
 def empty_input_image():
-    SELECTED_COLOR = random.choice(COLORS)
-    return { 'background': Image.new("L", (512, 512), 255),
+    return [{ 'background': Image.new("L", (512, 512), 255),
              'layers': [Image.new("L", (512, 512), 255),Image.new("L", (512, 512), 255)],
-             'composite': Image.new("L", (512, 512), 255)}
+             'composite': Image.new("L", (512, 512), 255)}, random.choice(COLORS)]
 
 def pil_image_to_data_uri(img, format='PNG'):
     buffered = BytesIO()
@@ -35,7 +34,7 @@ def pil_image_to_data_uri(img, format='PNG'):
     return f"data:image/{format.lower()};base64,{img_str}"
 
 
-def run(image, item_name):
+def run(image, item_name, color_name):
     print("sketch updated")
     print(image)
     empty_image = Image.new("L", (512, 512), 255)
@@ -45,7 +44,7 @@ def run(image, item_name):
         ones = empty_image
         return ones
     print(item_name.split()[1])
-    prompt = SELECTED_COLOR + " " + item_name.split()[1] + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
+    prompt = color_name + " " + item_name.split()[1] + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
     inverted_image = ImageOps.invert(image["composite"])
     converted_image = inverted_image.convert("RGB")
     image_t = TF.to_tensor(converted_image) > 0.5
@@ -195,14 +194,16 @@ with gr.Blocks(css="style.css", head = head) as demo:
                     </button>""",elem_id="eraser_container")
             with gr.Group(elem_id="output_image_container", elem_classes="image_container"):
                 result = gr.Image(label="Result",  height="100%", width="100%", elem_id="output_image", show_label=False, show_download_button=True,container=False,)
+                color = gr.Radio(choices=COLORS, value=DEFAULT_COLOR, show_label=False, container=False, visible=False)
     with gr.Row(elem_id="radio_row"):
-        item = gr.Radio(choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAME, show_label=False, container=False)
+        item = gr.Radio(choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAMEDEFAULT_COLOR, show_label=False, container=False)
 
     demo.load(None,None,None,js=scripts)
-    inputs = [image, item]
+    inputs = [image, item, color]
     outputs = [result]
     item.change(fn=run, inputs=inputs, outputs=outputs)
+    color.change(fn=run, inputs=inputs, outputs=outputs)
     image.change(fn=run, inputs=inputs, outputs=outputs, trigger_mode="always_last")
-    image.clear(fn=empty_input_image, outputs=image)
+    image.clear(fn=empty_input_image, outputs=[image , color])
 if __name__ == "__main__":
     demo.queue().launch(debug=True, allowed_paths=["."])