Update project_controller.py

project_controller.py

@@ -5,16 +5,15 @@ router = APIRouter()
 service_config = ServiceConfig()
 service_config.load()
 
-@router.get("/list")
-def list_projects():
-    return service_config.projects
-
 @router.post("/add")
 async def add_project(request: Request):
     data = await request.json()
     project_name = data.get("project_name")
+    if not project_name:
+        raise HTTPException(status_code=400, detail="project_name cannot be empty")
     if project_name in service_config.projects:
         raise HTTPException(status_code=400, detail="Project already exists")
+
     service_config.projects[project_name] = {
         "enabled": False,
         "versions": []
@@ -24,38 +23,22 @@ async def add_project(request: Request):
         json.dump(service_config, f, indent=2)
     return {"message": f"Project {project_name} added"}
 
-@router.post("/enable")
-async def enable_project(request: Request):
+@router.post("/publish")
+async def publish_project(request: Request):
     data = await request.json()
     project_name = data.get("project_name")
-    if project_name not in service_config.projects:
+    project = service_config.projects.get(project_name)
+    if not project:
         raise HTTPException(status_code=404, detail="Project not found")
-    service_config.projects[project_name]["enabled"] = True
-    with open(service_config.config_path, "w", encoding="utf-8") as f:
-        import json
-        json.dump(service_config, f, indent=2)
-    return {"message": f"Project {project_name} enabled"}
 
-@router.post("/disable")
-async def disable_project(request: Request):
-    data = await request.json()
-    project_name = data.get("project_name")
-    if project_name not in service_config.projects:
-        raise HTTPException(status_code=404, detail="Project not found")
-    service_config.projects[project_name]["enabled"] = False
-    with open(service_config.config_path, "w", encoding="utf-8") as f:
-        import json
-        json.dump(service_config, f, indent=2)
-    return {"message": f"Project {project_name} disabled"}
+    latest_version = project["versions"][-1] if project["versions"] else None
+    if not latest_version:
+        raise HTTPException(status_code=400, detail="No version exists to publish")
 
-@router.delete("/delete")
-async def delete_project(request: Request):
-    data = await request.json()
-    project_name = data.get("project_name")
-    if project_name not in service_config.projects:
-        raise HTTPException(status_code=404, detail="Project not found")
-    del service_config.projects[project_name]
-    with open(service_config.config_path, "w", encoding="utf-8") as f:
-        import json
-        json.dump(service_config, f, indent=2)
-    return {"message": f"Project {project_name} deleted"}
+    llm = latest_version.get("llm", {})
+    if not llm.get("repo_id"):
+        raise HTTPException(status_code=400, detail="repo_id is required")
+    if llm.get("use_fine_tune") and not llm.get("fine_tune_zip"):
+        raise HTTPException(status_code=400, detail="fine_tune_zip is required when use_fine_tune is true")
+
+    return {"message": f"Project {project_name} passed publish checks"}