Update app.py

app.py

@@ -272,7 +272,7 @@ class Session():
 
 sesh = Session()
 
-def submitPhoto(img):
+def submitPhoto(img, rooms):
     sesh.current_img_signature = str(img.sum())
     sesh.img_data[sesh.current_img_signature] = img
     sesh.Palettes[sesh.current_img_signature] = {}
@@ -283,7 +283,9 @@ def submitPhoto(img):
     sesh.Palette_img = np.zeros((200,200,3)).astype(np.uint8)
     sesh.Palette_img[:] = 128
     sesh.items_layouts[sesh.current_img_signature] = bed_layout_new 
-    return [(v,str(k)) for k,v in sesh.img_data.items()], gr.Image(label='Upload Room Photo')
+    rooms.append((img, sesh.current_img_signature))
+    #return [(v,str(k)) for k,v in sesh.img_data.items()], gr.Image(label='Upload Room Photo
+    return rooms, gr.Image(label='Upload Room Photo')
 
 def summarisePerStore():
     sesh.summary_joined = {}
@@ -2584,7 +2586,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), js=js_func) as demo:
     close4.click(closeSidePanel, None, [items_display, item_side, room_side, colors_side, merchant, replace, ads, store_summary])
  
 
-    upload_photo.click(submitPhoto, upload, [choose_room, upload])
+    upload_photo.click(submitPhoto, [upload, choose_room], [choose_room, upload])
     choose_room.select(makePalettes, None, [color_Palettes, room2decorate, room_items])
 
     #upload_photo.click(submitPhoto, upload, [working_room, upload])