From 2a83399effcb12eb7ef1c41e2352eae955af785d Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 24 Jun 2024 16:52:00 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20generate=20=E5=8F=96=E6=B6=88?= =?UTF-8?q?=E6=8E=A5=E5=8F=A3=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/api/api_generate_image.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py index 07303ad..23fbb7f 100644 --- a/app/api/api_generate_image.py +++ b/app/api/api_generate_image.py @@ -1,6 +1,8 @@ import json import logging + from fastapi import APIRouter, BackgroundTasks, HTTPException + from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel from app.schemas.response_template import ResponseModel 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() -@router.get("/generate_cancel/{tasks_id}>") +@router.get("/generate_cancel/{tasks_id}") def generate_image(tasks_id: str): try: logger.info(f"generate_cancel request item is : @@@@@@:{tasks_id}") @@ -53,7 +55,7 @@ def generate_single_logo(request_item: GenerateSingleLogoImageModel, background_ 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): try: 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() -@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): try: 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() -@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): try: logger.info(f"generate_relight_image_cancel_cancel request item is : @@@@@@:{tasks_id}")