Upload folder using huggingface_hub

apps/text_tools.py

@@ -200,17 +200,18 @@ def calculate_llava_next_tokens(text, image_groups, tokenizer, max_pixels):
 
 # --- Actual UI Logic ---
 
-def run_calculation(text, model, img_data, img_max_pixels, vid_count, vid_frames, vid_w, vid_h):
-    # Construct image groups from Dataframe
+def run_calculation(text, model, img_max_pixels, vid_count, vid_frames, vid_w, vid_h, *args):
+    # Parse variable number of image group arguments
+    # args structure: c1, w1, h1, c2, w2, h2, ...
     image_groups = []
-    if img_data is not None:
-        # Handle different Dataframe formats (list of lists or pandas)
-        # Gradio usually returns list of lists if type='array' (default?)
-        # Let's assume list of lists for now, or handle pandas if needed.
-        # But wait, Gradio Dataframe `value` is list of lists.
-        for row in img_data:
+    
+    # Group args into triplets
+    for i in range(0, len(args), 3):
+        if i + 2 < len(args):
+            c = args[i]
+            w = args[i+1]
+            h = args[i+2]
             try:
-                c, w, h = row
                 c = int(c)
                 w = int(w)
                 h = int(h)
@@ -281,16 +282,53 @@ def create_ui():
             text_input = gr.Textbox(lines=5, label="输入文本 (Text)", placeholder="输入 Prompt...")
             
             with gr.Accordion("🖼️ 图片设置 (Images)", open=True):
-                img_data = gr.Dataframe(
-                    headers=["数量", "宽 (px)", "高 (px)"],
-                    datatype=["number", "number", "number"],
-                    value=[[0, 1080, 1920]],
-                    label="图片列表 (可添加多行)",
-                    col_count=(3, "fixed"),
-                    interactive=True
+                # Group 1 (Always visible)
+                with gr.Row():
+                    img_c_1 = gr.Number(value=1, label="图片数量 (Group 1)", precision=0)
+                    img_w_1 = gr.Number(value=1024, label="宽 (px)")
+                    img_h_1 = gr.Number(value=1024, label="高 (px)")
+                
+                # Group 2 (Hidden by default)
+                with gr.Row(visible=False) as group_2:
+                    img_c_2 = gr.Number(value=1, label="图片数量 (Group 2)", precision=0)
+                    img_w_2 = gr.Number(value=1024, label="宽 (px)")
+                    img_h_2 = gr.Number(value=1024, label="高 (px)")
+                    
+                # Group 3 (Hidden by default)
+                with gr.Row(visible=False) as group_3:
+                    img_c_3 = gr.Number(value=1, label="图片数量 (Group 3)", precision=0)
+                    img_w_3 = gr.Number(value=1024, label="宽 (px)")
+                    img_h_3 = gr.Number(value=1024, label="高 (px)")
+
+                # Group 4 (Hidden by default)
+                with gr.Row(visible=False) as group_4:
+                    img_c_4 = gr.Number(value=1, label="图片数量 (Group 4)", precision=0)
+                    img_w_4 = gr.Number(value=1024, label="宽 (px)")
+                    img_h_4 = gr.Number(value=1024, label="高 (px)")
+
+                add_group_btn = gr.Button("➕ 增加一组图片 (Add Group)", size="sm")
+                
+                # State to track visible groups
+                visible_groups = gr.State(1)
+
+                def add_group(curr_count):
+                    next_count = curr_count + 1
+                    # Return updates for group_2, group_3, group_4
+                    return (
+                        next_count, 
+                        gr.update(visible=next_count >= 2),
+                        gr.update(visible=next_count >= 3),
+                        gr.update(visible=next_count >= 4)
+                    )
+
+                add_group_btn.click(
+                    add_group, 
+                    [visible_groups], 
+                    [visible_groups, group_2, group_3, group_4]
                 )
+
                 with gr.Row():
-                    img_max_pixels = gr.Number(value=512*512, label="Max Pixels (最大像素限制)", precision=0)
+                    img_max_pixels = gr.Number(value=1280*1280, label="Max Pixels (最大像素限制)", precision=0)
             
             with gr.Accordion("🎥 视频设置 (Videos)", open=False):
                 with gr.Row():
@@ -313,6 +351,12 @@ def create_ui():
     
     btn.click(
         run_calculation, 
-        [text_input, model_select, img_data, img_max_pixels, vid_count, vid_frames, vid_w, vid_h], 
+        [
+            text_input, model_select, img_max_pixels, vid_count, vid_frames, vid_w, vid_h,
+            img_c_1, img_w_1, img_h_1,
+            img_c_2, img_w_2, img_h_2,
+            img_c_3, img_w_3, img_h_3,
+            img_c_4, img_w_4, img_h_4
+        ], 
         [out_json, out_file]
     )