import json
import logging
from common.db import fetchone, execute, get_analysis_raw_data, save_analysis_report
from integrations.llm.llm_service import LLMService
from integrations.llm.prompt import report_prompt, plan_prompt
from integrations.llm.schemas.report import ReportOutput
from integrations.llm.schemas.plan import PlanOutput
from models.status import AnalysisStatus

logger = logging.getLogger(__name__)


async def generate_report(analysis_run_id: str) -> ReportOutput:
    run = await fetchone(
        "SELECT hospital_id FROM analysis_runs WHERE analysis_run_id = %s",
        (analysis_run_id,),
    )
    clinic_row = await fetchone(
        "SELECT raw_data FROM hospital_baseinfo WHERE hospital_id = %s",
        (run["hospital_id"],),
    )
    raw_data = clinic_row["raw_data"] if clinic_row else None
    clinic = json.loads(raw_data) if isinstance(raw_data, str) else (raw_data or {})
    raw = await get_analysis_raw_data(analysis_run_id)

    input_data = {
        "clinic_name":    clinic.get("clinicName"),
        "clinic_name_en": clinic.get("clinicNameEn"),
        "address":        clinic.get("address"),
        "phone":          clinic.get("phone"),
        "slogan":         clinic.get("slogan"),
        "services":       json.dumps(clinic.get("services", []), ensure_ascii=False),
        "doctors":        json.dumps(clinic.get("doctors", []), ensure_ascii=False),
        **{
            channel: json.dumps(data, ensure_ascii=False) if data else None
            for channel, data in raw.items()
        },
    }

    return await LLMService(provider="perplexity").generate(report_prompt, input_data)


async def generate_plan(analysis_run_id: str) -> PlanOutput:
    run = await fetchone(
        "SELECT hospital_id, report_data FROM analysis_runs WHERE analysis_run_id = %s",
        (analysis_run_id,),
    )
    clinic_row = await fetchone(
        "SELECT raw_data FROM hospital_baseinfo WHERE hospital_id = %s",
        (run["hospital_id"],),
    )
    raw_data = clinic_row["raw_data"] if clinic_row else None
    clinic = json.loads(raw_data) if isinstance(raw_data, str) else (raw_data or {})
    report_data = run["report_data"]
    report = json.loads(report_data) if isinstance(report_data, str) else report_data

    input_data = {
        "clinic_name":    clinic.get("clinicName"),
        "clinic_name_en": clinic.get("clinicNameEn"),
        "address":        clinic.get("address"),
        "phone":          clinic.get("phone"),
        "slogan":         clinic.get("slogan"),
        "services":       json.dumps(clinic.get("services", []), ensure_ascii=False),
        "doctors":        json.dumps(clinic.get("doctors", []), ensure_ascii=False),
        "report":         json.dumps(report, ensure_ascii=False) if report else None,
    }

    return await LLMService(provider="perplexity").generate(plan_prompt, input_data)


async def run_report_task(analysis_run_id: str) -> None:
    logger.info("[report] start run=%s", analysis_run_id)
    result = await generate_report(analysis_run_id)
    await save_analysis_report(analysis_run_id, result.model_dump())
    logger.info("[report] done run=%s", analysis_run_id)


async def run_plan_task(analysis_run_id: str) -> None:
    logger.info("[plan] start run=%s", analysis_run_id)
    result = await generate_plan(analysis_run_id)
    await execute(
        "UPDATE analysis_runs SET plan_data = %s WHERE analysis_run_id = %s",
        (json.dumps(result.model_dump(), ensure_ascii=False), analysis_run_id),
    )
    logger.info("[plan] done run=%s", analysis_run_id)