TSLiveWallpaper/TSLiveWallpaper/Business/TSAIListVC/TSAIList+Enmu.swift

//
//  TSAIList+Enmu.swift
//  TSLiveWallpaper
//
//  Created by 100Years on 2025/6/10.
//

enum TSGeneratorImageStyle:String {
    case enhance = "enhance"   //照片变高清
    case colorize = "colorize"   //黑白变颜色
    case descratch = "descratch"   //旧照片修复
    
    case enlighten = "enlighten"   //调整光线
    case recreate = "recreate"   //修复和上色老照片
    case creatVideo = "creatVideo"   //修复和上色老照片
    
    var imageMaxKb:Int{
        switch self {
        case .descratch:
            return 5*1024
        default:
            return 10*1024
        }
    }
    
    var imageMaxBitSize:Int{
        switch self {
        case .descratch:
            return kUploadImageMaxBit5Size
        default:
            return kUploadImageMaxBit10Size
        }
    }
    
    var userDefaultsKey:String{
        return self.rawValue
    }
    
    var config:TSAIListHintBaseVC.Config{
        switch self {
        case .descratch:
            return .descratchConfig
        case .colorize:
            return .colorizeConfig
        case .enhance:
            return .enhanceConfig
        case .enlighten:
            return .enlightenConfig
        case .recreate:
            return .recreateConfig
        case .creatVideo:
            return .animatedConfig
        }
    }
    
    var advance:Bool{
        return false
    }
    //服务器生成图片所用的模型
    var aiModel:String {
        return "kie"
    }
    
    
    var generatorBtnTitle:String {
        switch self {
        case .enhance:
            return "Enhance Photo".localized
        case .colorize:
            return "Colorize Photo".localized
        case .descratch:
            return "Descratch Photo".localized
        case .enlighten:
            return "Enlighten Photo".localized
        case .recreate:
            return "Recreate Photo".localized
        default:
            return "Generate".localized
        }
    }
}