超分新增发布rabbitmq消息

app/api/api_super_resolution.py

@@ -1,14 +1,30 @@
-from fastapi import APIRouter
+import json
+import logging
+
+from fastapi import APIRouter, BackgroundTasks
 
 from app.schemas.super_resolution import SuperResolutionModel
-from app.service.super_resolution.service import SuperResolution
+from app.service.super_resolution.service import SuperResolution, infer_cancel
 
 router = APIRouter()
+logger = logging.getLogger()
 
 
 @router.post("super_resolution")
-def super_resolution(request_item: SuperResolutionModel):
-    service = SuperResolution()
-    sr_result_url = service.sr_result(request_item.sr_image_url, request_item.sr_xn)
-    response = {"sr_result_url": sr_result_url}
-    return {"code": 200, "message": "ok", "data": response}
+def super_resolution(request_item: SuperResolutionModel, background_tasks: BackgroundTasks):
+    try:
+        service = SuperResolution(request_item)
+        background_tasks.add_task(service.sr_result)
+        code = 200
+        message = "access"
+    except Exception as e:
+        code = 000
+        message = e
+        logger.warning(e)
+    return {"code": code, "message": message}
+
+
+@router.get("sr_cancel/{tasks_id}>")
+def super_resolution(tasks_id):
+    result = infer_cancel(tasks_id)
+    return {"code": 200, "message": result['message'], "data": result['data']}