完善推理模块的主体功能

2025-03-04 17:22:50 +08:00
parent fa6c344e84
commit 5f4ed473d7
4 changed files with 25 additions and 3 deletions
--- a/app/model/crud/project_detect_crud.py
+++ b/app/model/crud/project_detect_crud.py
@ -171,3 +171,10 @@ def get_log_imgs(log_id: int, session: Session):
    result = [ProjectDetectLogImgOut.from_orm(img) for img in query.all()]
    return result

+
+def get_log_img_url(img_id: int, session: Session):
+    result = session.query(ProjectDetectLogImg).filter_by(id=img_id).first()
+    if result is None:
+        return None
+    else:
+        return result.image_url