IgorSlinko commited on
Commit
1ae03ef
Β·
1 Parent(s): d9831c7

Add validation: Start must be < End, default End=30

Browse files

- Validate that Start < End (strictly less, not equal)
- Show error message if validation fails
- Change default End value from 50 to 30

Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -70,7 +70,7 @@ def get_routed_steps(total_steps: int, strategy: str, params: dict) -> set:
70
 
71
  elif strategy == "Replace part of trajectory":
72
  mode = params.get("mode", "Percentages")
73
- start, end = parse_start_end(params.get("start", 0), params.get("end", 50), total_steps, mode)
74
  routed = set(range(start, min(end, total_steps)))
75
 
76
  return routed
@@ -1313,7 +1313,7 @@ def build_app():
1313
  )
1314
  with gr.Row():
1315
  start_step_1 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
1316
- end_step_1 = gr.Number(label="End", value=50, minimum=0, precision=0, interactive=True)
1317
 
1318
  add_model_2_btn = gr.Button("+ Add another model", size="sm", visible=False)
1319
 
@@ -1349,7 +1349,7 @@ def build_app():
1349
  )
1350
  with gr.Row():
1351
  start_step_2 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
1352
- end_step_2 = gr.Number(label="End", value=50, minimum=0, precision=0, interactive=True)
1353
 
1354
  add_model_3_btn = gr.Button("+ Add another model", size="sm", visible=False)
1355
 
@@ -1385,7 +1385,7 @@ def build_app():
1385
  )
1386
  with gr.Row():
1387
  start_step_3 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
1388
- end_step_3 = gr.Number(label="End", value=50, minimum=0, precision=0, interactive=True)
1389
 
1390
  gr.Markdown("---")
1391
  route_btn = gr.Button("πŸš€ Let's ROUTE!!", variant="primary", size="lg", interactive=False)
@@ -1587,6 +1587,13 @@ def build_app():
1587
  strategy_params["mode"] = part_mode_1_val
1588
  strategy_params["start"] = start_1_val
1589
  strategy_params["end"] = end_1_val
 
 
 
 
 
 
 
1590
 
1591
  total_base_tokens = {"uncached_input": 0, "cache_read": 0, "cache_creation": 0, "completion": 0}
1592
  total_routing_tokens = {"uncached_input": 0, "cache_read": 0, "cache_creation": 0, "completion": 0}
 
70
 
71
  elif strategy == "Replace part of trajectory":
72
  mode = params.get("mode", "Percentages")
73
+ start, end = parse_start_end(params.get("start", 0), params.get("end", 30), total_steps, mode)
74
  routed = set(range(start, min(end, total_steps)))
75
 
76
  return routed
 
1313
  )
1314
  with gr.Row():
1315
  start_step_1 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
1316
+ end_step_1 = gr.Number(label="End", value=30, minimum=0, precision=0, interactive=True)
1317
 
1318
  add_model_2_btn = gr.Button("+ Add another model", size="sm", visible=False)
1319
 
 
1349
  )
1350
  with gr.Row():
1351
  start_step_2 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
1352
+ end_step_2 = gr.Number(label="End", value=30, minimum=0, precision=0, interactive=True)
1353
 
1354
  add_model_3_btn = gr.Button("+ Add another model", size="sm", visible=False)
1355
 
 
1385
  )
1386
  with gr.Row():
1387
  start_step_3 = gr.Number(label="Start", value=0, minimum=0, precision=0, interactive=True)
1388
+ end_step_3 = gr.Number(label="End", value=30, minimum=0, precision=0, interactive=True)
1389
 
1390
  gr.Markdown("---")
1391
  route_btn = gr.Button("πŸš€ Let's ROUTE!!", variant="primary", size="lg", interactive=False)
 
1587
  strategy_params["mode"] = part_mode_1_val
1588
  strategy_params["start"] = start_1_val
1589
  strategy_params["end"] = end_1_val
1590
+ if start_1_val >= end_1_val:
1591
+ yield (
1592
+ gr.update(visible=True, value="❌ Start must be less than End"),
1593
+ gr.update(visible=False),
1594
+ None, None,
1595
+ )
1596
+ return
1597
 
1598
  total_base_tokens = {"uncached_input": 0, "cache_read": 0, "cache_creation": 0, "completion": 0}
1599
  total_routing_tokens = {"uncached_input": 0, "cache_read": 0, "cache_creation": 0, "completion": 0}