Explorar o código

refactor: Change 'ocr_data' type hint to 'Dict' and replace 'tool' with 'json_format' in annotation mode setup for improved clarity

zhch158_admin hai 10 horas
pai
achega
1126fcc6da
Modificáronse 1 ficheiros con 3 adicións e 3 borrados
  1. 3 3
      table_line_generator/editor/mode_setup.py

+ 3 - 3
table_line_generator/editor/mode_setup.py

@@ -20,7 +20,7 @@ from .analysis_controls import create_analysis_section
 
 
 def setup_new_annotation_mode(
-    ocr_data: list,
+    ocr_data: Dict,
     image: Image.Image,
     display_config: Dict
 ) -> Tuple:
@@ -31,7 +31,7 @@ def setup_new_annotation_mode(
         (generator, structure, undo_stack, line_width, display_mode, zoom_level, show_line_numbers)
     """
     # 🔑 获取当前工具类型
-    tool = st.session_state.get('current_tool', 'ppstructv3')
+    json_format = st.session_state.get('current_json_format', 'ppstructure')
     
     # 初始化生成器
     if 'generator' not in st.session_state:
@@ -53,7 +53,7 @@ def setup_new_annotation_mode(
     # 分析控件
     structure = create_analysis_section(
         st.session_state.generator,
-        tool=tool  # 🔑 传入工具类型
+        json_format=json_format  # 🔑 传入数据格式类型
     )
     
     # 显示控件