from fastapi import APIRouter
from pydantic import BaseModel
from services.classification_service import (
    classify_issue,
    summarize_issue,
    classify_and_summarize,
)

router = APIRouter(tags=["classification"])


class ClassifyRequest(BaseModel):
    description: str
    detail_notes: str = ""


class SummarizeRequest(BaseModel):
    description: str
    detail_notes: str = ""
    solution: str = ""


@router.post("/classify")
async def classify(req: ClassifyRequest):
    try:
        result = await classify_issue(req.description, req.detail_notes)
        return {"available": True, **result}
    except Exception as e:
        return {"available": False, "error": str(e)}


@router.post("/summarize")
async def summarize(req: SummarizeRequest):
    try:
        result = await summarize_issue(req.description, req.detail_notes, req.solution)
        return {"available": True, **result}
    except Exception as e:
        return {"available": False, "error": str(e)}


@router.post("/classify-and-summarize")
async def classify_and_summarize_endpoint(req: ClassifyRequest):
    try:
        result = await classify_and_summarize(req.description, req.detail_notes)
        return {"available": True, **result}
    except Exception as e:
        return {"available": False, "error": str(e)}