import io
import logging

from fastapi import APIRouter, HTTPException
from starlette.responses import StreamingResponse

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):
    server = GenerateImage()
    byte_stream = server.get_result(prompt)
    # 返回流式响应
    return StreamingResponse(byte_stream, media_type="image/png")