Move routing charts to end + add Indexes/Percentages toggle

- Move routing_plots_row to the end (after Tokens/Cost per Trajectory)
- Add Radio button for 'Replace part of trajectory': Indexes vs Percentages
- Indexes mode: start/end are step numbers (0-based)
- Percentages mode: start/end are percentages (0-100)
- Default: Percentages with start=0, end=50

app.py

@@ -29,19 +29,22 @@ _calculated_tokens_cache = {}
 _trajectory_steps_cache = {}
 
 
-def parse_step_or_ratio(value: float, total_steps: int) -> int:
+def parse_start_end(start: float, end: float, total_steps: int, mode: str) -> tuple[int, int]:
     """
-    Parse a value as either step number or ratio.
+    Parse start and end values based on mode.
 
-    If value is integer (e.g., 3.0, 5.0) -> treat as step number
-    If value is float with decimal (e.g., 0.5, 0.25) -> treat as ratio of total_steps
+    Args:
+        start: start value
+        end: end value
+        total_steps: total number of steps in trajectory
+        mode: "Indexes" or "Percentages"
 
-    Returns: step index (0-based)
+    Returns: (start_idx, end_idx) - both 0-based
     """
-    if value == int(value) and value >= 1:
-        return int(value)
+    if mode == "Indexes":
+        return int(start), int(end)
     else:
-        return int(value * total_steps)
+        return int(start * total_steps / 100), int(end * total_steps / 100)
 
 
 def get_routed_steps(total_steps: int, strategy: str, params: dict) -> set:
@@ -66,8 +69,8 @@ def get_routed_steps(total_steps: int, strategy: str, params: dict) -> set:
             routed = set(range(0, total_steps, k))
 
     elif strategy == "Replace part of trajectory":
-        start = parse_step_or_ratio(params.get("start", 0), total_steps)
-        end = parse_step_or_ratio(params.get("end", 0.5), total_steps)
+        mode = params.get("mode", "Percentages")
+        start, end = parse_start_end(params.get("start", 0), params.get("end", 50), total_steps, mode)
         routed = set(range(start, min(end, total_steps)))
 
     return routed
@@ -1221,14 +1224,14 @@ def build_app():
                         plot_tokens = gr.Plot(label="Token Usage by Type")
                         plot_tokens_cost = gr.Plot(label="Cost by Token Type ($)")
 
-                    with gr.Row(visible=False) as routing_plots_row:
-                        routing_tokens_plot = gr.Plot(label="Tokens by Type (per Model)")
-                        routing_cost_plot = gr.Plot(label="Cost by Type (per Model)")
-
                     with gr.Row():
                         plot_stacked = gr.Plot(label="Tokens per Trajectory")
                         plot_cost_breakdown = gr.Plot(label="Cost per Trajectory ($)")
 
+                    with gr.Row(visible=False) as routing_plots_row:
+                        routing_tokens_plot = gr.Plot(label="Tokens by Type (per Model)")
+                        routing_cost_plot = gr.Plot(label="Cost by Type (per Model)")
+
             with gr.Column(scale=1):
                 selected_folder = gr.State("")
                 gr.Markdown("### Selected Model")
@@ -1302,9 +1305,15 @@ def build_app():
                                 random_pct_1 = gr.Number(label="Percentage (%)", value=50, minimum=0, maximum=100, precision=0, interactive=True)
                             with gr.Row(visible=False) as every_k_params_1:
                                 step_k_1 = gr.Number(label="k", value=2, minimum=1, precision=0, interactive=True)
-                            with gr.Row(visible=False) as part_params_1:
-                                start_step_1 = gr.Number(label="Start (int=step; 0,0-1,0=ratio)", value=0, minimum=0, precision=2, interactive=True)
-                                end_step_1 = gr.Number(label="End (int=step; 0,0-1,0=ratio)", value=0.5, minimum=0, precision=2, interactive=True)
+                            with gr.Column(visible=False) as part_params_1:
+                                part_mode_1 = gr.Radio(
+                                    choices=["Indexes", "Percentages"],
+                                    value="Percentages",
+                                    label="Mode",
+                                )
+                                with gr.Row():
+                                    start_step_1 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
+                                    end_step_1 = gr.Number(label="End", value=50, minimum=0, precision=0, interactive=True)
 
                         add_model_2_btn = gr.Button("+ Add another model", size="sm", visible=False)
 
@@ -1332,9 +1341,15 @@ def build_app():
                                     random_pct_2 = gr.Number(label="Percentage (%)", value=50, minimum=0, maximum=100, precision=0, interactive=True)
                                 with gr.Row(visible=False) as every_k_params_2:
                                     step_k_2 = gr.Number(label="k", value=2, minimum=1, precision=0, interactive=True)
-                                with gr.Row(visible=False) as part_params_2:
-                                    start_step_2 = gr.Number(label="Start (int=step; 0,0-1,0=ratio)", value=0, minimum=0, precision=2, interactive=True)
-                                    end_step_2 = gr.Number(label="End (int=step; 0,0-1,0=ratio)", value=0.5, minimum=0, precision=2, interactive=True)
+                                with gr.Column(visible=False) as part_params_2:
+                                    part_mode_2 = gr.Radio(
+                                        choices=["Indexes", "Percentages"],
+                                        value="Percentages",
+                                        label="Mode",
+                                    )
+                                    with gr.Row():
+                                        start_step_2 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
+                                        end_step_2 = gr.Number(label="End", value=50, minimum=0, precision=0, interactive=True)
 
                             add_model_3_btn = gr.Button("+ Add another model", size="sm", visible=False)
 
@@ -1362,9 +1377,15 @@ def build_app():
                                     random_pct_3 = gr.Number(label="Percentage (%)", value=50, minimum=0, maximum=100, precision=0, interactive=True)
                                 with gr.Row(visible=False) as every_k_params_3:
                                     step_k_3 = gr.Number(label="k", value=2, minimum=1, precision=0, interactive=True)
-                                with gr.Row(visible=False) as part_params_3:
-                                    start_step_3 = gr.Number(label="Start (int=step; 0,0-1,0=ratio)", value=0, minimum=0, precision=2, interactive=True)
-                                    end_step_3 = gr.Number(label="End (int=step; 0,0-1,0=ratio)", value=0.5, minimum=0, precision=2, interactive=True)
+                                with gr.Column(visible=False) as part_params_3:
+                                    part_mode_3 = gr.Radio(
+                                        choices=["Indexes", "Percentages"],
+                                        value="Percentages",
+                                        label="Mode",
+                                    )
+                                    with gr.Row():
+                                        start_step_3 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
+                                        end_step_3 = gr.Number(label="End", value=50, minimum=0, precision=0, interactive=True)
 
                     gr.Markdown("---")
                     route_btn = gr.Button("🚀 Let's ROUTE!!", variant="primary", size="lg", interactive=False)
@@ -1500,7 +1521,7 @@ def build_app():
             state_data,
             base_input, base_cache_read, base_cache_creation, base_completion,
             routing_model_1_val, r1_input, r1_cache_read, r1_cache_creation, r1_completion,
-            strategy_1_val, random_pct_1_val, step_k_1_val, start_1_val, end_1_val,
+            strategy_1_val, random_pct_1_val, step_k_1_val, part_mode_1_val, start_1_val, end_1_val,
             source, overhead, with_cache
         ):
             if state_data is None:
@@ -1563,6 +1584,7 @@ def build_app():
             elif strategy_1_val == "Replace every step k":
                 strategy_params["k"] = step_k_1_val
             elif strategy_1_val == "Replace part of trajectory":
+                strategy_params["mode"] = part_mode_1_val
                 strategy_params["start"] = start_1_val
                 strategy_params["end"] = end_1_val
 
@@ -1688,7 +1710,7 @@ def build_app():
                 trajectories_state,
                 price_input, price_cache_read, price_cache_creation, price_completion,
                 routing_model_1, routing_price_1_input, routing_price_1_cache_read, routing_price_1_cache_creation, routing_price_1_completion,
-                strategy_1, random_pct_1, step_k_1, start_step_1, end_step_1,
+                strategy_1, random_pct_1, step_k_1, part_mode_1, start_step_1, end_step_1,
                 token_source, thinking_overhead, use_cache,
             ],
             outputs=[routing_result, routing_plots_row, routing_tokens_plot, routing_cost_plot],