using Ollama;

namespace AuthorBuddy.Web.Models;

public class ModelStatus
{
    public string Status { get; private set; } = "Unknown";
    public double Progress { get; private set; }

    public void SetModelStatus(PullModelResponse modelResponse)
    {
        if (modelResponse.Total > 0 && modelResponse.Completed.HasValue && modelResponse.Completed > 0)
        {
            Progress = (double)modelResponse.Completed.Value / (double)modelResponse.Total * 100;
            Status = modelResponse.Status?.ToString() ?? "Unknown";
        }
    }
}