import logging

from fastapi import APIRouter, HTTPException

from app.schemas.response_template import ResponseModel
from app.service.generate_image.agent_generate import GenerateImage

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


@router.get("/agent_generate_image")
def generate_image(prompt: str):
    try:
        server = GenerateImage()
        data = server.get_result(prompt)
    except Exception as e:
        logger.warning(f"generate_image Run Exception @@@@@@:{e}")
        raise HTTPException(status_code=404, detail=str(e))
    return ResponseModel(data=data)