File size: 866 Bytes
624109c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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))
|