Update app.py

app.py

@@ -547,7 +547,7 @@ def updateItemDisplay(seshid, budget=False, pallete_change=False):
         sesh.updated_display = realised_display 
         summarisePerStore(seshid)
 
-        sesh.product_pd_counts = [(product,pd_count) for store in sesh.summary_joined for product,pd_count in sesh.summary_joined[store].items()]
+        """sesh.product_pd_counts = [(product,pd_count) for store in sesh.summary_joined for product,pd_count in sesh.summary_joined[store].items()]
         urls = [product_pd_count[1]['product_dict']['href'] for product_pd_count in sesh.product_pd_counts]
         sesh.responses = asyncio.run(main(urls))
         item_locator_data = {data[1]: sesh.item_data_[sesh.current_img_signature][sesh.room_type][sesh.room_style][sesh.price][sesh.Palette_ids[sesh.current_img_signature]][data[1]][data[0]] for data in sesh.display_items_[sesh.current_img_signature] if data[1] not in ['Palette', 'Living Room', 'Bedroom']}
@@ -559,7 +559,24 @@ def updateItemDisplay(seshid, budget=False, pallete_change=False):
             if tuple(idxs) not in sesh.locator_genid: sesh.locator_genid[tuple(idxs)] = gen_uid = str(time.time())
         json_data = [json.dumps({'room_type': sesh.room_type, 'item_locator_data': item_locator_data[product_pd_count[0]], 'gen_uid': sesh.locator_genid[tuple(item_locator_data[product_pd_count[0]])], 'response':responses[i], 'item': product_pd_count[0]}) for i,product_pd_count in enumerate(sesh.product_pd_counts) if product_pd_count[0] not in ['Palette', 'Living Room', 'Bedroom']]
         responses = asyncio.run(main_post(urls, headers, json_data, timeout=.1))
-        print('dump_text_gen: ', responses)
+        print('dump_text_gen: ', responses)"""
+
+        
+        item_locator_data = content['item_locator_data']
+        responses = content['responses']
+        items = content['items']
+        number_of_items = min(max(content['number_of_items'], 3), 7)
+        room_type = content['room_type']
+        video_format = content['video_format']
+
+        formats = ['Youtube Long-Form', 'Youtube Short-Form', 'Facebook Long-Form', 'Instagram Reel', 'Tiktok', 'LinkedIn']
+        number_of_items = [7, 3, 7, 3, 3, 7]
+        urls = [sesh.item_data[img][sesh.item_data_keys[img]['room_type']][sesh.item_data_keys[img]['room_style']][sesh.item_data_keys[img]['price']][sesh.item_data_keys[img]['Paletteid']][itemidx_item[1]][itemidx_item[0]]['href'] for itemidx_item in sesh.display_items_[img]]]
+        sesh.responses = asyncio.run(main(urls))
+        responses = [r.decode('ISO-8859-1') if type(r) != type(None) else r for r in sesh.responses]
+        item_locator_data = [sesh.item_data_[sesh.current_img_signature][sesh.room_type][sesh.room_style][sesh.price][sesh.Palette_ids[sesh.current_img_signature]][data[1]][data[0]] for data in sesh.display_items_[sesh.current_img_signature] if data[1] not in ['Palette', 'Living Room', 'Bedroom']]
+        items = [data[1] for data in sesh.display_items_[sesh.current_img_signature] if data[1] not in ['Palette', 'Living Room', 'Bedroom']]]
+        json_data = [json.dumps({'video_format': fmt, 'number_of_items': number_of_items[i], 'room_type': sesh.room_type, 'item_locator_data': item_locator_data, 'gen_uid': sesh.locator_genid[tuple(item_locator_data[product_pd_count[0]])], 'responses':responses, 'items': items}) for i,fmt in enumerate(formats)]
 
         setSesh(seshid, sesh)
         return gr.Gallery(realised_display, label='Room Items', columns=5)