Update app.py

app.py

@@ -90,6 +90,30 @@ def load_labels(df: pd.DataFrame):
     return tag_names, rating_indexes, general_indexes, character_indexes
 
 
+def add_substitute_row(current):
+    """
+    Append an empty [original, substitute] row to the substitutes dataframe.
+    Works with type='array' (list of lists).
+    """
+    if current is None:
+        current = []
+    # Make sure we have a plain list of lists
+    current = list(current)
+    current.append(["", ""])
+    return current
+
+
+def add_exclusion_row(current):
+    """
+    Append an empty [tag] row to the exclusions dataframe.
+    """
+    if current is None:
+        current = []
+    current = list(current)
+    current.append([""])
+    return current
+
+
 class VideoTagger:
     """
     Wraps a WD-style ONNX model and tag metadata,
@@ -499,46 +523,61 @@ with gr.Blocks(title=TITLE) as demo:
                     combined_tags = gr.Textbox(
                         label="Combined Unique Tags (All Frames)",
                         lines=6,
-                        buttons=list['copy'],
+                        buttons=["copy"],
                     )
                     debug_info = gr.JSON(
                         label="Details / Debug Info",
                     )
 
-        # ---------------- TAB 2: TAG CONTROL ----------------
-        with gr.Tab("Tag Control"):
-            gr.Markdown("### Tag Substitutes")
-            gr.Markdown(
-                "Add rows where **Original Tag** will be replaced by **Substitute Tag** "
-                "in the final combined output (after all frames are processed)."
-            )
-
-            tag_substitutes_df = gr.Dataframe(
-                headers=["Original Tag", "Substitute Tag"],
-                datatype=["str", "str"],
-                row_count=1,
-                row_limits=(1, None),
-                column_count=2,
-                type="array",
-                label="Tag Substitutes",
-                interactive=True,
-            )
-
-            gr.Markdown("### Tag Exclusions")
-            gr.Markdown(
-                "Add tags that should be **removed entirely** from the final combined output."
-            )
-
-            tag_exclusions_df = gr.Dataframe(
-                headers=["Tag to Exclude"],
-                datatype=["str"],
-                row_count=1,
-                row_limits=(1, None),
-                column_count=1,
-                type="array",
-                label="Tag Exclusions",
-                interactive=True,
-            )
+                # ---------------- TAB 2: TAG CONTROL ----------------
+                with gr.Tab("Tag Control"):
+                    gr.Markdown("### Tag Substitutes")
+                    gr.Markdown(
+                        "Add rows where **Original Tag** will be replaced by **Substitute Tag** "
+                        "in the final combined output (after all frames are processed)."
+                    )
+        
+                    with gr.Row():
+                        tag_substitutes_df = gr.Dataframe(
+                            headers=["Original Tag", "Substitute Tag"],
+                            datatype=["str", "str"],
+                            row_count=1,          # just start with 1 empty row
+                            column_count=2,
+                            type="array",
+                            label="Tag Substitutes",
+                            interactive=True,
+                        )
+                        add_sub_row_btn = gr.Button("➕ Add substitute", scale=0)
+        
+                    gr.Markdown("### Tag Exclusions")
+                    gr.Markdown(
+                        "Add tags that should be **removed entirely** from the final combined output."
+                    )
+        
+                    with gr.Row():
+                        tag_exclusions_df = gr.Dataframe(
+                            headers=["Tag to Exclude"],
+                            datatype=["str"],
+                            row_count=1,          # start with 1 row
+                            column_count=1,
+                            type="array",
+                            label="Tag Exclusions",
+                            interactive=True,
+                        )
+                        add_ex_row_btn = gr.Button("➕ Add exclusion", scale=0)
+
+    add_sub_row_btn.click(
+        fn=add_substitute_row,
+        inputs=tag_substitutes_df,
+        outputs=tag_substitutes_df,
+    )
+
+    add_ex_row_btn.click(
+        fn=add_exclusion_row,
+        inputs=tag_exclusions_df,
+        outputs=tag_exclusions_df,
+    )
+
 
     # Wiring the button AFTER all components are defined
     run_button.click(