fix:完成实时流检测联调

src/components/WebRtcPlayer.vue

@@ -1,22 +1,16 @@
 <template>
   <div class="webrtc-wrapper">
-    <video 
+    <video
-      ref="videoRef" 
+      ref="videoRef"
-      autoplay 
+      autoplay
-      muted 
+      muted
-      playsinline 
+      playsinline
       class="video-player"
     ></video>
-    
+
-    <!-- 加载状态 -->
     <div v-if="loading" class="loading-overlay">
       <span class="spinner"></span>
-      <span>连接中...</span>
+      <span>{{ loadingText }}</span>
-    </div>
-
-    <!-- 录制状态指示器 -->
-    <div v-if="isRecording" class="recording-indicator">
-      <span class="dot"></span> 录制中 {{ recordTime }}s
     </div>
 
     <div v-if="error" class="error-overlay">{{ error }}</div>
@@ -24,206 +18,195 @@
 </template>
 
 <script setup>
-import { ref, onMounted, onUnmounted, watch } from 'vue';
+import { ref, onMounted, onUnmounted, watch } from 'vue'
+
 const props = defineProps({
   src: { type: String, required: true }
-});
+})
+
+const emit = defineEmits(['connection-status'])
+
+const videoRef = ref(null)
+const error = ref('')
+const loading = ref(false)
+const loadingText = ref('连接中...')
+
+let pc = null
+let sessionUrl = ''
+let queuedCandidates = []
+let offerData = null
+let retryTimer = null
+let retryCount = 0
+const MAX_RETRIES = 30
+const RETRY_INTERVAL = 2000
+let destroyed = false
+
+const whepUrl = () => {
+  const base = props.src.replace(/\/+$/, '')
+  return base + '/whep'
+}
 
-// 监听 src 变化，重新连接
+function parseOffer(sdp) {
-watch(() => props.src, (newSrc, oldSrc) => {
+  const ret = { iceUfrag: '', icePwd: '', medias: [] }
-  if (newSrc && newSrc !== oldSrc) {
+  for (const line of sdp.split('\r\n')) {
-    disconnect();
+    if (line.startsWith('m=')) ret.medias.push(line.slice('m='.length))
-    connect();
+    else if (ret.iceUfrag === '' && line.startsWith('a=ice-ufrag:')) ret.iceUfrag = line.slice('a=ice-ufrag:'.length)
+    else if (ret.icePwd === '' && line.startsWith('a=ice-pwd:')) ret.icePwd = line.slice('a=ice-pwd:'.length)
   }
-});
+  return ret
-
+}
-// 暴露给父组件的方法
+
-const emit = defineEmits(['record-status-change','record-complete','connection-status']);
+function generateSdpFragment(od, candidates) {
-
+  const byMedia = {}
-const videoRef = ref(null);
+  for (const c of candidates) {
-const error = ref('');
+    const mid = c.sdpMLineIndex
-const loading = ref(false);
+    if (!byMedia[mid]) byMedia[mid] = []
-let pc = null;
+    byMedia[mid].push(c)
-
-// --- 录屏相关状态 ---
-const isRecording = ref(false);
-const recordTime = ref(0);
-let mediaRecorder = null;
-let recordedChunks = [];
-let timerInterval = null;
-
-// 1. 断开 WebRTC 连接
-const disconnect = () => {
-  if (pc) {
-    pc.close();
-    pc = null;
   }
-  if (videoRef.value) {
+  let frag = 'a=ice-ufrag:' + od.iceUfrag + '\r\n' + 'a=ice-pwd:' + od.icePwd + '\r\n'
-    videoRef.value.srcObject = null;
+  let mid = 0
+  for (const media of od.medias) {
+    if (byMedia[mid]) {
+      frag += 'm=' + media + '\r\n' + 'a=mid:' + mid + '\r\n'
+      for (const c of byMedia[mid]) frag += 'a=' + c.candidate + '\r\n'
+    }
+    mid++
   }
-  error.value = '';
+  return frag
-};
+}
+
+function linkToIceServers(links) {
+  if (!links) return []
+  return links.split(', ').map((link) => {
+    const m = link.match(/^<(.+?)>; rel="ice-server"(; username="(.*?)"; credential="(.*?)"; credential-type="password")?/i)
+    const ret = { urls: [m[1]] }
+    if (m[3] !== undefined) {
+      ret.username = JSON.parse(`"${m[3]}"`)
+      ret.credential = JSON.parse(`"${m[4]}"`)
+      ret.credentialType = 'password'
+    }
+    return ret
+  })
+}
+
+function sendLocalCandidates(candidates) {
+  fetch(sessionUrl, {
+    method: 'PATCH',
+    headers: { 'Content-Type': 'application/trickle-ice-sdpfrag', 'If-Match': '*' },
+    body: generateSdpFragment(offerData, candidates)
+  }).catch(() => {})
+}
+
+function cleanupConnection() {
+  if (pc) { pc.close(); pc = null }
+  if (videoRef.value) videoRef.value.srcObject = null
+  sessionUrl = ''
+  queuedCandidates = []
+  offerData = null
+}
+
+function cancelRetry() {
+  if (retryTimer) {
+    clearTimeout(retryTimer)
+    retryTimer = null
+  }
+}
+
+async function connect() {
+  if (!props.src || destroyed) return
+  cancelRetry()
+  cleanupConnection()
+
+  loading.value = true
+  loadingText.value = retryCount > 0 ? `连接中... (第 ${retryCount} 次重试)` : '连接中...'
+  error.value = ''
 
-// 2. 连接 WebRTC
-const connect = async () => {
-  loading.value = true;
-  error.value = '';
   try {
-    pc = new RTCPeerConnection();
+    const optionsRes = await fetch(whepUrl(), { method: 'OPTIONS' })
+    const iceServers = linkToIceServers(optionsRes.headers.get('Link'))
+
+    pc = new RTCPeerConnection({ iceServers, sdpSemantics: 'unified-plan' })
+    pc.addTransceiver('video', { direction: 'sendrecv' })
+    pc.addTransceiver('audio', { direction: 'sendrecv' })
+
     pc.ontrack = (event) => {
       if (videoRef.value && event.streams[0]) {
-        videoRef.value.srcObject = event.streams[0];
+        videoRef.value.srcObject = event.streams[0]
-        emit('connection-status', true);
+        emit('connection-status', true)
+        retryCount = 0
+        loading.value = false
+        error.value = ''
+      }
+    }
+
+    pc.onicecandidate = (evt) => {
+      if (evt.candidate !== null) {
+        if (sessionUrl) sendLocalCandidates([evt.candidate])
+        else queuedCandidates.push(evt.candidate)
       }
-    };
+    }
-    pc.addTransceiver('video', { direction: 'recvonly' });
-    pc.addTransceiver('audio', { direction: 'recvonly' });
 
-    const offer = await pc.createOffer();
+    const offer = await pc.createOffer()
-    await pc.setLocalDescription(offer);
+    offerData = parseOffer(offer.sdp)
+    await pc.setLocalDescription(offer)
 
-    const res = await fetch(props.src, {
+    const res = await fetch(whepUrl(), {
       method: 'POST',
       headers: { 'Content-Type': 'application/sdp' },
-      body: pc.localDescription.sdp
+      body: offer.sdp
-    });
+    })
-    
+
-    if (!res.ok) throw new Error(`Status: ${res.status}`);
+    if (!res.ok) {
-    
+      const errBody = await res.text().catch(() => '')
-    const answer = await res.text();
+      const isNotPublishing = errBody.includes('no one is publishing')
-    await pc.setRemoteDescription({ type: 'answer', sdp: answer });
+      if (isNotPublishing) throw new Error('NOT_PUBLISHING')
-  } catch (e) {
+      throw new Error(errBody || `HTTP ${res.status}`)
-    error.value = '连接失败，请检查视频地址';
-    emit('connection-status', false);
-    console.error(e);
-  } finally {
-    let timeout = setTimeout(() => {
-      loading.value = false;
-      clearTimeout(timeout);
-    }, 1500);
-  }
-};
-
-// 2. 截图功能
-const takeSnapshot = () => {
-  if (!videoRef.value) return null;
-  
-  const video = videoRef.value;
-  const canvas = document.createElement('canvas');
-  canvas.width = video.videoWidth;
-  canvas.height = video.videoHeight;
-  
-  const ctx = canvas.getContext('2d');
-  ctx.drawImage(video, 0, 0, canvas.width, canvas.height);
-  
-  // 转换为 Blob (格式可选: image/png, image/jpeg)
-  return new Promise((resolve) => {
-    canvas.toBlob((blob) => {
-      resolve(blob);
-    }, 'image/jpeg', 0.9);
-  });
-};
-
-// 3. 开始录屏
-const startRecord = () => {
-  if (!videoRef.value || isRecording.value) return;
-
-  try {
-    // 捕获视频流的 MediaStream
-    // 注意：captureStream 是实验性 API，但在现代 Chrome/Edge/Firefox 中支持良好
-    const stream = videoRef.value.captureStream ? videoRef.value.captureStream() : videoRef.value.mozCaptureStream();
-    
-    if (!stream) {
-      error.value = '浏览器不支持捕获视频流';
-      return;
     }
 
-    recordedChunks = [];
+    const location = res.headers.get('location')
-    // 创建 MediaRecorder
+    if (location) sessionUrl = new URL(location, whepUrl()).toString()
-    // mimeType 可以根据需要选择，例如 'video/webm; codecs=vp9'
-    mediaRecorder = new MediaRecorder(stream, { mimeType: 'video/webm' });
 
-    mediaRecorder.ondataavailable = (event) => {
+    const answer = await res.text()
-      if (event.data.size > 0) {
+    await pc.setRemoteDescription({ type: 'answer', sdp: answer })
-        recordedChunks.push(event.data);
+
-      }
+    if (queuedCandidates.length > 0 && sessionUrl) {
-    };
+      sendLocalCandidates(queuedCandidates)
-
+      queuedCandidates = []
-    mediaRecorder.onstop = () => {
+    }
-        const blob = new Blob(recordedChunks, { type: 'video/webm' });
+
-        recordedChunks = []; // 清空数组释放内存
+    retryCount = 0
-        
-        // 【关键】将 blob 发送给父组件
-        emit('record-complete', blob);
-        
-        console.log('录制完成，Blob大小:', blob.size);
-    };
-
-    mediaRecorder.start();
-    isRecording.value = true;
-    recordTime.value = 0;
-    
-    // 计时器
-    timerInterval = setInterval(() => {
-      recordTime.value++;
-    }, 1000);
-
-    emit('record-status-change', true);
-
-  } catch (err) {
-    console.error('录制启动失败:', err);
-    error.value = '录制启动失败';
-  }
-};
-
-// 4. 停止录屏
-const stopRecord = () => {
-  if (mediaRecorder && isRecording.value) {
-    mediaRecorder.stop();
-    isRecording.value = false;
-    clearInterval(timerInterval);
-    emit('record-status-change', false);
-  }
-};
-
-// 5. 处理录制完成（这里可以改为直接上传后端）
-const handleRecordComplete = (blob) => {
-  console.log('录制完成，文件大小:', blob.size, 'bytes');
-  
-  // 【示例】如何上传给后端
-  // uploadVideoToServer(blob);
-  
-  // 如果需要在前端预览或下载，可以创建 URL
-  // const url = URL.createObjectURL(blob);
-};
-
-// 【可选】上传函数示例
-const uploadVideoToServer = async (blob) => {
-  const formData = new FormData();
-  formData.append('file', blob, `record_${new Date().getTime()}.webm`);
-  
-  try {
-    // 替换为你的后端上传接口
-    // await axios.post('/api/upload/video', formData);
-    console.log('视频上传成功');
   } catch (e) {
-    console.error('上传失败', e);
+    const isNotPublishing = e.message === 'NOT_PUBLISHING'
+    cleanupConnection()
+
+    if (isNotPublishing && retryCount < MAX_RETRIES && !destroyed) {
+      retryCount++
+      loadingText.value = `等待推流中... (${retryCount}/${MAX_RETRIES})`
+      retryTimer = setTimeout(() => connect(), RETRY_INTERVAL)
+    } else {
+      error.value = isNotPublishing ? '视频流尚未就绪，请稍后重试' : '连接失败，请检查视频地址'
+      emit('connection-status', false)
+      loading.value = false
+      console.error('[WebRTC]', e.message === 'NOT_PUBLISHING' ? 'stream not publishing' : e)
+    }
   }
-};
+}
 
-onMounted(() => connect());
+watch(() => props.src, (newSrc, oldSrc) => {
+  if (newSrc && newSrc !== oldSrc) {
+    retryCount = 0
+    cancelRetry()
+    connect()
+  }
+})
+
+onMounted(() => connect())
 
 onUnmounted(() => {
-  if (pc) pc.close();
+  destroyed = true
-  if (isRecording.value) stopRecord();
+  cancelRetry()
-});
+  cleanupConnection()
-
+})
-// 暴露方法给父组件
-defineExpose({
-  takeSnapshot,
-  startRecord,
-  stopRecord
-});
 </script>
 
 <style scoped>
@@ -253,30 +236,4 @@ defineExpose({
 @keyframes spin {
   to { transform: rotate(360deg); }
 }
-
+</style>
-.recording-indicator {
-  position: absolute;
-  top: 10px;
-  right: 10px;
-  background: rgba(0, 0, 0, 0.6);
-  color: white;
-  padding: 4px 8px;
-  border-radius: 4px;
-  font-size: 12px;
-  display: flex;
-  align-items: center;
-  gap: 6px;
-}
-.dot {
-  width: 8px;
-  height: 8px;
-  background-color: red;
-  border-radius: 50%;
-  animation: blink 1s infinite;
-}
-@keyframes blink {
-  0% { opacity: 1; }
-  50% { opacity: 0.4; }
-  100% { opacity: 1; }
-}
-</style>

src/views/home/components/ControlButtons.vue

@@ -12,7 +12,7 @@
       :type="algorithmRunning ? 'danger' : 'success'" 
       size="default"
       @click="$emit('toggleAlgorithm')"
-      :disabled="offlineVideoMissing || isLiveMode"
+      :disabled="!isLiveMode && offlineVideoMissing"
     >
       {{ algorithmRunning ? '关闭算法' : '启动算法' }}
     </el-button>

src/views/home/components/VideoPlayer.vue

@@ -4,7 +4,6 @@
     <WebRtcPlayer
       v-if="videoMode === 'live'"
       :src="webrtcUrl"
-      @connection-status="$emit('connectionChange', $event)"
     />
     <!-- 离线视频模式 -->
     <div v-else-if="videoMode === 'offline'" class="offline-player">

src/views/home/index.vue

@@ -35,7 +35,6 @@
         :webrtc-url="webrtcUrl"
         :offline-video-url="offlineVideoUrl"
         :video-autoplay-trigger="videoAutoplayTrigger"
-        @connection-change="onConnectionChange"
       />
 
       <DetectionProgress
@@ -58,6 +57,21 @@
     <div class="right-section">
       <ChartsPanel :process-stats="processStats" :counts="detectCounts" />
     </div>
+
+    <!-- SOP 未完成弹窗 -->
+    <el-dialog
+      v-model="sopAlert.visible"
+      :title="sopAlert.title"
+      width="500px"
+      :close-on-click-modal="false"
+      :close-on-press-escape="false"
+      draggable
+    >
+      <div style="white-space: pre-line; line-height: 1.8;">{{ sopAlert.content }}</div>
+      <template #footer>
+        <el-button type="primary" @click="sopAlert.visible = false">知道了</el-button>
+      </template>
+    </el-dialog>
   </div>
 </template>
 
@@ -78,26 +92,328 @@ import { downloadAsZip } from '@/utils/zip.js'
 
 // ============ 视频流相关 ============
 const videoMode = ref('idle')
-const streamAddress = ref('')
+const streamAddress = ref('rtsp://10.23.22.97:8556/camera2')
-const webrtcUrl = computed(() => {
+const webrtcUrl = ref('')
-  return import.meta.env.VITE_WEBRTC_URL || `http://${streamAddress.value}/stream`
-})
 const streamRunning = ref(false)
 const isLiveMode = computed(() => videoMode.value === 'live')
+const RAW_WS_URL = 'ws://10.23.22.43:8001/ws/raw_rtsp'
+const DETECT_WS_URL = 'ws://10.23.22.43:8001/ws/detect_rtsp'
+let rawWsClient = null
+let detectWsClient = null
+
+function destroyRawWs() {
+  if (rawWsClient) {
+    rawWsClient.destroy()
+    rawWsClient = null
+  }
+}
+
+function destroyDetectWs() {
+  if (detectWsClient) {
+    detectWsClient.destroy()
+    detectWsClient = null
+  }
+}
+
+function stopRawStream() {
+  if (rawWsClient && rawWsClient.getReadyState() === WebSocket.OPEN) {
+    rawWsClient.send({ type: 'stop' })
+  }
+  destroyRawWs()
+}
+
+let detectStopCallback = null
+
+function stopDetectStream(onDone) {
+  if (detectWsClient && detectWsClient.getReadyState() === WebSocket.OPEN) {
+    detectStopCallback = onDone || null
+    detectWsClient.send({ type: 'stop' })
+    addLog('已发送停止请求，等待检测结果...', 'info')
+  } else {
+    // 没有活跃的检测连接，立即回调
+    if (onDone) onDone()
+  }
+}
+
+function buildRtspUrl(input) {
+  return input.startsWith('rtsp://') ? input : `rtsp://${input}`
+}
+
+function connectRawStream() {
+  destroyRawWs()
+  webrtcUrl.value = ''
+  addLog('正在连接原始流服务...', 'info')
+
+  rawWsClient = new WebSocketClient({
+    url: RAW_WS_URL,
+    autoReconnect: false,
+    heartbeatInterval: 0,
+    heartbeatTimeout: 0,
+    debug: true
+  })
+
+  rawWsClient.on('open', () => {
+    addLog('原始流服务已连接，请求流地址', 'info')
+    rawWsClient.send({ type: 'start_raw' })
+  })
+
+  rawWsClient.on('message', (msg) => {
+    switch (msg.type) {
+      case 'ready':
+        addLog(msg.message || '原始流服务就绪', 'info')
+        break
+
+      case 'processing_started':
+        addLog(msg.message || '原始流发布已启动', 'info')
+        break
+
+      case 'webrtc_stream':
+        webrtcUrl.value = msg.webrtc_url
+        addLog('已获取原始 WebRTC 流，开始播放', 'success')
+        ElMessage.success('原始视频流已连接')
+        break
+
+      case 'stopping':
+        addLog(msg.message || '原始流正在关闭', 'warning')
+        break
+
+      case 'error':
+        addLog('流服务错误: ' + (msg.message || ''), 'error')
+        ElMessage.error(msg.message || '流服务出错')
+        break
+    }
+  })
+
+  rawWsClient.on('close', () => {
+    addLog('原始流连接已关闭', 'warning')
+    if (videoMode.value === 'live' && !algorithmRunning.value) {
+      streamRunning.value = false
+      videoMode.value = 'idle'
+    }
+  })
+
+  rawWsClient.on('error', () => {
+    addLog('原始流连接失败', 'error')
+    ElMessage.error('连接原始流服务失败')
+    if (!algorithmRunning.value) {
+      videoMode.value = 'idle'
+      streamRunning.value = false
+    }
+  })
+
+  rawWsClient.connect()
+}
+
+function connectDetectStream() {
+  destroyDetectWs()
+  webrtcUrl.value = ''
+  addLog('正在连接检测流服务...', 'info')
+
+  // 两步握手状态: 0-初始, 1-已发送switch, 2-已发送start_rtsp
+  let handshakeStep = 0
+
+  detectWsClient = new WebSocketClient({
+    url: DETECT_WS_URL,
+    autoReconnect: false,
+    heartbeatInterval: 0,
+    heartbeatTimeout: 0,
+    debug: true
+  })
+
+  detectWsClient.on('open', () => {
+    addLog('检测流服务已连接，开启算法开关...', 'info')
+    handshakeStep = 1
+    detectWsClient.send({ type: 'switch', enabled: true })
+  })
+
+  detectWsClient.on('message', (msg) => {
+    switch (msg.type) {
+      case 'ready':
+        addLog(msg.message || '检测流服务就绪', 'info')
+        // 如果 open 已发 switch 但 ready 晚到, 忽略
+        break
+
+      case 'switch':
+        if (handshakeStep === 1) {
+          // switch 响应收到，发送 start_rtsp 开始检测
+          addLog('算法开关已开启，发送 RTSP 地址', 'info')
+          handshakeStep = 2
+          detectWsClient.send({
+            type: 'start_rtsp',
+            rtsp_url: buildRtspUrl(streamAddress.value),
+            options: {
+              conf: 0.6,
+              iou: 0.45,
+              skip: 5,
+              imgsz: 640,
+              save_video: true,
+              publish_webrtc: true,
+              publish_fps: 25
+            }
+          })
+        }
+        break
+
+      case 'processing_started':
+        addLog(msg.message || 'RTSP 检测已启动', 'success')
+        break
+
+      case 'webrtc_stream':
+        webrtcUrl.value = msg.webrtc_url
+        addLog('已获取检测 WebRTC 流，开始播放', 'success')
+        ElMessage.success('检测流已连接')
+        break
+
+      case 'pause':
+        paused.value = Boolean(msg.paused)
+        addLog(msg.message || (paused.value ? '检测流已暂停' : '检测流已恢复'), 'warning')
+        break
+
+      case 'counts':
+        mergeCounts(msg.counts || [])
+        break
+
+      case 'progress':
+        if (msg.frame) {
+          const progress = msg.total_frames > 0
+            ? (msg.frame / msg.total_frames * 100)
+            : 0
+          detectProgress.value = Math.round(progress)
+          detectStatusText.value = msg.is_rtsp
+            ? `处理中 (第 ${msg.frame} 帧)`
+            : `处理中 (${msg.frame}/${msg.total_frames || '?'})`
+        }
+        break
+
+      case 'log':
+        addLog(msg.message || '', 'info')
+        break
+
+      case 'process_status':
+        addLog(msg.message || '', 'warning')
+        // 更新流程统计
+        processStats.completed += msg.is_complete ? 1 : 0
+        processStats.incomplete += msg.is_complete ? 0 : 1
+        if (msg.should_alert || !msg.is_complete) {
+          const detail = (msg.missing_parts || [])
+            .map(item => `步骤${item.step_num} ${item.step_name}：${item.part_name}`)
+            .join('\n')
+          Object.assign(sopAlert, {
+            visible: true,
+            title: `SOP 流程 #${msg.process_id} 未完成`,
+            content: detail
+          })
+        }
+        break
+
+      case 'stopping':
+        addLog(msg.message || '检测流正在关闭', 'warning')
+        break
+
+      case 'done':
+        addLog(msg.message || '检测完成', 'success')
+        // 切换为离线视频模式，显示处理后的结果视频（类似离线视频逻辑）
+        if (msg.output_video_url) {
+          const videoUrl = HTTP_BASE.replace(/\/$/, '') + msg.output_video_url
+          if (offlineVideoUrl.value) {
+            URL.revokeObjectURL(offlineVideoUrl.value)
+          }
+          offlineVideoUrl.value = videoUrl
+          videoMode.value = 'offline'
+          detectStatusText.value = '检测完成'
+          detectProgress.value = 100
+          addLog('检测完成，视频已更新', 'success')
+          // 流程统计（类似离线视频 done 处理）
+          const processes = msg.processes || []
+          processStats.completed = processes.filter(p => p.is_complete).length
+          processStats.incomplete = processes.filter(p => !p.is_complete).length
+          videoAutoplayTrigger.value++
+          exportVideoUrl.value = videoUrl
+        }
+        if (msg.report_url) {
+          exportReportUrl.value = HTTP_BASE.replace(/\/$/, '') + msg.report_url
+        }
+        if (msg.counts) mergeCounts(msg.counts)
+        if (exportVideoUrl.value || exportReportUrl.value) {
+          canExport.value = true
+        }
+        algorithmRunning.value = false
+        paused.value = false
+        destroyDetectWs()
+        // 执行回调
+        if (detectStopCallback) {
+          const cb = detectStopCallback
+          detectStopCallback = null
+          cb()
+        }
+        break
+
+      case 'error':
+        addLog('检测流错误: ' + (msg.message || ''), 'error')
+        ElMessage.error(msg.message || '检测流出错')
+        algorithmRunning.value = false
+        destroyDetectWs()
+        connectRawStream()
+        break
+    }
+  })
+
+  detectWsClient.on('close', () => {
+    addLog('检测流连接已关闭', 'warning')
+    algorithmRunning.value = false
+  })
+
+  detectWsClient.on('error', () => {
+    addLog('检测流连接失败', 'error')
+    ElMessage.error('连接检测流服务失败')
+    algorithmRunning.value = false
+  })
+
+  detectWsClient.connect()
+}
 
 function switchStream() {
   if (!streamAddress.value) {
     ElMessage.warning('请输入视频流地址')
     return
   }
+  stopRawStream()
   videoMode.value = 'live'
+  webrtcUrl.value = ''
+  algorithmRunning.value = false
+  paused.value = false
   clearAllLiveData()
+  stopDetectStream(() => connectRawStream())
 }
 
 function toggleStream() {
-  streamRunning.value = !streamRunning.value
   if (streamRunning.value) {
-    // 防呆：如果已有上传的离线视频，清空它，避免启动算法按钮未禁用
+    // 关闭视频流
+    stopRawStream()
+    if (algorithmRunning.value) {
+      // 算法正在运行，等待 done 后自动切换到结果视频
+      stopDetectStream(() => {
+        algorithmRunning.value = false
+        streamRunning.value = false
+        webrtcUrl.value = ''
+        paused.value = false
+        ending.value = false
+        ElMessage.info('视频流已关闭')
+        addLog('视频检测完成，已切换到结果视频', 'info')
+      })
+    } else {
+      stopDetectStream()
+      algorithmRunning.value = false
+      streamRunning.value = false
+      videoMode.value = 'idle'
+      webrtcUrl.value = ''
+      paused.value = false
+      ending.value = false
+      ElMessage.info('视频流已关闭')
+      addLog('视频流已关闭', 'info')
+    }
+  } else {
+    // 启动视频流
     if (offlineVideoUrl.value) {
       URL.revokeObjectURL(offlineVideoUrl.value)
       offlineVideoUrl.value = ''
@@ -105,18 +421,23 @@ function toggleStream() {
       resetDetectionState()
       addLog('已清空离线视频文件，如需检测请重新上传', 'warning')
     }
+    if (!streamAddress.value) {
+      ElMessage.warning('请先输入视频流地址')
+      return
+    }
+    streamRunning.value = true
     videoMode.value = 'live'
+    webrtcUrl.value = ''
     clearAllLiveData()
-  } else {
+    // 如果原始流还在运行（由"切换"启动），先关闭
-    videoMode.value = 'idle'
+    if (rawWsClient) {
-    ElMessage.info('视频流已关闭')
+      addLog('正在关闭原始流，切换到推算流...', 'info')
+      stopRawStream()
+    }
+    connectDetectStream()
   }
 }
 
-function onConnectionChange(status) {
-  streamRunning.value = status
-}
-
 // ============ 离线视频相关 ============
 const offlineVideoUrl = ref('')
 const detectionFile = ref(null)
@@ -153,6 +474,7 @@ const DEFAULT_COUNTS = [
 ]
 const detectCounts = reactive(DEFAULT_COUNTS.map(c => ({ ...c })))
 const processStats = reactive({ completed: 0, incomplete: 0 })
+const sopAlert = reactive({ visible: false, title: '', content: '' })
 
 function mergeCounts(newCounts) {
   newCounts.forEach(newItem => {
@@ -232,6 +554,18 @@ function toggleAlgorithm() {
     return
   }
 
+  // 直播模式下关闭算法 → 停止检测流，等待 done 后切换到结果视频
+  if (algorithmRunning.value && videoMode.value === 'live') {
+    addLog('算法已关闭，等待检测结果...', 'info')
+    stopDetectStream(() => {
+      algorithmRunning.value = false
+      paused.value = false
+      ElMessage.info('算法已关闭')
+      addLog('检测结果已保存，切换到结果视频', 'info')
+    })
+    return
+  }
+
   algorithmRunning.value = !algorithmRunning.value
 
   if (algorithmRunning.value) {
@@ -243,7 +577,6 @@ function toggleAlgorithm() {
       resetDetectionState()
       detectStatusText.value = '正在连接 WebSocket...'
       addLog('开始连接算法检测服务', 'info')
-      // 触发视频自动播放
       videoAutoplayTrigger.value++
 
       wsClient = new WebSocketClient({
@@ -331,11 +664,9 @@ function toggleAlgorithm() {
             detectStatusText.value = '检测完成'
             detectProgress.value = 100
             addLog('检测完成，视频已更新', 'success')
-            // 更新流程统计（饼图数据）
             const processes = msg.processes || []
             processStats.completed = processes.filter(p => p.is_complete).length
             processStats.incomplete = processes.filter(p => !p.is_complete).length
-            // 更新检测计数（柱状图数据）
             mergeCounts(msg.counts || [])
             ElMessage.success('视频检测完成')
             algorithmRunning.value = false
@@ -343,7 +674,6 @@ function toggleAlgorithm() {
             ending.value = false
             wsClient.close()
             videoAutoplayTrigger.value++
-            // 保存导出数据
             exportVideoUrl.value = videoUrl
             exportReportUrl.value = msg.report_url ? (HTTP_BASE.replace(/\/$/, '') + msg.report_url) : ''
             canExport.value = true
@@ -379,6 +709,23 @@ function toggleAlgorithm() {
       })
 
       wsClient.connect()
+    } else if (videoMode.value === 'live') {
+      // 直播模式：检测流已在"启动视频流"中连接，仅更新状态
+      algorithmRunning.value = true
+      paused.value = false
+      ending.value = false
+      flowLogs.value = []
+      resetDetectionState()
+      addLog('算法检测已启用', 'success')
+      if (!detectWsClient && !rawWsClient) {
+        // 如果还没有任何流连接，主动连接检测流
+        connectDetectStream()
+      } else if (rawWsClient) {
+        // 如果当前是原始流，切换到检测流
+        addLog('正在关闭原始流，切换到检测流...', 'info')
+        stopRawStream()
+        connectDetectStream()
+      }
     } else if (videoMode.value === 'offline' && !detectionFile.value) {
       ElMessage.warning('请先上传离线视频')
       algorithmRunning.value = false
@@ -408,6 +755,22 @@ function toggleAlgorithm() {
 }
 
 function togglePause() {
+  if (isLiveMode.value && algorithmRunning.value) {
+    // 直播检测模式 → 用检测流 WS
+    if (!detectWsClient || detectWsClient.getReadyState() !== WebSocket.OPEN) return
+    paused.value = !paused.value
+    detectWsClient.send({ type: 'pause', paused: paused.value })
+    addLog(paused.value ? '发送暂停请求...' : '发送恢复请求...', 'info')
+    return
+  }
+  if (isLiveMode.value) {
+    // 原始流模式 → 用原始流 WS
+    if (!rawWsClient || rawWsClient.getReadyState() !== WebSocket.OPEN) return
+    paused.value = !paused.value
+    rawWsClient.send({ type: 'pause', paused: paused.value })
+    addLog(paused.value ? '发送暂停请求...' : '发送恢复请求...', 'info')
+    return
+  }
   if (!wsClient || wsClient.getReadyState() !== WebSocket.OPEN) return
   paused.value = !paused.value
   wsClient.send({ type: 'pause', paused: paused.value })
@@ -444,9 +807,12 @@ onUnmounted(() => {
     URL.revokeObjectURL(offlineVideoUrl.value)
   }
   if (wsClient) {
+    wsClient.send({ type: 'stop' })
     wsClient.destroy()
     wsClient = null
   }
+  stopRawStream()
+  stopDetectStream()
 })
 </script>