fix: count_df

src/modules/app.py

@@ -1,4 +1,5 @@
 import re
+import math
 import itertools
 import pandas as pd
 import gradio as gr
@@ -75,9 +76,11 @@ with gr.Blocks() as demo:
 
                 headers = ['ImageId', 'PosX', 'PosY', 'PosPx', 'PosPy', 'SizeX', 'SizeY', 'SizePx', 'SizePy']
 
-                @gr.render(inputs=[count_url, cm_img_input, idv_img_input])
-                def update_df(*items):
-                    image_count = sum(len(item) for item in items if item is not None)
+                @gr.render(inputs=[count_url,  idv_img_input])
+                def update_df(*items): # TODO
+                    items = 1 if items is None else items
+                    image_count = math.prod(len(item) for item in items if item is not None)
+
                     if image_count is not None:
                         positions = calculate_layout(image_count)
                     df = pd.DataFrame(positions, columns=headers)
@@ -103,7 +106,8 @@ with gr.Blocks() as demo:
             ['demo/demo.jpg'],
             ['demo/logo.png']
             ],
-            [{'files': ['demo/demo.md', 'demo/demo2.md']},
+            # [{'files': ['demo/demo.md', 'demo/demo2.md']},
+            [{'files': ['demo/demo.md']},
             ['demo/demo.jpg'],
             ['demo/img/number-1.png', 'demo/img/number-2.png', 'demo/img/number-3.png', 'demo/img/number-4.png', 'demo/img/number-5.png', ]
             ]