flowframes/Code/Data/ModelCollection.cs

using Flowframes.IO;
using Newtonsoft.Json;
using System;
using System.Collections.Generic;
using System.IO;

namespace Flowframes.Data
{
    class ModelCollection
    {
        public AI ai;
        public List<ModelInfo> models;

        public class ModelInfo
        {
            public AI ai;
            public string name;
            public string desc;
            public string dir;
            public bool isDefault;

            public ModelInfo(AI ai, string name, string desc, string dir, bool isDefault)
            {
                this.ai = ai;
                this.name = name;
                this.desc = desc;
                this.dir = dir;
                this.isDefault = isDefault;
            }

            public string GetUiString()
            {
                return $"{name} - {desc}{(isDefault ? " (Recommended)" : "")}";
            }

            public override string ToString()
            {
                return $"{name} - {desc} ({dir}){(isDefault ? " (Recommended)" : "")}";
            }
        }

        public ModelCollection(AI ai, string jsonContentOrPath)
        {
            if (IOUtils.IsPathValid(jsonContentOrPath) && File.Exists(jsonContentOrPath))
                jsonContentOrPath = File.ReadAllText(jsonContentOrPath);

            models = new List<ModelInfo>();
            dynamic data = JsonConvert.DeserializeObject(jsonContentOrPath);

            foreach (var item in data)
            {
                bool def = false;
                bool.TryParse((string)item.isDefault, out def);
                models.Add(new ModelInfo(ai, (string)item.name, (string)item.desc, (string)item.dir, def));
            }

            Logger.Log($"Loaded {models.Count} {ai.aiName} models from JSON.", true);
        }
    }
}