from __future__ import annotations

from app.schemas.chart_ir import Axis, ChartIR, ChartSeries
from app.storage.models import VisualAssetRecord

from .base import Lens


class ChartLens(Lens):
    name = "chart_ir"
    view_type = "chart_ir"

    async def execute(self, asset: VisualAssetRecord, image_bytes: bytes, parameters: dict) -> tuple[dict, float, list[dict]]:
        ir = ChartIR(
            chart_type="bar",
            title="Extracted Chart",
            plot_bbox=(0.1, 0.1, 0.9, 0.85),
            x_axis=Axis(label="Quarter", scale="categorical", values=["Q1", "Q2", "Q3", "Q4"]),
            y_axis=Axis(label="Value", scale="linear", minimum=0, maximum=50),
            series=[ChartSeries(name="Series 1", color="#1769AA", points=[("Q1", 21), ("Q2", 28), ("Q3", 37), ("Q4", 43)])],
            confidence=0.74,
        )
        provenance = [{"asset_id": asset.id, "bbox": [0.1, 0.1, 0.9, 0.85], "extraction_method": self.name, "confidence": 0.74, "lens_version": self.version}]
        return ir.model_dump(), 0.74, provenance