Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import os
 import json
 import tempfile
@@ -46,7 +47,6 @@ def _new_session(
         cfg=cfg.to_dict(),
         seed=int(seed),
     )
-    # record root event (step 0) as a receipt too
     rec.record_event(state=state, action="RESET")
     return {"cfg": cfg, "state": state, "rec": rec}
 
@@ -91,7 +91,6 @@ def ui_step(sess: Optional[Dict[str, Any]]):
         world, pov, status = _render(state)
         return sess, world, pov, status + "  (done)", _receipt_tail(rec)
 
-    # Agent decides action internally (simple deterministic heuristic in sim)
     new_state, action = step_sim(cfg, state)
     sess["state"] = new_state
     rec.record_event(state=new_state, action=action)
@@ -153,14 +152,14 @@ def ui_verify(zip_file):
     return json.dumps(report, indent=2)
 
 
-def ui_replay(zip_file, export_gif: bool, watermark: bool):
+def ui_replay(zip_file, export_gif_flag: bool, watermark: bool):
     if zip_file is None:
         return None, None, "Upload a receipt bundle ZIP."
 
     zip_path = zip_file if isinstance(zip_file, str) else zip_file.name
     replay = replay_from_receipt_bundle_zip(
         zip_path,
-        export_gif=bool(export_gif),
+        export_gif_flag=bool(export_gif_flag),
         watermark=bool(watermark),
     )
 
@@ -220,7 +219,7 @@ with gr.Blocks(title=APP_TITLE) as demo:
         with gr.Tab("Replay"):
             replay_upload = gr.File(label="Upload receipt bundle ZIP")
             with gr.Row():
-                replay_export_gif = gr.Checkbox(value=True, label="Export replay GIF")
+                replay_export_gif_flag = gr.Checkbox(value=True, label="Export replay GIF")
                 replay_watermark = gr.Checkbox(value=True, label="Watermark final hash on frames")
             btn_replay = gr.Button("Replay From Bundle", variant="primary")
             with gr.Row():
@@ -229,7 +228,6 @@ with gr.Blocks(title=APP_TITLE) as demo:
             replay_gif = gr.File(label="Replayed GIF (if exported)")
             replay_msg = gr.Textbox(label="Replay result", interactive=False)
 
-    # Wiring
     btn_new.click(
         ui_new,
         inputs=[seed, size, walls_pct, coins, hazards, pov_radius],
@@ -268,7 +266,7 @@ with gr.Blocks(title=APP_TITLE) as demo:
 
     btn_replay.click(
         ui_replay,
-        inputs=[replay_upload, replay_export_gif, replay_watermark],
+        inputs=[replay_upload, replay_export_gif_flag, replay_watermark],
         outputs=[replay_world, replay_pov, replay_gif, replay_msg],
     )