import asyncio

from fastapi import UploadFile, File

from ocr.api.message.schemas import CreateMessageRequest
from ocr.api.openai_requests import generate_report, generate_changes
from ocr.api.report import report_router
from ocr.api.report.db_requests import (get_all_reports_obj,
                                        delete_all_reports,
                                        get_report_obj_by_id,
                                        save_report_obj,
                                        get_last_report_obj, change_report_report_obj, change_changes_report_obj)
from ocr.api.report.dto import Paging
from ocr.api.report.model import ReportModel
from ocr.api.report.schemas import AllReportResponse
from ocr.api.utils import divide_images, clean_response, extract_text_from_images
from ocr.core.wrappers import OcrResponseWrapper


@report_router.get('/all')
async def get_all_reports() -> OcrResponseWrapper[AllReportResponse]:
    reports = await get_all_reports_obj()
    response = AllReportResponse(
        paging=Paging(pageSize=len(reports), pageIndex=0, totalCount=len(reports)),
        data=reports
    )
    return OcrResponseWrapper(data=response)


@report_router.delete('/all')
async def delete_all_report() -> OcrResponseWrapper:
    await delete_all_reports()
    return OcrResponseWrapper()


@report_router.get('/{reportId}')
async def get_report(reportId: str) -> OcrResponseWrapper[ReportModel]:
    report = await get_report_obj_by_id(reportId)
    return OcrResponseWrapper(data=report)


@report_router.patch('/{reportId}/report')
async def change_report_report(
        reportId: str,
        data: CreateMessageRequest
) -> OcrResponseWrapper[ReportModel]:
    report = await change_report_report_obj(reportId, data)
    return OcrResponseWrapper(data=report)


@report_router.patch('/{reportId}/changes')
async def change_report_changes(
        reportId: str,
        data: CreateMessageRequest
) -> OcrResponseWrapper[ReportModel]:
    report = await change_changes_report_obj(reportId, data)
    return OcrResponseWrapper(data=report)


@report_router.post('')
async def create_report(
        file: UploadFile = File(...),
) -> OcrResponseWrapper[ReportModel]:
    try:
        last_report, contents = await asyncio.gather(get_last_report_obj(), file.read())
        report, changes = None, None
        images = divide_images(contents)
        text_content = extract_text_from_images(images)
        if last_report:
            report, changes = await asyncio.gather(
                generate_report(text_content),
                generate_changes(text_content, last_report.report)
            )
        else:
            report = await generate_report(text_content)
        report = await save_report_obj(clean_response(report), clean_response(changes), text_content, file.filename)
    finally:
        await file.close()
    return OcrResponseWrapper(data=report)