feat generate 迁移

app/api/api_generate_image.py

@@ -0,0 +1,27 @@
+import logging
+from fastapi import APIRouter, BackgroundTasks
+from app.schemas.generate_image import GenerateImageModel
+from app.service.generate_image.service import GenerateImage, infer_cancel
+
+router = APIRouter()
+logger = logging.getLogger()
+
+
+@router.post("/generate_image")
+def generate_image(request_item: GenerateImageModel, background_tasks: BackgroundTasks):
+    try:
+        service = GenerateImage(request_item)
+        background_tasks.add_task(service.get_result)
+        code = 200
+        message = "access"
+    except Exception as e:
+        code = 400
+        message = e
+        logger.warning(e)
+    return {"code": code, "message": message}
+
+
+@router.get("/generate_cancel/{tasks_id}>")
+def generate_image(tasks_id):
+    result = infer_cancel(tasks_id)
+    return {"code": 200, "message": result['message'], "data": result['data']}

app/api/api_route.py

@@ -2,8 +2,10 @@ from fastapi import APIRouter
 
 from app.api import api_test
 from app.api import api_super_resolution
+from app.api import api_generate_image
 
 router = APIRouter()
 
 router.include_router(api_test.router, tags=["test"], prefix="/test")
-router.include_router(api_super_resolution.router, tags=["api_super_resolution"], prefix="/api")
+router.include_router(api_super_resolution.router, tags=["super_resolution"], prefix="/api")
+router.include_router(api_generate_image.router, tags=["generate_image"], prefix="/api")

app/api/api_test.py

@@ -1,8 +1,6 @@
 import logging
-
 from fastapi import APIRouter
-
-from app.core.config import RABBITMQ_QUEUES
+from app.core.config import SR_RABBITMQ_QUEUES, GI_RABBITMQ_QUEUES
 
 logger = logging.getLogger()
 router = APIRouter()
@@ -10,6 +8,6 @@ router = APIRouter()
 
 @router.get("")
 def test():
-    logger.info(RABBITMQ_QUEUES)
+    logger.info(SR_RABBITMQ_QUEUES)
     logger.info("test")
-    return {"message": RABBITMQ_QUEUES}
+    return {"SR_RABBITMQ_QUEUES message": SR_RABBITMQ_QUEUES, "GI_RABBITMQ_QUEUES": GI_RABBITMQ_QUEUES}