Update app.py

app.py

@@ -17,9 +17,9 @@ from src.pix2pix_turbo import Pix2Pix_Turbo
 
 model = Pix2Pix_Turbo("sketch_to_image_stochastic")
 
-ITEMS_NAMES = [ "Lamp","Bag","Sofa","Chair","Car","Motorbike","Building"]
+ITEMS_NAMES = [ "💡 Lamp","👜 Bag","🛋️ Sofa","🪑 Chair","🏎️ Car","🏍️ Motorbike","🏠 Building", "⌚ Watch"]
 MAX_SEED = np.iinfo(np.int32).max
-DEFAULT_ITEM_NAME = "Lamp"
+DEFAULT_ITEM_NAME = "💡 Lamp"
 
 
 def pil_image_to_data_uri(img, format='PNG'):
@@ -56,7 +56,7 @@ def update_canvas(use_line, use_eraser):
         _color = "#ffffff"
         brush_size = 20
     if use_line:
-        _color = "#000011"
+        _color = "#000000"
         brush_size = 4
     return gr.update(brush_radius=brush_size, brush_color=_color, interactive=True)
 
@@ -96,18 +96,6 @@ async () => {
       return false
     }
 
-    globalThis.UNDO_SKETCH_FUNCTION = () => {
-        console.log("undo sketch function")
-        var button_undo = document.querySelector('#input_image > div.image-container.svelte-p3y7hu > div.svelte-s6ybro > button:nth-child(1)');
-        // Create a new 'click' event
-        var event = new MouseEvent('click', {
-            'view': window,
-            'bubbles': true,
-            'cancelable': true
-        });
-        button_undo.dispatchEvent(event);
-    }
-
     globalThis.DELETE_SKETCH_FUNCTION = () => {
         console.log("delete sketch function")
         var button_del = document.querySelector('#input_image > div.image-container.svelte-p3y7hu > div.svelte-s6ybro > button:nth-child(2)');
@@ -178,7 +166,6 @@ with gr.Blocks(css="style.css") as demo:
     eraser = gr.Checkbox(label="eraser", value=False, elem_id="cb-eraser")
     with gr.Row(elem_id="main_row"):
         with gr.Column(elem_id="column_input"):
-            gr.Markdown("## INPUT", elem_id="input_header")
             image = gr.Image(
                 source="canvas", tool="color-sketch", type="pil", image_mode="L",
                 invert_colors=True, shape=(512, 512), brush_radius=4, height=440, width=440,
@@ -189,16 +176,14 @@ with gr.Blocks(css="style.css") as demo:
             <div class="button-row">
                 <div id="my-div-pencil" class="pad2"> <button id="my-toggle-pencil" onclick="return togglePencil(this)"></button> </div>
                 <div id="my-div-eraser" class="pad2"> <button id="my-toggle-eraser" onclick="return toggleEraser(this)"></button> </div>
-                <div class="pad2"> <button id="my-button-undo" onclick="return UNDO_SKETCH_FUNCTION(this)"></button> </div>
                 <div class="pad2"> <button id="my-button-clear" onclick="return DELETE_SKETCH_FUNCTION(this)"></button> </div>
             </div>
             """)
             with gr.Row():
-                item = gr.Dropdown(label="Item", choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAME, scale=1)
+                item = gr.Dropdown(label="What do you want to design? 🧑‍🎨 ", choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAME, scale=1)
        
 
         with gr.Column(elem_id="column_output"):
-            gr.Markdown("## OUTPUT", elem_id="output_header")
             result = gr.Image(label="Result", height=440, width=440, elem_id="output_image", show_label=False, show_download_button=True)
             download_output = gr.Button("Download output", elem_id="download_output")