Update app.py

app.py

@@ -1,24 +1,40 @@
 import json
 import os
 import tempfile
-from typing import Any, Dict, Optional, Tuple
+import zipfile
+from typing import Any, Dict, Optional, Tuple, List
 
 import gradio as gr
 
 from drp.bundle import verify_bundle
 from drp.diff import diff_bundles
 from drp.report import render_report_markdown
+from drp.pdf_report import write_pdf_report
 from drp.simulate import make_demo_bundle_zip, fork_patch_bundle
 
 
 APP_TITLE = "TimelineDiff — Differential Reproducibility Protocol (DRP)"
 
 
+def _tmp_dir() -> str:
+    return tempfile.mkdtemp(prefix="drp_")
+
+
 def _tmp_path(name: str) -> str:
-    d = tempfile.mkdtemp(prefix="drp_")
+    d = _tmp_dir()
     return os.path.join(d, name)
 
 
+def _as_alignment_rows(alignment: List[Dict[str, Any]]) -> List[List[Any]]:
+    """
+    Convert alignment dict rows to Dataframe rows.
+    """
+    rows = []
+    for r in alignment:
+        rows.append([r.get("i"), r.get("status"), r.get("kind_a"), r.get("step_a"), r.get("kind_b"), r.get("step_b")])
+    return rows
+
+
 def ui_verify(bundle_file) -> Tuple[str, Dict[str, Any]]:
     if not bundle_file:
         return "Upload a bundle zip first.", {"ok": False}
@@ -28,30 +44,52 @@ def ui_verify(bundle_file) -> Tuple[str, Dict[str, Any]]:
     return msg, summary
 
 
-def ui_diff(bundle_a, bundle_b) -> Tuple[str, Dict[str, Any], str]:
+def ui_diff(bundle_a, bundle_b) -> Tuple[str, str, Dict[str, Any], List[List[Any]], str, str, str]:
     if not bundle_a or not bundle_b:
-        return "Upload two bundles.", {"error": "missing input"}, ""
+        return "Upload two bundles.", "", {"error": "missing input"}, [], "", "", ""
 
     diff = diff_bundles(bundle_a.name, bundle_b.name)
     md = render_report_markdown(diff)
 
-    # export report artifacts
-    out_zip = _tmp_path("drp-diff-report.zip")
-    out_md = _tmp_path("report.md")
-    out_json = _tmp_path("diff.json")
+    # Summary header (quick context above JSON)
+    s = diff.get("summary", {})
+    parts = []
+    if s.get("first_divergence_index") is None:
+        parts.append("Timelines identical")
+    else:
+        parts.append(f"Identical until step/index {s.get('first_divergence_index')}")
+    parts.append(f"{s.get('diff_event_count', 0)} events differ")
+    if s.get("missing_event_count", 0):
+        parts.append(f"{s.get('missing_event_count')} missing")
+    if s.get("final_reward_delta") is not None:
+        parts.append(f"Final reward delta: {s.get('final_reward_delta'):.6g}")
+    summary_line = " · ".join(parts)
+
+    # Export artifacts
+    out_dir = _tmp_dir()
+    out_md = os.path.join(out_dir, "report.md")
+    out_json = os.path.join(out_dir, "diff.json")
+    out_pdf = os.path.join(out_dir, "report.pdf")
+    out_zip = os.path.join(out_dir, "drp-diff-report.zip")
 
     with open(out_md, "w", encoding="utf-8") as f:
         f.write(md)
     with open(out_json, "w", encoding="utf-8") as f:
         json.dump(diff, f, ensure_ascii=False, indent=2)
 
-    import zipfile
+    # PDF: render from markdown lines (plain, reliable)
+    md_lines = md.splitlines()
+    write_pdf_report(out_pdf, "DRP Differential Report", md_lines)
+
     with zipfile.ZipFile(out_zip, "w", compression=zipfile.ZIP_DEFLATED) as z:
         z.write(out_md, arcname="report.md")
+        z.write(out_pdf, arcname="report.pdf")
         z.write(out_json, arcname="diff.json")
 
-    headline = "✅ Diff complete."
-    return headline, diff, out_zip
+    alignment_rows = _as_alignment_rows(diff.get("alignment", []))
+
+    status = "✅ Diff complete."
+    return status, summary_line, diff, alignment_rows, out_zip, out_pdf, out_md
 
 
 def ui_generate(seed_a: int, seed_b: int, chaos: float) -> Tuple[str, str, str]:
@@ -89,19 +127,37 @@ def ui_fork(source_bundle, fork_index: int, patch_kind: str, patch_step: str, pa
 
 
 with gr.Blocks(title=APP_TITLE) as demo:
-    gr.Markdown(f"# {APP_TITLE}\nDiff two agent timelines. Find first divergence. Export a forensic report.")
+    gr.Markdown(f"# {APP_TITLE}\nDiff two agent timelines. Find first divergence. Export forensic reports.")
 
     with gr.Tab("Diff two bundles"):
         with gr.Row():
             bundle_a = gr.File(label="Bundle A (.zip)", file_types=[".zip"])
             bundle_b = gr.File(label="Bundle B (.zip)", file_types=[".zip"])
+
         run = gr.Button("Compute differential report", variant="primary")
 
         status = gr.Textbox(label="Status", interactive=False)
+        summary = gr.Textbox(label="Summary (fast context)", interactive=False)
+
+        alignment_table = gr.Dataframe(
+            headers=["i", "status", "kind_a", "step_a", "kind_b", "step_b"],
+            label="Timeline alignment (scan for first divergence)",
+            interactive=False,
+            wrap=True,
+        )
+
         diff_json = gr.JSON(label="Diff JSON (summary + per-event diffs)")
-        report_zip = gr.File(label="Download drp-diff-report.zip")
 
-        run.click(fn=ui_diff, inputs=[bundle_a, bundle_b], outputs=[status, diff_json, report_zip])
+        with gr.Row():
+            report_zip = gr.File(label="Download drp-diff-report.zip (MD + PDF + JSON)")
+            report_pdf = gr.File(label="Download report.pdf")
+            report_md = gr.File(label="Download report.md")
+
+        run.click(
+            fn=ui_diff,
+            inputs=[bundle_a, bundle_b],
+            outputs=[status, summary, diff_json, alignment_table, report_zip, report_pdf, report_md],
+        )
 
     with gr.Tab("Verify bundle integrity"):
         bundle_v = gr.File(label="Bundle (.zip)", file_types=[".zip"])
@@ -130,6 +186,7 @@ with gr.Blocks(title=APP_TITLE) as demo:
         with gr.Row():
             patch_kind = gr.Textbox(label="Patch kind (optional)", placeholder="e.g. tool_result")
             patch_step = gr.Textbox(label="Patch step (optional)", placeholder="e.g. t12.tool_result")
+
         patch_payload = gr.Textbox(
             label="Patch payload JSON (optional)",
             lines=8,
@@ -142,7 +199,7 @@ with gr.Blocks(title=APP_TITLE) as demo:
         fork_btn.click(fn=ui_fork, inputs=[src, fork_index, patch_kind, patch_step, patch_payload], outputs=[fork_msg, fork_file])
 
     gr.Markdown(
-        "Tip: In a real agent exporter, emit DRP events at: LLM calls, tool calls/results, memory writes, planner steps, guardrails."
+        "Exporter tip: emit DRP events at LLM calls, tool calls/results, memory writes, planner steps, guardrails, and state snapshots."
     )
 
 if __name__ == "__main__":