Skip to content

Commit 44db6e6

Browse files
abrichrclaude
andcommitted
fix: regenerate suggested steps after task restart
When the user presses 'R' to restart a task, the QEMU hard reset produces a new stable screenshot, but the suggested steps were not regenerated. The stale steps from the previous screenshot were displayed. Now _generate_steps() is called again with the fresh screenshot after every restart. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 6d0a3fd commit 44db6e6

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

scripts/record_waa_demos.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -882,6 +882,16 @@ def _hard_reset_task_env() -> bytes:
882882
step_idx = 0
883883
print(f"\n VNC: {vnc_url}")
884884
print(f" Task: {instruction}\n")
885+
886+
# Re-generate steps from the new stable screenshot
887+
print(" Generating suggested steps...")
888+
suggested = _generate_steps(before_png, instruction, task_config)
889+
print()
890+
print(" ┌─ SUGGESTED STEPS ──────────────────────────────")
891+
for line in suggested.splitlines():
892+
print(f" │ {line}")
893+
print(" └────────────────────────────────────────────────")
894+
print()
885895
print(" Task restarted. Continue recording.\n")
886896
continue
887897

0 commit comments

Comments
 (0)