From c0746aab57aa0ca40759b05d422bfe74f9bc95fb Mon Sep 17 00:00:00 2001
From: scdotbox <128663994+scdotbox@users.noreply.github.com>
Date: Sat, 3 May 2025 15:20:04 +0800
Subject: [PATCH] Update video_reader.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

增加本地视频下载文件的检查
---
 backend/app/utils/video_reader.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/backend/app/utils/video_reader.py b/backend/app/utils/video_reader.py
index 7085610..6922bd6 100644
--- a/backend/app/utils/video_reader.py
+++ b/backend/app/utils/video_reader.py
@@ -104,6 +104,9 @@ class VideoReader:
     def run(self)->list[str]:
         logger.info("🚀 开始提取视频帧...")
         try:
+            # 确保目录存在
+            os.makedirs(self.frame_dir, exist_ok=True)
+            os.makedirs(self.grid_dir, exist_ok=True)
             #清空帧文件夹
             for file in os.listdir(self.frame_dir):
                 if file.startswith("frame_"):