Class AutoFigureAgent

autofigure/agent.py:50–468

Main class for generating scientific figures using AI. Example usage: ```python from autofigure import AutoFigureAgent, Config config = Config(generation_api_key="your-api-key") agent = AutoFigureAgent(config) result = agent.generate( d

class AutoFigureAgent:
    """
    Main class for generating scientific figures using AI.

    Example usage:
        ```python
        from autofigure import AutoFigureAgent, Config

        config = Config(generation_api_key="your-api-key")
        agent = AutoFigureAgent(config)

        result = agent.generate(
            description="A flowchart showing the training pipeline",
            max_iterations=5
        )

        print(f"Generated: {result.svg_path}")
        ```
    """

    def __init__(self, config: Config):
        """
        Initialize AutoFigureAgent.

        Args:
            config: AutoFigure configuration
        """
        self.config = config
        self._extractor = None
        self._enhancer = None

        # Validate configuration
        errors = config.validate()
        if errors:
            print(f"[AutoFigureAgent] Configuration warnings: {errors}")

        # Update generator CONFIG from SDK config
        update_config_from_sdk(config)

    @property
    def extractor(self) -> MethodologyExtractor:
        """Lazy-load methodology extractor."""
        if self._extractor is None:
            self._extractor = MethodologyExtractor(self.config)
        return self._extractor

    @property
    def enhancer(self) -> ImageEnhancer:
        """Lazy-load image enhancer."""
        if self._enhancer is None:
            self._enhancer = ImageEnhancer(self.config)
        return self._enhancer

    def generate(
        self,
        description: str,
        max_iterations: Optional[int] = None,
        output_format: str = "svg",