feat（新功能）:

fix（修复bug）:
docs（文档变更）:
refactor（重构）:
test(增加测试): Agent generate test

app/api/api_agent_generate_image.py

@@ -1,6 +1,8 @@
+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
@@ -11,10 +13,7 @@ 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 data
+    byte_stream = server.get_result(prompt)
+    # 返回流式响应
+    return StreamingResponse(byte_stream, media_type="image/png")

app/service/generate_image/agent_generate.py

@@ -7,6 +7,7 @@
 @Date    ：2023/7/26 12:01:05
 @detail  ：
 """
+import io
 import logging
 from datetime import timedelta
 
@@ -52,15 +53,18 @@ class GenerateImage:
         image = result.as_numpy("generated_image")
         image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_RGB2BGR)
         _, img_byte_array = cv2.imencode('.jpg', image_result)
-        object_name = f'test.jpg'
-        req = oss_upload_image(bucket='test', object_name=object_name, image_bytes=img_byte_array)
-        url = self.minio_client.get_presigned_url(
-            "GET",
-            "test",
-            object_name,
-            expires=timedelta(hours=2),
-        )
-        return url
+        byte_stream = io.BytesIO(img_byte_array)
+        byte_stream.seek(0)
+
+        # object_name = f'test.jpg'
+        # req = oss_upload_image(bucket='test', object_name=object_name, image_bytes=img_byte_array)
+        # url = self.minio_client.get_presigned_url(
+        #     "GET",
+        #     "test",
+        #     object_name,
+        #     expires=timedelta(hours=2),
+        # )
+        return byte_stream
 
 
 if __name__ == '__main__':