from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
from app.models.load_emotional_model import llm
from app.services.document_loader import extract_text_from_pdf_path
from app.services.create_prompt import create_prompt_emotion

router = APIRouter()


class PdfRequest(BaseModel):
    emotion: str
    course_id: str
    lesson_id: str


@router.post("/rewrite-pdf-emotion")
async def rewrite_pdf_emotion(request: PdfRequest) -> dict:
    try:
        prompt = create_prompt_emotion()
        text = extract_text_from_pdf_path(request.course_id, request.lesson_id)
        chain = prompt | llm
        rewritten_text = chain.invoke({"style": request.emotion, "content": text})
        return {"rewritten_text": rewritten_text}

    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))