Update app.py

app.py

@@ -208,6 +208,12 @@ def getSesh(seshid):
 def setSesh(seshid, sesh):
     all_sesh[seshid] = sesh
 
+default_img = np.zeros((200,200,3)).astype(np.uint8)
+default_img[:] = 64
+
+Palette_img = np.zeros((200,200,3)).astype(np.uint8)
+Palette_img[:] = 128
+
 def submitPhoto(img, rooms, seshid):
     print('submitPhoto reset rooms: ', rooms, seshid)
     seshid = str(time.time()) if not seshid or seshid == 'SESHID' else seshid
@@ -1707,7 +1713,8 @@ with gr.Blocks(theme=gr.themes.Monochrome(), js=js_func) as demo:
         num_of_items = gr.Label(label='Total')
     with gr.Row():
         with gr.Column(scale=5) as items_display:
-            room_items = gr.Gallery([((sesh.Palette_img if item == 'Palette' else sesh.default_img ) if item in ['Palette', 'Living Room', 'Bedroom'] else room_item_placeholder_images_dict['LIVING ROOM FURNITURE'][item], item) for item in living_layout_new], label='Room Items', columns=5, height=400, interactive=True)
+            #room_items = gr.Gallery([((sesh.Palette_img if item == 'Palette' else sesh.default_img ) if item in ['Palette', 'Living Room', 'Bedroom'] else room_item_placeholder_images_dict['LIVING ROOM FURNITURE'][item], item) for item in living_layout_new], label='Room Items', columns=5, height=400, interactive=True)
+            room_items = gr.Gallery([((Palette_img if item == 'Palette' else default_img ) if item in ['Palette', 'Living Room', 'Bedroom'] else room_item_placeholder_images_dict['LIVING ROOM FURNITURE'][item], item) for item in living_layout_new], label='Room Items', columns=5, height=400, interactive=True)
         with gr.Column(scale=1, visible=False) as style_side:
             #dsdf = gr.Image("C:\\Users\\Wayne\\Downloads\\agent-first.gif", show_label=False, show_download_button=False)
             close1 = gr.Button('Close')