feat design 功能迁移

app/api/api_design.py

@@ -0,0 +1,28 @@
+import logging
+import time
+
+from fastapi import APIRouter
+
+from app.schemas.design import DesignModel
+from app.service.design.service import generate
+
+router = APIRouter()
+logger = logging.getLogger()
+
+
+@router.post("/design")
+def design(request_data: DesignModel):
+    try:
+        logger.info(f"design request item is : @@@@@@:{request_data}")
+        code = 200
+        message = "access"
+        start_time = time.time()
+        data = generate(request_data=request_data)
+        logger.info(f"design Run time is @@@@@@:{time.time() - start_time}")
+    except Exception as e:
+        code = 400
+        message = str(e)
+        data = str(e)
+        logger.warning(f"design Run Exception @@@@@@:{e}")
+    logger.info({"code": code, "message": message, "data": data})
+    return {"code": code, "message": message, "data": data}

app/api/api_route.py

@@ -4,6 +4,7 @@ from app.api import api_test
 from app.api import api_super_resolution
 from app.api import api_generate_image
 from app.api import api_attribute_retrieve
+from app.api import api_design
 
 router = APIRouter()
 
@@ -11,3 +12,4 @@ router.include_router(api_test.router, tags=["test"], prefix="/test")
 router.include_router(api_super_resolution.router, tags=["super_resolution"], prefix="/api")
 router.include_router(api_generate_image.router, tags=["generate_image"], prefix="/api")
 router.include_router(api_attribute_retrieve.router, tags=["attribute_retrieve"], prefix="/api")
+router.include_router(api_design.router, tags=['design'], prefix="/api")