from typing import List

from sqlmodel import Session
from entity.frame_analysis_result import FrameAnalysisResultCreate, FrameAnalysisResult


class FrameAnalysisResultService:

    def __init__(self, db: Session):
        self.db = db

    def add_frame_analysis_results(self, results: List[FrameAnalysisResultCreate]):
        new_results = [FrameAnalysisResult.model_validate(result) for result in results]
        self.db.add_all(new_results)
        self.db.commit()
        for result in new_results:
            self.db.refresh(result)
        return new_results