Function extract_elements

modules/sam3_info_extractor.py:1091–1129

快捷函数 - 一行代码提取元素 Args: image_path: 图片路径 groups: 要处理的组列表（默认全部） Returns: ProcessingResult 使用示例: # 提取所有元素 result = extract_elements("test.png") # 只提取图片和箭头 result = extract_elements("test.png", groups

(image_path: str, 
                     groups: List[PromptGroup] = None)

# ======================== 快捷函数 ========================
def extract_elements(image_path: str, 
                     groups: List[PromptGroup] = None) -> ProcessingResult:
    """
    快捷函数 - 一行代码提取元素
    
    Args:
        image_path: 图片路径
        groups: 要处理的组列表（默认全部）
        
    Returns:
        ProcessingResult
        
    使用示例:
        # 提取所有元素
        result = extract_elements("test.png")
        
        # 只提取图片和箭头
        result = extract_elements("test.png", groups=[PromptGroup.IMAGE, PromptGroup.ARROW])
    """
    extractor = Sam3InfoExtractor()
    context = ProcessingContext(image_path=image_path)
    
    if groups:
        all_elements = []
        for group in groups:
            result = extractor.extract_by_group(context, group)
            all_elements.extend(result.elements)
        
        for i, elem in enumerate(all_elements):
            elem.id = i
        
        return ProcessingResult(
            success=True,
            elements=all_elements,
            canvas_width=context.canvas_width,
            canvas_height=context.canvas_height
        )
    
    return extractor.process(context)

def extract_with_prompts(image_path: str, 
                         prompts: List[str],