aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat

Browse Source 
fix generate 取消接口修复
This commit is contained in:
zhouchengrong
2024-06-24 16:52:00 +08:00
parent 26a56b4d9f
commit 2a83399eff
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/api/api_generate_image.py
View File

@@ -1,6 +1,8 @@
import json import json
import logging import logging
from fastapi import APIRouter, BackgroundTasks, HTTPException from fastapi import APIRouter, BackgroundTasks, HTTPException
from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel
from app.schemas.response_template import ResponseModel from app.schemas.response_template import ResponseModel
from app.service.generate_image.service_generate_image import GenerateImage, infer_cancel as generate_image_infer_cancel from app.service.generate_image.service_generate_image import GenerateImage, infer_cancel as generate_image_infer_cancel
@@ -26,7 +28,7 @@ def generate_image(request_item: GenerateImageModel, background_tasks: Backgroun
return ResponseModel() return ResponseModel()
@router.get("/generate_cancel/{tasks_id}>") @router.get("/generate_cancel/{tasks_id}")
def generate_image(tasks_id: str): def generate_image(tasks_id: str):
try: try:
logger.info(f"generate_cancel request item is : @@@@@@:{tasks_id}") logger.info(f"generate_cancel request item is : @@@@@@:{tasks_id}")
@@ -53,7 +55,7 @@ def generate_single_logo(request_item: GenerateSingleLogoImageModel, background_
return ResponseModel() return ResponseModel()
@router.get("/generate_single_logo_cancel/{tasks_id}>") @router.get("/generate_single_logo_cancel/{tasks_id}")
def generate_single_logo_image(tasks_id: str): def generate_single_logo_image(tasks_id: str):
try: try:
logger.info(f"generate_single_logo_cancel request item is : @@@@@@:{tasks_id}") logger.info(f"generate_single_logo_cancel request item is : @@@@@@:{tasks_id}")
@@ -80,7 +82,7 @@ def generate_product_image(request_item: GenerateProductImageModel, background_t
return ResponseModel() return ResponseModel()
@router.get("/generate_product_image_cancel_cancel/{tasks_id}>") @router.get("/generate_product_image_cancel_cancel/{tasks_id}")
def generate_product_image(tasks_id: str): def generate_product_image(tasks_id: str):
try: try:
logger.info(f"generate_product_image_cancel_cancel request item is : @@@@@@:{tasks_id}") logger.info(f"generate_product_image_cancel_cancel request item is : @@@@@@:{tasks_id}")
@@ -107,7 +109,7 @@ def generate_relight_image(request_item: GenerateProductImageModel, background_t
return ResponseModel() return ResponseModel()
@router.get("/generate_relight_image_cancel_cancel/{tasks_id}>") @router.get("/generate_relight_image_cancel_cancel/{tasks_id}")
def generate_relight_image(tasks_id: str): def generate_relight_image(tasks_id: str):
try: try:
logger.info(f"generate_relight_image_cancel_cancel request item is : @@@@@@:{tasks_id}") logger.info(f"generate_relight_image_cancel_cancel request item is : @@@@@@:{tasks_id}")