import logging
import time

from fastapi import APIRouter

from app.schemas.prompt_generation import PromptGenerationImageModel
from app.service.prompt_generation.chatgpt_for_translation import translate_to_en

router = APIRouter()
logger = logging.getLogger()


@router.post("/translateToEN")
def prompt_generation(request_data: PromptGenerationImageModel):
    try:
        logger.info(f"prompt_translate to English request data :  @@@@@@:{request_data}")
        code = 200
        message = "OK!"
        start_time = time.time()
        data = translate_to_en(request_data.text)
        logger.info(f"prompt_generation Run time is @@@@@@:{time.time() - start_time}")
    except Exception as e:
        code = 400
        message = str(e)
        data = str(e)
        logger.warning(f"prompt_generation Run Exception @@@@@@:{e}")
    logger.info({"code": code, "message": message, "data": data})
    return {"code": code, "msg": message, "data": data}