Procházet zdrojové kódy

feat: 修改初始化方法,支持通过配置字典传入配置,移除对 load_config 的依赖

zhch158_admin před 1 týdnem
rodič
revize
19be083b28
1 změnil soubory, kde provedl 9 přidání a 3 odebrání
  1. 9 3
      streamlit_validator_core.py

+ 9 - 3
streamlit_validator_core.py

@@ -7,7 +7,7 @@ from typing import Dict, List, Optional
 import json
 
 from ocr_validator_utils import (
-    load_config, load_ocr_data_file, process_ocr_data,
+    load_ocr_data_file, process_ocr_data,
     get_ocr_statistics, find_available_ocr_files_multi_source, 
     get_data_source_display_name
 )
@@ -17,8 +17,14 @@ from ocr_validator_layout import OCRLayoutManager
 class StreamlitOCRValidator:
     """核心验证器类"""
     
-    def __init__(self):
-        self.config = load_config()
+    def __init__(self, config_dict: Dict = None):  # 🎯 参数名改为 config_dict
+        """
+        初始化验证器
+        
+        Args:
+            config_dict: 配置字典(从 ConfigManager.to_validator_config() 生成)
+        """
+        self.config = config_dict  # 🎯 直接赋值
         self.ocr_data = []
         self.md_content = ""
         self.image_path = ""