const API_ENDPOINT = "/api/detect"; const VIEW_LABELS = { overview: "系统总览", inspection: "视频巡检", events: "告警中心", risks: "风险台账", channels: "通知通道", settings: "系统设置", }; const STATUS_LABELS = { pending: "待处置", acknowledged: "已确认", resolved: "已解决", }; const elements = Object.fromEntries( [ "ackEventsCount", "alertStatus", "allEventsCount", "channelCount", "channelSummary", "clearButton", "confirmFramesValue", "cooldownValue", "currentClock", "detectionList", "emptyState", "eventTableBody", "eventTableMeta", "feishuChannelBadge", "fireCount", "globalStatus", "inspectionStatus", "intervalSelect", "lastUpdated", "loadingState", "maxConfidence", "menuButton", "modelStatusDetail", "modelStatusValue", "notificationDot", "overlayCanvas", "overviewMessage", "pendingEventCount", "pendingEventsCount", "pendingNavCount", "readyApi", "readyFeishu", "readyWechat", "readyWeights", "recentEventList", "refreshEventsButton", "resolvedEventsCount", "riskChannelValue", "riskConfidenceValue", "riskPendingValue", "riskStatus", "runDetectionButton", "settingConfidence", "settingConfirmFrames", "settingCooldown", "settingImageSize", "settingIou", "settingWeights", "sidebar", "sidebarStatus", "sidebarStatusDot", "smokeCount", "sourceLabel", "todayEventCount", "toastRegion", "trendChart", "videoClock", "videoInput", "videoMeta", "videoPreview", "viewBreadcrumb", "viewTitle", "wechatChannelBadge", ].map((id) => [id, document.querySelector(`#${id}`)]) ); const frameCanvas = document.createElement("canvas"); let videoUrl = null; let sessionId = null; let detectionActive = false; let requestInFlight = false; let lastDetectionAt = 0; let dashboardData = null; let eventData = []; let eventFilter = ""; function showToast(message, type = "info") { const toast = document.createElement("div"); toast.className = `toast ${type}`; toast.textContent = message; elements.toastRegion.appendChild(toast); setTimeout(() => toast.remove(), 3600); } function switchView(viewName) { if (!VIEW_LABELS[viewName]) return; document.querySelectorAll("[data-view-panel]").forEach((panel) => { panel.classList.toggle("is-active", panel.dataset.viewPanel === viewName); }); document.querySelectorAll("[data-view]").forEach((button) => { button.classList.toggle("is-active", button.dataset.view === viewName); }); elements.viewTitle.textContent = VIEW_LABELS[viewName]; elements.viewBreadcrumb.textContent = VIEW_LABELS[viewName]; elements.sidebar.classList.remove("is-open"); if (viewName === "events") loadEvents(); if (["overview", "risks", "channels", "settings"].includes(viewName)) loadDashboard(); } function setServiceStatus(online) { elements.globalStatus.textContent = online ? "服务运行正常" : "服务连接异常"; elements.globalStatus.className = `status-chip ${online ? "status-online" : "status-offline"}`; elements.sidebarStatus.textContent = online ? "服务在线" : "服务离线"; elements.sidebarStatusDot.className = `status-dot ${online ? "is-online" : "is-offline"}`; elements.readyApi.textContent = online ? "正常" : "异常"; elements.readyApi.className = online ? "ready" : "not-ready"; } function enabledChannelNames(channels = {}) { const names = []; if (channels.wechat) names.push("企业微信"); if (channels.feishu) names.push("飞书"); return names; } function updateClock() { elements.currentClock.textContent = new Date().toLocaleString("zh-CN", { hour12: false, month: "2-digit", day: "2-digit", hour: "2-digit", minute: "2-digit", second: "2-digit", }); } function formatDate(value) { if (!value) return "--"; return new Date(value).toLocaleString("zh-CN", { hour12: false }); } function formatTime(seconds) { if (!Number.isFinite(seconds)) return "00:00"; const minutes = Math.floor(seconds / 60); const remaining = Math.floor(seconds % 60); return `${String(minutes).padStart(2, "0")}:${String(remaining).padStart(2, "0")}`; } function updateDashboard(data) { dashboardData = data; const { summary, system, detection } = data; const channels = system.alert_channels || {}; const channelNames = enabledChannelNames(channels); elements.todayEventCount.textContent = String(summary.today); elements.pendingEventCount.textContent = String(summary.pending); elements.pendingNavCount.textContent = String(summary.pending); elements.notificationDot.hidden = summary.pending === 0; elements.modelStatusValue.textContent = system.weights_available ? "运行正常" : "权重缺失"; elements.modelStatusDetail.textContent = system.weights_available ? "模型文件已就绪" : "请检查 YOLO_WEIGHTS"; elements.modelStatusValue.className = "text-value"; elements.channelCount.textContent = String(channelNames.length); elements.channelSummary.textContent = channelNames.length ? channelNames.join(" + ") : "尚未启用机器人"; elements.overviewMessage.textContent = summary.pending ? `当前有 ${summary.pending} 条告警事件等待处置,请尽快进入告警中心确认。` : "当前无待处置事件,模型与视频巡检服务保持监测状态。"; elements.readyWeights.textContent = system.weights_available ? "已就绪" : "缺失"; elements.readyWeights.className = system.weights_available ? "ready" : "not-ready"; setChannelReady(elements.readyWechat, channels.wechat); setChannelReady(elements.readyFeishu, channels.feishu); setChannelBadge(elements.wechatChannelBadge, channels.wechat); setChannelBadge(elements.feishuChannelBadge, channels.feishu); elements.riskPendingValue.textContent = String(summary.pending); elements.riskChannelValue.textContent = `${channelNames.length} / 2`; elements.riskConfidenceValue.textContent = `${Math.round(detection.confidence * 100)}%`; elements.confirmFramesValue.textContent = `${detection.confirm_frames} 帧`; elements.cooldownValue.textContent = `${detection.cooldown_seconds} 秒`; elements.settingWeights.textContent = system.weights; elements.settingImageSize.textContent = `${detection.image_size} px`; elements.settingConfidence.textContent = detection.confidence.toFixed(2); elements.settingIou.textContent = detection.iou.toFixed(2); elements.settingConfirmFrames.textContent = `${detection.confirm_frames} 帧`; elements.settingCooldown.textContent = `${detection.cooldown_seconds} 秒`; renderRecentEvents(data.recent_events || []); renderTrend(eventData.length ? eventData : data.recent_events || []); } function setChannelReady(element, enabled) { element.textContent = enabled ? "已启用" : "未配置"; element.className = enabled ? "ready" : "optional"; } function setChannelBadge(element, enabled) { element.textContent = enabled ? "已启用" : "未配置"; element.className = `channel-badge ${enabled ? "enabled" : ""}`; } async function loadDashboard() { try { const response = await fetch("/api/dashboard"); if (!response.ok) throw new Error(`API ${response.status}`); const data = await response.json(); setServiceStatus(true); updateDashboard(data); } catch (error) { setServiceStatus(false); showToast(`无法读取管理数据:${error.message}`, "error"); } } async function loadEvents() { try { const query = eventFilter ? `?status=${eventFilter}` : ""; const response = await fetch(`/api/events${query}`); if (!response.ok) throw new Error(`API ${response.status}`); const result = await response.json(); eventData = result.events || []; renderEventTable(eventData); updateEventSummary(result.summary); renderTrend(eventData); setServiceStatus(true); } catch (error) { setServiceStatus(false); showToast(`无法读取告警事件:${error.message}`, "error"); } } function updateEventSummary(summary) { elements.allEventsCount.textContent = String(summary.total); elements.pendingEventsCount.textContent = String(summary.pending); elements.ackEventsCount.textContent = String(summary.acknowledged); elements.resolvedEventsCount.textContent = String(summary.resolved); elements.pendingNavCount.textContent = String(summary.pending); elements.notificationDot.hidden = summary.pending === 0; } function renderRecentEvents(events) { if (!events.length) { elements.recentEventList.innerHTML = "
暂无告警事件。开始视频巡检后,满足连续帧条件的告警会显示在这里。
"; return; } elements.recentEventList.innerHTML = events.map((event) => { const primary = event.classes.includes("fire") ? "fire" : "smoke"; const label = event.classes.map(classLabel).join("、"); return `暂无检测结果
'; const context = elements.overlayCanvas.getContext("2d"); context.clearRect(0, 0, elements.overlayCanvas.width, elements.overlayCanvas.height); } function drawDetections(detections = []) { const video = elements.videoPreview; if (!video.videoWidth || !video.videoHeight) return; const canvas = elements.overlayCanvas; canvas.width = video.videoWidth; canvas.height = video.videoHeight; const context = canvas.getContext("2d"); context.clearRect(0, 0, canvas.width, canvas.height); detections.forEach((detection) => { const [x1, y1, x2, y2] = detection.box || []; const color = detection.class === "fire" ? "#ff665e" : "#41d6c3"; context.strokeStyle = color; context.lineWidth = Math.max(2, canvas.width / 320); context.strokeRect(x1, y1, x2 - x1, y2 - y1); context.fillStyle = color; context.font = `${Math.max(13, canvas.width / 60)}px sans-serif`; context.fillText(`${detection.class} ${Math.round(detection.confidence * 100)}%`, x1 + 4, Math.max(18, y1 - 6)); }); } function updateDetectionResults(result) { const detections = result.detections || []; const smoke = detections.filter((item) => item.class === "smoke").length; const fire = detections.filter((item) => item.class === "fire").length; const max = detections.reduce((highest, item) => Math.max(highest, item.confidence || 0), 0); elements.smokeCount.textContent = String(smoke); elements.fireCount.textContent = String(fire); elements.maxConfidence.textContent = max ? `${Math.round(max * 100)}%` : "--"; elements.riskStatus.textContent = fire ? "高风险" : smoke ? "需关注" : "正常"; elements.detectionList.innerHTML = detections.length ? detections.map((item) => `未发现烟雾或火焰目标
'; const alert = result.alert || {}; const channelNames = enabledChannelNames(alert.notification_channels); if (alert.triggered) { const labels = alert.classes.map(classLabel).join("、"); elements.alertStatus.textContent = channelNames.length ? `已触发 ${channelNames.join(" + ")} 告警:${labels}` : `已生成告警事件:${labels}(机器人未配置)`; elements.alertStatus.className = "alert-status alert-triggered"; showToast(`检测到${labels},已生成告警事件`, "error"); loadDashboard(); } else { const fireFrames = alert.consecutive?.fire || 0; const smokeFrames = alert.consecutive?.smoke || 0; elements.alertStatus.textContent = channelNames.length ? `${channelNames.join(" + ")}已启用 · 连续帧 火焰 ${fireFrames} / 烟雾 ${smokeFrames}` : `机器人未配置 · 连续帧 火焰 ${fireFrames} / 烟雾 ${smokeFrames}`; elements.alertStatus.className = "alert-status"; } drawDetections(detections); elements.lastUpdated.textContent = `推理 ${result.inference_ms ?? "--"} ms`; } function captureFrame() { const video = elements.videoPreview; frameCanvas.width = video.videoWidth; frameCanvas.height = video.videoHeight; frameCanvas.getContext("2d").drawImage(video, 0, 0); return new Promise((resolve) => frameCanvas.toBlob(resolve, "image/jpeg", 0.88)); } async function detectCurrentFrame(timestamp) { if (!detectionActive || requestInFlight || !sessionId || elements.videoPreview.paused || elements.videoPreview.ended) return; const interval = Number(elements.intervalSelect.value); if (timestamp - lastDetectionAt < interval) return; lastDetectionAt = timestamp; requestInFlight = true; elements.loadingState.hidden = false; try { const frame = await captureFrame(); if (!frame) throw new Error("无法截取视频帧"); const form = new FormData(); form.append("file", frame, "video-frame.jpg"); const response = await fetch(`${API_ENDPOINT}?session_id=${encodeURIComponent(sessionId)}`, { method: "POST", body: form }); if (!response.ok) throw new Error(`API ${response.status}`); updateDetectionResults(await response.json()); elements.inspectionStatus.textContent = "检测运行中"; elements.inspectionStatus.className = "status-chip status-online"; setServiceStatus(true); } catch (error) { elements.inspectionStatus.textContent = "检测异常"; elements.inspectionStatus.className = "status-chip status-offline"; showToast(`视频检测失败:${error.message}`, "error"); } finally { requestInFlight = false; elements.loadingState.hidden = true; } } function scheduleDetection(timestamp) { detectCurrentFrame(timestamp); if (detectionActive) requestAnimationFrame(scheduleDetection); } async function startDetection() { if (!elements.videoPreview.src) return; detectionActive = true; lastDetectionAt = -Infinity; elements.runDetectionButton.textContent = "停止检测"; elements.intervalSelect.disabled = true; try { await elements.videoPreview.play(); requestAnimationFrame(scheduleDetection); } catch { stopDetection(); showToast("浏览器未允许视频播放,请手动点击播放后重试。", "error"); } } function stopDetection() { detectionActive = false; elements.runDetectionButton.textContent = "开始连续检测"; elements.intervalSelect.disabled = false; elements.inspectionStatus.textContent = elements.videoPreview.src ? "检测已暂停" : "等待视频"; elements.inspectionStatus.className = "status-chip status-idle"; } async function resetSession() { const previous = sessionId; sessionId = crypto.randomUUID(); if (previous) fetch(`/api/sessions/${encodeURIComponent(previous)}`, { method: "DELETE" }).catch(() => {}); } document.querySelectorAll("[data-view], [data-view-jump]").forEach((control) => { control.addEventListener("click", () => switchView(control.dataset.view || control.dataset.viewJump)); }); document.querySelectorAll("[data-event-filter]").forEach((button) => { button.addEventListener("click", () => { eventFilter = button.dataset.eventFilter; document.querySelectorAll("[data-event-filter]").forEach((item) => item.classList.toggle("is-active", item === button)); loadEvents(); }); }); elements.eventTableBody.addEventListener("click", (event) => { const button = event.target.closest("[data-event-id]"); if (button) updateEventStatus(button.dataset.eventId, button.dataset.eventStatus); }); elements.menuButton.addEventListener("click", () => elements.sidebar.classList.toggle("is-open")); elements.refreshEventsButton.addEventListener("click", loadEvents); elements.runDetectionButton.addEventListener("click", () => detectionActive ? stopDetection() : startDetection()); elements.videoPreview.addEventListener("ended", stopDetection); elements.videoPreview.addEventListener("timeupdate", () => { elements.videoClock.textContent = `${formatTime(elements.videoPreview.currentTime)} / ${formatTime(elements.videoPreview.duration)}`; }); elements.videoPreview.addEventListener("seeked", async () => { drawDetections([]); await resetSession(); }); elements.videoPreview.addEventListener("resize", () => drawDetections([])); elements.videoInput.addEventListener("change", async () => { const [file] = elements.videoInput.files; if (!file) return; stopDetection(); elements.videoPreview.pause(); if (videoUrl) URL.revokeObjectURL(videoUrl); videoUrl = URL.createObjectURL(file); elements.videoPreview.src = videoUrl; elements.videoPreview.hidden = false; elements.emptyState.hidden = true; elements.sourceLabel.textContent = file.name; elements.videoMeta.textContent = `${(file.size / 1024 / 1024).toFixed(1)} MB · 本地视频`; elements.runDetectionButton.disabled = false; elements.inspectionStatus.textContent = "视频已就绪"; elements.inspectionStatus.className = "status-chip status-online"; await resetSession(); clearDetectionResults(); }); elements.clearButton.addEventListener("click", async () => { stopDetection(); elements.videoPreview.pause(); elements.videoPreview.removeAttribute("src"); elements.videoPreview.load(); elements.videoPreview.hidden = true; if (videoUrl) URL.revokeObjectURL(videoUrl); videoUrl = null; elements.videoInput.value = ""; elements.sourceLabel.textContent = "未选择视频"; elements.videoMeta.textContent = "支持 MP4、WebM 等浏览器可播放格式"; elements.emptyState.hidden = false; elements.runDetectionButton.disabled = true; await resetSession(); clearDetectionResults(); }); updateClock(); setInterval(updateClock, 1000); resetSession(); Promise.all([loadDashboard(), loadEvents()]);