Преглед на файлове

feat: 添加文件路径检查,确保切换数据源时路径有效

zhch158_admin преди 1 седмица
родител
ревизия
788e93532b
променени са 1 файла, в които са добавени 4 реда и са изтрити 0 реда
  1. 4 0
      streamlit_validator_core.py

+ 4 - 0
streamlit_validator_core.py

@@ -81,6 +81,8 @@ class StreamlitOCRValidator:
                 print(f"✅ 切换到OCR数据源: {source_key}")
             else:
                 print(f"⚠️ 数据源 {source_key} 没有可用文件")
+        else:
+            raise FileNotFoundError(f"找不到文件路径: {source_key}")
     
     def switch_to_verify_source(self, source_key: str):
         """切换到指定验证数据源"""
@@ -96,6 +98,8 @@ class StreamlitOCRValidator:
                 print(f"✅ 切换到验证数据源: {source_key}")
             else:
                 print(f"⚠️ 验证数据源 {source_key} 没有可用文件")
+        else:
+            raise FileNotFoundError(f"找不到文件路径: {source_key}")
 
     def load_ocr_data(self, json_path: str, md_path: Optional[str] = None, image_path: Optional[str] = None):
         """加载OCR相关数据"""