feat 更新响应模板

fix

app/api/api_super_resolution.py

@@ -1,8 +1,9 @@
 import json
 import logging
 
-from fastapi import APIRouter, BackgroundTasks
+from fastapi import APIRouter, BackgroundTasks, HTTPException
 
+from app.schemas.response_template import ResponseModel
 from app.schemas.super_resolution import SuperResolutionModel
 from app.service.super_resolution.service import SuperResolution, infer_cancel
 
@@ -13,18 +14,22 @@ logger = logging.getLogger()
 @router.post("/super_resolution")
 def super_resolution(request_item: SuperResolutionModel, background_tasks: BackgroundTasks):
     try:
+        logger.info(f"super_resolution request item is : @@@@@@:{request_item}")
         service = SuperResolution(request_item)
         background_tasks.add_task(service.sr_result)
-        code = 200
-        message = "OK!"
     except Exception as e:
-        code = 400
-        message = e
-        logger.warning(e)
-    return {"code": code, "msg": message}
+        logger.warning(f"super_resolution Run Exception @@@@@@:{e}")
+        raise HTTPException(status_code=404, detail=str(e))
+    return ResponseModel()
 
 
 @router.get("/sr_cancel/{tasks_id}>")
 def super_resolution(tasks_id):
-    result = infer_cancel(tasks_id)
-    return {"code": 200, "msg": result['message'], "data": result['data']}
+    try:
+        logger.info(f"sr_cancel request item is : @@@@@@:{tasks_id}")
+        data = infer_cancel(tasks_id)
+        logger.info(f"sr_cancel response @@@@@@:{json.dumps(data, indent=4)}")
+    except Exception as e:
+        logger.warning(f"sr_cancel Run Exception @@@@@@:{e}")
+        raise HTTPException(status_code=404, detail=str(e))
+    return ResponseModel(data=data['data'])