21 lines
582 B
Python
21 lines
582 B
Python
|
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)
|