完善推理模块的主体功能

app/api/common/view_img.py

@@ -4,7 +4,7 @@ from starlette.responses import FileResponse
 from sqlalchemy.orm import Session
 
 from app.model.crud.project_image_crud import get_img_url
-from app.model.crud.project_detect_crud import get_detect_img_url
+from app.model.crud.project_detect_crud import get_detect_img_url, get_log_img_url
 from app.config.config_reader import images_url
 from app.db.db_session import get_db
 
@@ -75,4 +75,19 @@ def view_detect_thumb(image_id: int, session: Session = Depends(get_db)):
     return FileResponse(image_path, media_type='image/jpeg')
 
 
+@view.get("/view_log_img/{image_id}")
+def view_log_img(image_id: int, session: Session = Depends(get_db)):
+    """
+    查看推理完成后图片
+    :param session:
+    :param image_id: 图片id
+    :return:
+    """
+    url = get_log_img_url(image_id, session)
+    image_path = os.path.join(url)
+    # 检查文件是否存在以及是否是文件
+    if not os.path.isfile(image_path):
+        raise HTTPException(status_code=404, detail="Image not found")
+    return FileResponse(image_path, media_type='image/jpeg')
+