Update app.py

app.py

@@ -94,15 +94,6 @@ def _j(obj) -> str:
 
 
 def ui_quickstart_run(sign_all: bool, current_sk: str, current_pk: str):
-    """
-    One-click guided demo:
-    - optionally generate + use keys
-    - start session
-    - append a realistic sequence of events
-    - verify
-    - finalise + export
-    - fill the other tabs automatically
-    """
     status_lines = []
 
     # Decide keys
@@ -125,18 +116,20 @@ def ui_quickstart_run(sign_all: bool, current_sk: str, current_pk: str):
         sign_start=bool(sign_all),
         sk_hex=sk_hex,
     )
+
     if not sid:
         status_lines.append(f"[FAIL] start_session: {start_msg}")
-        # Keep UI safe: don’t overwrite fields on failure
+        quick_msg = "\n".join(status_lines)
         return (
             (sk_hex if sign_all else current_sk),
             (pk_hex if sign_all else current_pk),
             "", "", "", "", "",  # session ids fanout
-            "\n".join(status_lines),
+            quick_msg,            # start_status
             [], "No timeline.",
             "FAIL", False, "Quickstart failed at start_session.",
             None, "Not finalised.",
             None, "No export.",
+            quick_msg,            # quick_status
         )
 
     status_lines.append(f"[OK] Started session: {sid}")
@@ -178,9 +171,8 @@ def ui_quickstart_run(sign_all: bool, current_sk: str, current_pk: str):
         pk_hex=(pk_hex if sign_all else ""),
         require_signatures=bool(sign_all),
     )
-    status_lines.append(f"[{ 'OK' if verify_ok else 'FAIL' }] verify_session (pre-finalise): {verify_status}")
+    status_lines.append(f"[{'OK' if verify_ok else 'FAIL'}] verify_session (pre-finalise): {verify_status}")
 
-    # Finalise + export only if verification ok
     anchor = None
     fin_msg = "Not finalised."
     export_path = None
@@ -202,18 +194,18 @@ def ui_quickstart_run(sign_all: bool, current_sk: str, current_pk: str):
     else:
         status_lines.append("[SKIP] Finalise/export skipped because verification failed.")
 
-    # Fan-out session id into all tabs + prefill timeline/verify/finalise/export panels
-    start_status_msg = f"Quickstart complete. session_id={sid}"
+    quick_msg = f"Quickstart complete. session_id={sid}\n\n" + "\n".join(status_lines)
 
     return (
         (sk_hex if sign_all else current_sk),
         (pk_hex if sign_all else current_pk),
-        sid, sid, sid, sid, sid,  # session ids fanout
-        start_status_msg + "\n\n" + "\n".join(status_lines),
+        sid, sid, sid, sid, sid,        # session ids fanout
+        quick_msg,                      # start_status
         tl_rows, tl_msg,
         verify_status, verify_ok, verify_report,
         anchor, fin_msg,
         export_path, export_msg,
+        quick_msg,                      # quick_status
     )
 
 
@@ -225,6 +217,20 @@ with gr.Blocks(title="RFT Agent Flight Recorder — Black Box Trace + Third-Part
         "**Key safety note:** Public demo. Do not paste production private keys here."
     )
 
+    # ------------------------------------------------------------
+    # Quickstart FIRST (most important UX improvement)
+    # ------------------------------------------------------------
+    with gr.Tab("Quickstart (1-click)"):
+        gr.Markdown(
+            "## Quickstart (1-click)\n"
+            "If you just want to see the full workflow **without guessing what to click**, use this.\n\n"
+            "**This will:** start a session → append events → verify → finalise → export a ZIP proof bundle → "
+            "and auto-fill the other tabs with the `session_id`, timeline, reports, and export file."
+        )
+        quick_sign = gr.Checkbox(label="Sign everything (Ed25519) + generate fresh keys for this run", value=False)
+        quick_run = gr.Button("Run Quickstart Demo Now")
+        quick_status = gr.Textbox(label="Quickstart status", lines=14)
+
     with gr.Tab("Keys"):
         sk = gr.Textbox(label="Ed25519 Private Key (hex)", lines=2)
         pk = gr.Textbox(label="Ed25519 Public Key (hex)", lines=2)
@@ -254,7 +260,6 @@ with gr.Blocks(title="RFT Agent Flight Recorder — Black Box Trace + Third-Part
         start_btn = gr.Button("Start New Session")
         start_status = gr.Textbox(label="Status", lines=2)
 
-        # fan-out session id to all places
         start_btn.click(
             fn=ui_start_session,
             inputs=[model_id, run_mode, notes, sign_start, sk],
@@ -391,36 +396,21 @@ with gr.Blocks(title="RFT Agent Flight Recorder — Black Box Trace + Third-Part
         diff_out = gr.Textbox(label="Diff", lines=14)
         diff_btn.click(fn=text_diff, inputs=[before, after], outputs=[diff_out])
 
-    # NEW: Quickstart (1-click)
-    with gr.Tab("Quickstart (1-click)"):
-        gr.Markdown(
-            "## Quickstart (1-click)\n"
-            "If you just want to see the full workflow **without guessing what to click**, use this.\n\n"
-            "**This will:** start a session → append a realistic event sequence → verify → finalise → export a ZIP proof bundle → "
-            "and it will auto-fill the other tabs with the `session_id`, timeline, verification report, anchor, and export file."
-        )
-        quick_sign = gr.Checkbox(label="Sign everything (Ed25519) + generate fresh keys for this run", value=False)
-        quick_run = gr.Button("Run Quickstart Demo Now")
-        quick_status = gr.Textbox(label="Quickstart status", lines=14)
-
-        # Wire Quickstart button (updates existing components too)
-        quick_run.click(
-            fn=ui_quickstart_run,
-            inputs=[quick_sign, sk, pk],
-            outputs=[
-                sk,
-                pk,
-                sid_start, sid_record, sid_tl, sid_verify, sid_final,
-                start_status,
-                tl, tl_status,
-                verify_msg, verify_ok, verify_report,
-                anchor_out, fin_status,
-                export_file, export_status,
-            ],
-        )
-        # Also show the status in this tab by mirroring Start Session status
-        # (no extra logic; we just display whatever Start Session status was set to)
-        # Users see it immediately without tab switching.
-        gr.Markdown("Tip: after running, open **Timeline**, **Verify Session**, and **Finalise + Export** — everything is already populated.")
+    # Wire Quickstart click NOW that all components exist
+    quick_run.click(
+        fn=ui_quickstart_run,
+        inputs=[quick_sign, sk, pk],
+        outputs=[
+            sk,
+            pk,
+            sid_start, sid_record, sid_tl, sid_verify, sid_final,
+            start_status,
+            tl, tl_status,
+            verify_msg, verify_ok, verify_report,
+            anchor_out, fin_status,
+            export_file, export_status,
+            quick_status,
+        ],
+    )
 
 demo.launch()