from django.shortcuts import get_object_or_404, render

from control_plane.model_studio.models import TrainingProject


def model_studio(request):
    return render(request, "control_plane/model_studio.html", {"projects": TrainingProject.objects.select_related("current_champion", "baseline_evaluation").order_by("-updated_at")})


def model_studio_project(request, project_id):
    training_project = get_object_or_404(TrainingProject.objects.select_related("current_champion", "baseline_evaluation"), id=project_id)
    return render(request, "control_plane/model_studio_project.html", {"training_project": training_project, "program": training_project.overnight_programs.order_by("-created_at").first(), "experiments": training_project.experiments.order_by("-created_at")[:25], "clusters": training_project.failure_clusters.order_by("-priority")[:25]})