Method process

modules/text/restorer.py:92–133

Process image and return draw.io XML string. Args: image_path: Input image path. Returns: draw.io XML string.

(self, image_path: str)

        return self._pix2text_ocr
    
    def process(self, image_path: str) -> str:
        """
        Process image and return draw.io XML string.

        Args:
            image_path: Input image path.
        Returns:
            draw.io XML string.
        """
        image_path = Path(image_path)

        with Image.open(image_path) as img:
            image_width, image_height = img.size
        
        text_blocks = self.process_image(str(image_path))

        generator = MxGraphXMLGenerator(
            diagram_name=image_path.stem,
            page_width=image_width,
            page_height=image_height
        )
        
        text_cells = []
        for block in text_blocks:
            geo = block["geometry"]
            cell = generator.create_text_cell(
                text=block["text"],
                x=geo["x"],
                y=geo["y"],
                width=max(geo["width"], 20),
                height=max(geo["height"], 10),
                font_size=block.get("font_size", 12),
                is_latex=block.get("is_latex", False),
                rotation=geo.get("rotation", 0),
                font_weight=block.get("font_weight"),
                font_style=block.get("font_style"),
                font_color=block.get("font_color"),
                font_family=block.get("font_family")
            )
            text_cells.append(cell)
        
        return generator.generate_xml(text_cells)
    
    def process_image(self, image_path: str) -> List[Dict[str, Any]]:
        """Process image and return list of text blocks."""