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"
    
    case portraitFusion = "PortraitFusion"  //照片融合
    case removeBg = "removeBg"      //删除图片的背景
    case removeWatermark = "RemoveWatermark"      //移除水印
    case photoExpand = "ExpandPhoto"      // AI扩图
    case magicText = "MagicText"      // 神奇文字,也就是自定义提示词图生图
    
    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,.removeWatermark,.photoExpand,.removeBg:
            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
        case .portraitFusion:
            return .portraitFusionConfig
        case .ptp:
            return .ptpConfig
        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
        case .portraitFusion:
            return "Fusion".localized
        default:
            return "Photo".localized
        }
    }
    
    var needPhotos:Int {
        if self == .portraitFusion {
            return 2
        }
        return 1
    }
    
    
    var vipFreeNumType:VipFreeNumType{
        if self == .creatVideo {
            return .generalVideo
        }else if self == .removeBg {
            return .generalRemoveBg
        }else{
            return .general
        }
    }
}

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