20 lines
533 B
Python
20 lines
533 B
Python
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")
|