From 49807a3bf3aa1ac9fb57d77fe6abf65d99477dc3 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 13 Jun 2024 13:35:49 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20=E7=BF=BB=E8=AF=91=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3=E4=BF=AE=E6=94=B9template?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/api/api_attribute_retrieve.py | 4 ++-- app/api/api_chat_robot.py | 2 +- app/api/api_design.py | 2 +- app/api/api_design_pre_processing.py | 2 +- app/api/api_generate_image.py | 12 ++++++------ app/api/api_prompt_generation.py | 2 +- app/api/api_super_resolution.py | 2 +- 7 files changed, 13 insertions(+), 13 deletions(-) diff --git a/app/api/api_attribute_retrieve.py b/app/api/api_attribute_retrieve.py index fdecfa8..5b18ec5 100644 --- a/app/api/api_attribute_retrieve.py +++ b/app/api/api_attribute_retrieve.py @@ -18,7 +18,7 @@ def attribute_recognition(request_item: list[AttributeRecognitionModel]): service = AttributeRecognition(const=const, request_data=request_item) data = service.get_result() code = 200 - message = "access" + message = "OK!" logger.info(f"attribute_recognition response @@@@@@:{json.dumps(data, indent=4)}") except Exception as e: code = 400 @@ -35,7 +35,7 @@ def category_recognition(request_item: list[CategoryRecognitionModel]): service = CategoryRecognition(request_data=request_item) data = service.get_result() code = 200 - message = "access" + message = "OK!" logger.info(f"category_recognition response @@@@@@:{json.dumps(data, indent=4)}") except Exception as e: code = 400 diff --git a/app/api/api_chat_robot.py b/app/api/api_chat_robot.py index c394046..d919902 100644 --- a/app/api/api_chat_robot.py +++ b/app/api/api_chat_robot.py @@ -14,7 +14,7 @@ def chat_robot(request_data: ChatRobotModel): try: logger.info(f"chat_robot request item is : @@@@@@:{request_data}") code = 200 - message = "access" + message = "OK!" start_time = time.time() data = chat(post_data=request_data) logger.info(f"chat_robot Run time is @@@@@@:{time.time() - start_time}") diff --git a/app/api/api_design.py b/app/api/api_design.py index 0c48c81..79cf7e1 100644 --- a/app/api/api_design.py +++ b/app/api/api_design.py @@ -15,7 +15,7 @@ def design(request_data: DesignModel): try: logger.info(f"design request item is : @@@@@@:{request_data}") code = 200 - message = "access" + message = "OK!" start_time = time.time() data = generate(request_data=request_data) logger.info(f"design Run time is @@@@@@:{time.time() - start_time}") diff --git a/app/api/api_design_pre_processing.py b/app/api/api_design_pre_processing.py index 0c0089d..91bba6e 100644 --- a/app/api/api_design_pre_processing.py +++ b/app/api/api_design_pre_processing.py @@ -15,7 +15,7 @@ def design_pre_processing(request_data: DesignPreProcessingModel): try: logger.info(f"design_pre_processing request item is : @@@@@@:{request_data}") code = 200 - message = "access" + message = "OK!" start_time = time.time() server = DesignPreprocessing() data = server.pipeline(image_list=request_data.sketches) diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py index a74eb1b..9c979f1 100644 --- a/app/api/api_generate_image.py +++ b/app/api/api_generate_image.py @@ -18,7 +18,7 @@ def generate_image(request_item: GenerateImageModel, background_tasks: Backgroun service = GenerateImage(request_item) background_tasks.add_task(service.get_result) code = 200 - message = "access" + message = "OK!" except Exception as e: code = 400 message = e @@ -29,7 +29,7 @@ def generate_image(request_item: GenerateImageModel, background_tasks: Backgroun @router.get("/generate_cancel/{tasks_id}>") def generate_image(tasks_id): result = generate_image_infer_cancel(tasks_id) - return {"code": 200, "message": result['message'], "data": result['data']} + return {"code": 200, "message": "OK!", "data": result['data']} '''single logo''' @@ -42,7 +42,7 @@ def generate_single_logo(request_item: GenerateSingleLogoImageModel, background_ service = GenerateSingleLogoImage(request_item) background_tasks.add_task(service.get_result) code = 200 - message = "access" + message = "OK!" except Exception as e: code = 400 message = e @@ -53,7 +53,7 @@ def generate_single_logo(request_item: GenerateSingleLogoImageModel, background_ @router.get("/generate_single_logo_cancel/{tasks_id}>") def generate_single_logo_image(tasks_id): result = generate_single_logo_cancel(tasks_id) - return {"code": 200, "message": result['message'], "data": result['data']} + return {"code": 200, "message": "OK!", "data": result['data']} '''product image''' @@ -66,7 +66,7 @@ def generate_product_image(request_item: GenerateProductImageModel, background_t service = GenerateProductImage(request_item) background_tasks.add_task(service.get_result) code = 200 - message = "access" + message = "OK!" except Exception as e: code = 400 message = e @@ -77,4 +77,4 @@ def generate_product_image(request_item: GenerateProductImageModel, background_t @router.get("/generate_product_image_cancel_cancel/{tasks_id}>") def generate_single_logo_image(tasks_id): result = generate_product_image_cancel(tasks_id) - return {"code": 200, "message": result['message'], "data": result['data']} + return {"code": 200, "message": "OK!", "data": result['data']} diff --git a/app/api/api_prompt_generation.py b/app/api/api_prompt_generation.py index 5e71eec..8b930a1 100644 --- a/app/api/api_prompt_generation.py +++ b/app/api/api_prompt_generation.py @@ -15,7 +15,7 @@ def prompt_generation(request_data: PromptGenerationImageModel): try: logger.info(f"prompt_translate to English request data : @@@@@@:{request_data}") code = 200 - message = "access" + message = "OK!" start_time = time.time() data = translate_to_en(request_data.text) logger.info(f"prompt_generation Run time is @@@@@@:{time.time() - start_time}") diff --git a/app/api/api_super_resolution.py b/app/api/api_super_resolution.py index 63f4498..0d14ad7 100644 --- a/app/api/api_super_resolution.py +++ b/app/api/api_super_resolution.py @@ -16,7 +16,7 @@ def super_resolution(request_item: SuperResolutionModel, background_tasks: Backg service = SuperResolution(request_item) background_tasks.add_task(service.sr_result) code = 200 - message = "access" + message = "OK!" except Exception as e: code = 400 message = e