from pathlib import Path

lsh_path = Path(r"C:\Users\intel\AppData\Local\OptimusRuntime\Labs\AR74\12_ar88_authority_operational_closure\candidate_workspace\app\remediation\self_repair\live_shadow_hook.py")
content = lsh_path.read_text(encoding="utf-8")

old_sec = """                act_ev = update_payload.get("activity_event")
                if act_ev:
                    activities.append({
                        "timestamp": time.time(),
                        "event": act_ev,
                        "status": update_payload.get("status") or t_entry.get("status"),
                        "message": update_payload.get("message") or act_ev,
                        "incident_id": update_payload.get("incident_id"),
                        "shadow_run_id": update_payload.get("shadow_run_id"),
                    })"""

new_sec = """                act_ev = update_payload.get("activity_event")
                if act_ev:
                    import time
                    activities.append({
                        "timestamp": time.time(),
                        "event": act_ev,
                        "status": update_payload.get("status") or t_entry.get("status"),
                        "message": update_payload.get("message") or act_ev,
                        "incident_id": update_payload.get("incident_id"),
                        "shadow_run_id": update_payload.get("shadow_run_id"),
                    })"""

assert old_sec in content, "old_sec not found"
content = content.replace(old_sec, new_sec, 1)

# Ensure time is imported at top level as well
if "import time" not in content[:1000]:
    content = "import time\n" + content

lsh_path.write_text(content, encoding="utf-8")
print("SUCCESS: Fixed time import in live_shadow_hook.py!")