重构基本完成

app/api/business/project_api.py

@@ -13,6 +13,7 @@ from app.common import reponse_code as rc
 
 from typing import List
 from fastapi import APIRouter, Depends, Request, UploadFile, File, Form
+from fastapi.responses import StreamingResponse
 from sqlalchemy.orm import Session
 
 """项目管理API"""
@@ -150,3 +151,12 @@ def get_img_leafer(image_id: int, session: Session = Depends(get_db)):
     img_leafer_out = ps.get_img_leafer(image_id, session)
     return rc.response_success(data=img_leafer_out)
 
+
+@project.get("/run_train/{project_id}")
+def run_train(project_id: int, session: Session = Depends(get_db)):
+    project_info = pic.get_project_by_id(project_id, session)
+    if project_info is None:
+        return rc.response_error("项目查询错误")
+    if project_info.project_status == '1':
+        return rc.response_error("项目当前存在训练进程，请稍后再试")
+    return StreamingResponse(ps.run_train_yolo(project_info, session), media_type="text/plain")