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"   //修复和上色老照片
    case remove = "remove"      //删除图片的东西
    case ptp = "ptp"
    
    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{
        switch self {
        case .remove:
            return ""
        default:
            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
        default:
            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 "Adjust Light".localized
        case .recreate:
            return "Recreate Photo".localized
        case .creatVideo:
            return "Animated Photo".localized
        case .remove:
            return "Remove".localized
        default:
            return "Photo".localized
        }
    }
}

extension TSGeneratorImageStyle {
    
    var generateModel:TSGenerateModel? {
        let vm = TSAIListDataVM.shared
        switch self {
        case .enhance:
            return TSGenerateModel(json: vm.ptp_PhotoEnhance)
        case .colorize:
            return TSGenerateModel(json: vm.ptp_Colorize)
        case .descratch:
            return TSGenerateModel(json: vm.ptp_Descratch)
        case .enlighten:
            return TSGenerateModel(json: vm.ptp_Enlighten)
        case .recreate:
            return TSGenerateModel(json: vm.ptp_Recreate)
        case .creatVideo:
            return TSGenerateModel(json: vm.video_Creat)
        default:
            return nil
        }
    }
    
    
}