Prechádzať zdrojové kódy

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

zhch158_admin 1 týždeň pred
rodič
commit
788e93532b
1 zmenil súbory, kde vykonal 4 pridanie a 0 odobranie
  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相关数据"""