Record selected policy explicitly and reject stale playback

client.js

@@ -2,6 +2,9 @@
 const $ = id => document.getElementById(id);
 let job = null;
 let busy = false;
+let replayBusy = false;
+let localReplay = null;
+let replayRequestId = 0;
 let localResult = null;
 let localAnalysis = null;
 let pairedAnalysisPending = false;
@@ -58,7 +61,7 @@ function batchRunContext(){
   const trackInfo=trackLibrary?.info()??{source:'upload'},trackFile=$('track').files[0],network=networkSource(),backend=$('backend').value;
   const hasTrack=trackInfo.source==='builtin'||!!trackFile;
   let reason='';
-  if(busy||bundleRestoring||bucketOperation)reason='Finish the current operation before starting a batch.';
+  if(busy||replayBusy||bundleRestoring||bucketOperation)reason='Finish the current operation before starting a batch.';
   else if(job)reason='Finish or delete the current result before starting a batch.';
   else if(!capabilitiesReady)reason='Connect to the training service before running a batch.';
   else if($('run_mode')?.value==='evaluate')reason='Under Action in Run & results, choose Train a population before running a batch.';
@@ -70,7 +73,7 @@ function batchRunContext(){
     network:network?.name||(network?'Selected network':null),backend:backend.toUpperCase()};
 }
 function buttons() {
-  const settingsUnavailable=busy||bundleRestoring||bucketOperation;
+  const settingsUnavailable=busy||replayBusy||bundleRestoring||bucketOperation;
   const unavailable=settingsUnavailable||!!job;
   trackLibrary?.setBusy(unavailable);
   window.HFAuth?.setBusy(unavailable);
@@ -87,6 +90,13 @@ function buttons() {
   $('forgetLocal').disabled = busy || !localResult;
   $('track').disabled=unavailable;
   $('network').disabled=unavailable;
+  if($('replay_record'))$('replay_record').disabled=unavailable||batchActive||!capabilitiesReady||!replayNetworkSource();
+  if($('replay_duration'))$('replay_duration').disabled=settingsUnavailable;
+  if($('replay_download'))$('replay_download').disabled=replayBusy||!localReplay;
+  if($('replay_policy')){
+    const selected=networkSource();
+    $('replay_policy').textContent=selected?'Policy: '+(selected.name||'Selected network'):localResult?'Policy: latest received result. Choose a winner below to record a different policy.':'Choose or create a network to record a drive.';
+  }
   if($('batch_start')){
     const context=batchRunContext();
     window.setBatchRunContext?.(context);
@@ -438,7 +448,7 @@ function saveLocal() {
   setTimeout(()=>URL.revokeObjectURL(url),60000);
 }
 $('saveLocal').onclick=saveLocal;
-$('forgetLocal').onclick=()=>{localResult=null;localAnalysis=null;pairedAnalysisPending=false;renderAnalysisPanel();buttons();say('Browser result discarded. Downloaded files are unaffected.');};
+$('forgetLocal').onclick=()=>{invalidateReplay();localResult=null;localAnalysis=null;pairedAnalysisPending=false;renderAnalysisPanel();buttons();say('Browser result discarded. Downloaded files are unaffected.');};
 async function request(path, options = {}) {
   const response = await fetch(path, {...options, cache:'no-store', credentials:'same-origin'});
   if (!response.ok) {
@@ -455,19 +465,71 @@ async function encoded(file) {
 }
 const headers = () => ({'X-Job-Key':job.key});
 const networkSource=()=>stagedNetwork || $('network').files[0];
+function replayNetworkSource(){return networkSource()||localResult;}
+function invalidateReplay(reason){
+  replayRequestId++;
+  localReplay=null;
+  window.TrackReplay?.clearRecording?.(reason);
+  if($('replay_download'))$('replay_download').disabled=true;
+}
 window.addEventListener('ald:batch-draft-change',buttons);
 document.addEventListener('track-library-change',buttons);
+document.addEventListener('track-library-change',()=>invalidateReplay('Track changed. Record a new drive for this track.'));
+if($('replay_record'))$('replay_record').onclick=async()=>{
+  if(busy||replayBusy||job||batchActive||bundleRestoring||bucketOperation)return;
+  replayBusy=true;busy=true;
+  invalidateReplay('Recording a fresh CPU drive…');
+  const attempt=replayRequestId;
+  const controller=new AbortController();
+  const timeout=setTimeout(()=>controller.abort(),40000);
+  buttons();
+  try{
+    const track=await trackSource(),network=replayNetworkSource();
+    if(!track||!network)throw Error('Choose a track and a network before recording.');
+    const ticks=Number($('replay_duration').value);
+    if(![300,900].includes(ticks))throw Error('Choose a 5-second or 15-second recording.');
+    const [trackBytes,networkBytes]=await Promise.all([track.arrayBuffer(),network.arrayBuffer()]);
+    await validateNetworkBytes(new Uint8Array(networkBytes));
+    const [trackSha,networkSha,track64,network64]=await Promise.all([
+      sha(trackBytes),sha(networkBytes),encoded(track),encoded(network)
+    ]);
+    const response=await request('/api/replay',{method:'POST',signal:controller.signal,
+      headers:{'Content-Type':'application/json'},body:JSON.stringify({track:track64,network:network64,ticks})});
+    const raw=await response.text();
+    if(raw.length>2*1024*1024)throw Error('Recording exceeds the browser limit.');
+    const recording=JSON.parse(raw);
+    const currentTrack=await trackSource(),currentNetwork=replayNetworkSource();
+    if(attempt!==replayRequestId||!currentTrack||!currentNetwork)throw Error('Selection changed during recording. Record the current selection again.');
+    const [currentTrackSha,currentNetworkSha]=await Promise.all([
+      currentTrack.arrayBuffer().then(sha),currentNetwork.arrayBuffer().then(sha)
+    ]);
+    if(currentTrackSha!==trackSha||currentNetworkSha!==networkSha)throw Error('Selection changed during recording. Record the current selection again.');
+    if(!window.TrackReplay?.setRecording)throw Error('The recorded-drive viewer is unavailable. Reload this page.');
+    await window.TrackReplay.setRecording(recording,trackSha,networkSha);
+    if(attempt!==replayRequestId){window.TrackReplay.clearRecording();throw Error('Selection changed during recording. Record the current selection again.');}
+    localReplay=recording;
+  }catch(error){
+    if(attempt===replayRequestId&&$('replay_status'))$('replay_status').textContent=
+      error?.name==='AbortError'?'Recording timed out. Try a 5-second drive.':'Recording unavailable: '+error.message;
+  }finally{
+    clearTimeout(timeout);replayBusy=false;busy=false;buttons();
+  }
+};
+if($('replay_download'))$('replay_download').onclick=()=>{
+  if(localReplay)downloadTextFile(JSON.stringify(localReplay),'ald-recorded-drive.json','application/json');
+};
 async function refreshFreezeNetwork(file){
+  invalidateReplay('Network changed. Record a new drive for this policy.');
   if(!file){window.setFreezeNetwork?.(null);window.TrackReplay?.setStagedNetwork?.(null);return;}
   try{const network=await validateNetworkBytes(new Uint8Array(await file.arrayBuffer()));if(networkSource()===file){window.setFreezeNetwork?.(network);window.TrackReplay?.setStagedNetwork?.(network);}}
   catch(error){if(networkSource()===file){window.setFreezeNetwork?.(null);window.TrackReplay?.setStagedNetwork?.(null);if($('freeze_status'))$('freeze_status').textContent='Network metadata unavailable: '+error.message;}}
 }
 $('network').onchange=()=>{stagedNetwork=null;selectedWinner=null;if($('networkChoice'))$('networkChoice').textContent='Uploaded file selected.';refreshFreezeNetwork(networkSource());buttons();};
 window.TrainingClient={
-  isBusy:()=>busy||!!job||bundleRestoring,
+  isBusy:()=>busy||replayBusy||!!job||bundleRestoring,
   hasUnsavedState:()=>!!(networkSource()||localResult||$('track').files.length||trackLibrary?.info?.()?.source==='builtin'||$('editor_file')?.files.length||window.hasUnsavedNetworkDraft?.()),
   useNetwork(file){
-    if(busy||job)throw Error('Finish or delete the current result before changing the network');
+    if(busy||replayBusy||job)throw Error('Finish the current operation before changing the network');
     stagedNetwork=file;$('network').value='';
     if($('networkChoice'))$('networkChoice').textContent='Browser editor input: '+file.name+'. Not uploaded yet.';
     refreshFreezeNetwork(file);
@@ -961,6 +1023,8 @@ function showDetails(r){
   window.TrackReplay?.setRunAnalysis?.(localAnalysis,runReceipt||portableReceipt||(r?{report:r}:null));
 }
 $('run').onclick = async () => {
+  if(replayBusy)return;
+  invalidateReplay();
   const comparisonAttempt=!!comparisonPlan,comparisonRole=comparisonAttempt?comparisonPlan.roles[comparisonPlan.index]:null;
   const previousRunResultContext=runResultContext;
   cancelSteamLeaderboard();