frozen/ stuck connection optim

backend/engine.py

@@ -89,7 +89,7 @@ def run(model, video_path, line, config, on_frame, save_annotated=False, annotat
     if save_annotated:
         annotated_dir = os.path.join(tempfile.gettempdir(), "funky_reports")
         os.makedirs(annotated_dir, exist_ok=True)
-        annotated_path = os.path.join(annotated_dir, "annotated.mp4")
+        annotated_path = os.path.join(annotated_dir, f"annotated_{os.path.basename(video_path)}.mp4")
         fourcc = cv2.VideoWriter_fourcc(*"mp4v")
         writer = cv2.VideoWriter(annotated_path, fourcc, fps / stride, (out_w, out_h))
 
@@ -115,7 +115,7 @@ def run(model, video_path, line, config, on_frame, save_annotated=False, annotat
         vid_stride=stride,
         stream=True,
         verbose=False,
-        persist=True
+        persist=False  # MUST be False — True causes ByteTracker state leak across runs
     )
 
     a = line[0]

backend/server.py

@@ -46,15 +46,22 @@ def index():
 async def upload(file: UploadFile = File(...)):
     video_id = str(uuid.uuid4())[:8]
     path = UPLOAD_DIR / f"{video_id}.mp4"
-    
+
+    # Clean up any previous temp uploads to avoid stale state
+    for old_path in UPLOAD_DIR.glob("*.mp4"):
+        try:
+            old_path.unlink()
+        except Exception:
+            pass
+
     print(f"[BACKEND] Received upload request: {file.filename}")
     try:
         with open(path, "wb") as f:
             shutil.copyfileobj(file.file, f)
-        
+
         file_size = os.path.getsize(path)
         print(f"[BACKEND] Successfully stored: {path} ({file_size} bytes)")
-        
+
         videos[video_id] = str(path)
         return {"video_id": video_id}
     except Exception as e:
@@ -95,10 +102,8 @@ def generate_reports(video_id: str):
     # Include annotated video if it exists
     annotated_src = data.get("annotated_video")
     if annotated_src and os.path.exists(annotated_src):
-        import shutil
         dest = os.path.join(out_dir, "annotated.mp4")
-        if not os.path.exists(dest):
-            shutil.copy2(annotated_src, dest)
+        shutil.copy2(annotated_src, dest)  # Always overwrite — never skip
         files.append("annotated.mp4")
     return {"files": files}
 

frontend/vehicles.html

@@ -1284,8 +1284,28 @@
                 }
             };
 
+            let processingDone = false;
+
             ws.onclose = () => {
                 console.log('WS Closed');
+                if (!processingDone) {
+                    // Closed before done=True received — show error state
+                    document.getElementById('proc-label').innerText = 'Disconnected';
+                    const badge = document.getElementById('results-status-badge');
+                    if (badge) {
+                        badge.innerText = 'Connection Lost';
+                        badge.className = 'px-2.5 py-1 bg-red-900/40 text-red-300 text-[10px] font-bold rounded-full uppercase tracking-tighter';
+                    }
+                    document.getElementById('run-results-content').innerHTML = `
+                        <div class="flex flex-col items-center justify-center p-8 border-2 border-dashed border-red-900/40 rounded-2xl col-span-3 text-slate-400">
+                            <i class="fa-solid fa-triangle-exclamation text-2xl mb-3 text-red-400"></i>
+                            <span class="text-xs font-semibold mb-1">Processing connection was lost.</span>
+                            <span class="text-[10px] text-slate-500 mb-4">The server may have timed out or restarted. Please try again.</span>
+                            <button onclick="startNewAnalysis()" class="text-[10px] font-bold uppercase tracking-widest px-4 py-2 rounded-full" style="background:#111;border:1px solid #2a2a2a;color:#c89a6c">
+                                &larr; Start New Analysis
+                            </button>
+                        </div>`;
+                }
             };
 
             let lastUIUpdate = 0;
@@ -1294,6 +1314,7 @@
                 const d = JSON.parse(e.data);
 
                 if (d.done) {
+                    processingDone = true;
                     document.getElementById('proc-label').innerText = 'Complete';
                     document.getElementById('proc-bar').style.width = '100%';
                     document.getElementById('proc-pct').innerText = '100%';