feat : 代码梳理 移除所有敏感密钥 通过环境变量方式配置

app/api/api_pose_transform.py

@@ -4,55 +4,55 @@ import logging
 import requests
 from fastapi import APIRouter, BackgroundTasks, HTTPException
 
-from app.core.config import COMFYUI_SERVER_ADDRESS
+from app.core.config import settings
 from app.schemas.comfyui_i2v import ComfyuiI2VModel, ComfyuiFLF2VModel
 from app.schemas.pose_transform import PoseTransformModel
 from app.schemas.response_template import ResponseModel
 from app.service.comfyui_I2V.flf2v_server import ComfyUIServerFLF2V
 from app.service.comfyui_I2V.i2v_server import ComfyUIServerI2V
 from app.service.comfyui_I2V.pose2v_server import ComfyUIServerPose2V
-from app.service.generate_image.service_pose_transform import PoseTransformService, infer_cancel as pose_transform_infer_cancel
 
 router = APIRouter()
 logger = logging.getLogger()
 
+"""停用"""
 
-@router.post("/pose_transform")
-def pose_transform(request_item: PoseTransformModel, background_tasks: BackgroundTasks):
-    """
-        创建一个具有以下参数的请求体:
-        - **tasks_id**: 任务id 用于取消生成任务和获取生成结果
-        - **image_url**: 被生成图片的S3或minio url地址
-        - **pose_id**: 1
+# @router.post("/pose_transform")
+# def pose_transform(request_item: PoseTransformModel, background_tasks: BackgroundTasks):
+#     """
+#         创建一个具有以下参数的请求体:
+#         - **tasks_id**: 任务id 用于取消生成任务和获取生成结果
+#         - **image_url**: 被生成图片的S3或minio url地址
+#         - **pose_id**: 1
+#
+#
+#         示例参数：
+#         {
+#           "tasks_id": "123-89",
+#           "image_url": "aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png",
+#           "pose_id": "1"
+#         }
+#     """
+#     try:
+#         logger.info(f"pose_transform request item is : @@@@@@:{json.dumps(request_item.dict(),indent=4)}")
+#         service = PoseTransformService(request_item)
+#         background_tasks.add_task(service.get_result)
+#     except Exception as e:
+#         logger.warning(f"pose_transform Run Exception @@@@@@:{e}")
+#         raise HTTPException(status_code=404, detail=str(e))
+#     return ResponseModel()
 
 
-        示例参数：
-        {
-          "tasks_id": "123-89",
-          "image_url": "aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png",
-          "pose_id": "1"
-        }
-    """
-    try:
-        logger.info(f"pose_transform request item is : @@@@@@:{json.dumps(request_item.dict())}")
-        service = PoseTransformService(request_item)
-        background_tasks.add_task(service.get_result)
-    except Exception as e:
-        logger.warning(f"pose_transform Run Exception @@@@@@:{e}")
-        raise HTTPException(status_code=404, detail=str(e))
-    return ResponseModel()
-
-
-@router.get("/pose_transform_cancel/{tasks_id}")
-def pose_transform_cancel(tasks_id: str):
-    try:
-        logger.info(f"pose_transform_cancel request item is : @@@@@@:{tasks_id}")
-        data = pose_transform_infer_cancel(tasks_id)
-        logger.info(f"pose_transform_cancel response @@@@@@:{data}")
-    except Exception as e:
-        logger.warning(f"pose_transform_cancel Run Exception @@@@@@:{e}")
-        raise HTTPException(status_code=404, detail=str(e))
-    return ResponseModel(data=data['data'])
+# @router.get("/pose_transform_cancel/{tasks_id}")
+# def pose_transform_cancel(tasks_id: str):
+#     try:
+#         logger.info(f"pose_transform_cancel request item is : @@@@@@:{tasks_id}")
+#         data = pose_transform_infer_cancel(tasks_id)
+#         logger.info(f"pose_transform_cancel response @@@@@@:{data}")
+#     except Exception as e:
+#         logger.warning(f"pose_transform_cancel Run Exception @@@@@@:{e}")
+#         raise HTTPException(status_code=404, detail=str(e))
+#     return ResponseModel(data=data['data'])
 
 
 """
@@ -77,7 +77,7 @@ def comfyui_image_pose_2_video(request_item: PoseTransformModel, background_task
         }
     """
     try:
-        logger.info(f"image_pose_2_video request item is : @@@@@@:{json.dumps(request_item.dict())}")
+        logger.info(f"image_pose_2_video request item is : @@@@@@:{json.dumps(request_item.dict(),indent=4)}")
         service = ComfyUIServerPose2V(request_item)
         background_tasks.add_task(service.get_result)
     except Exception as e:
@@ -107,7 +107,7 @@ def comfyui_image_2_video(request_item: ComfyuiI2VModel, background_tasks: Backg
         }
     """
     try:
-        logger.info(f"image_2_video request item is : @@@@@@:{json.dumps(request_item.dict())}")
+        logger.info(f"image_2_video request item is : @@@@@@:{json.dumps(request_item.dict(),indent=4)}")
         service = ComfyUIServerI2V(request_item)
         background_tasks.add_task(service.get_result)
     except Exception as e:
@@ -139,7 +139,7 @@ def comfyui_flf_2_video(request_item: ComfyuiFLF2VModel, background_tasks: Backg
         }
     """
     try:
-        logger.info(f"flf_2_video request item is : @@@@@@:{json.dumps(request_item.dict())}")
+        logger.info(f"flf_2_video request item is : @@@@@@:{json.dumps(request_item.dict(),indent=4)}")
         service = ComfyUIServerFLF2V(request_item)
         background_tasks.add_task(service.get_result)
     except Exception as e:
@@ -153,7 +153,7 @@ def comfyui_i_2_video_cancel(tasks_id: str):
     try:
         logger.info(f"comfyui_i_2_video_cancel request item is : @@@@@@:{tasks_id}")
         response = requests.post(
-            f"http://{COMFYUI_SERVER_ADDRESS}/interrupt",
+            f"http://{settings.COMFYUI_SERVER_ADDRESS}/interrupt",
             json={"prompt_id": tasks_id}
         )
         data = {}