design batch 代码整理

app/api/api_design.py

@@ -6,8 +6,9 @@ from fastapi import APIRouter, HTTPException, UploadFile, File, Form
 
 from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel, DBGConfigModel
 from app.schemas.response_template import ResponseModel
-from app.service.design.model_process_service import model_transpose
-from app.service.design.service_design_batch_generate import start_design_batch_generate
+from app.service.design_batch.service import start_design_batch_generate
+# from app.service.design.model_process_service import model_transpose
+# from app.service.design.service_design_batch_generate import start_design_batch_generate
 from app.service.design_fast.design_generate import design_generate
 from app.service.design_fast.utils.redis_utils import Redis
 
@@ -236,7 +237,7 @@ def model_process(request_data: ModelProgressModel):
     try:
         logger.info(f"model_process request item is : @@@@@@:{json.dumps(request_data.dict())}")
 
-        data = model_transpose(image_path=request_data.model_path)
+        # data = model_transpose(image_path=request_data.model_path)
         logger.info(f"model_process response @@@@@@:{json.dumps(data)}")
     except Exception as e:
         logger.warning(f"model_process Run Exception @@@@@@:{e}")
@@ -251,20 +252,18 @@ def model_process(request_data: ModelProgressModel):
 async def design(file: UploadFile = File(...),
                  tasks_id: str = Form(...),
                  user_id: str = Form(...),
-                 priority: int = Form(...),
+                 file_name: str = Form(...),
                  total: int = Form(...)
                  ):
-    # file_content = await file.read()
     dbg_config = DBGConfigModel(
         tasks_id=tasks_id,
         user_id=user_id,
-        priority=priority,
+        file_name=file_name,
         total=total
     )
     contents = await file.read()
     file_name = file.filename
     await save_request_file(contents, file_name)
-
     return await start_design_batch_generate(dbg_config, contents)