TSLiveWallpaper/TSLiveWallpaper/Business/TSAIListVC/TSAIPhotoGeneratorVC/Model/TSAIListPhotoGeneratorModel.swift

//
//  TSAIListPhotoGeneratorModel.swift
//  TSLiveWallpaper
//
//  Created by 100Years on 2025/6/19.
//
import ObjectMapper
class TSAIListPhotoGeneratorModel {
    var upLoadImage:UIImage
    var generatorStyle:TSGeneratorImageStyle
    var expandEdge:UIEdgeInsets
    var expandViewSizes:(CGSize,CGSize)?
    var additionalPrompt:String//追加的提示词
    
    //预测宝宝
    var upLoadImages:[UIImage]?
    
    init(upLoadImage: UIImage,
         generatorStyle: TSGeneratorImageStyle,
         expandEdge:UIEdgeInsets = UIEdgeInsets(top: 0, left: 0, bottom: 0, right: 0),
         expandViewSizes:(CGSize,CGSize)? = nil,
         additionalPrompt:String = "",
         upLoadImages:[UIImage]? = nil
    ) {
        self.upLoadImage = upLoadImage
        self.generatorStyle = generatorStyle
        self.expandEdge = expandEdge
        self.expandViewSizes = expandViewSizes
        self.additionalPrompt = additionalPrompt
        self.upLoadImages = upLoadImages
    }
    
    var prompt:String = ""
    var inputText:String = ""
    var upLoadImageUrl:String?
    var model:String = ""   //决定生图的模型
}

class TSGenerateModel: TSBaseModel {
    var imageName:String = ""
    var imageText:String = ""
    var prompt:String = ""
    var isVip:Bool = false
    var specialStyle:Int = 0 //0 普通类型,1 热门类型,2 新类型,3 max,
    var input:Bool = false   //是否输入框
    var model:String = ""       //走新的通道,根据内容确定
    var unionType:Int = 0 //聚合类型,相同的为一类
    
    override func mapping(map: ObjectMapper.Map) {
        imageName               <- map["imageName"]
        imageText               <- map["imageText"]
        prompt                  <- map["prompt"]
        isVip                   <- map["isVip"]
        specialStyle            <- map["specialStyle"]
        input                   <- map["input"]
        model                   <- map["model"]
        unionType               <- map["unionType"]
    }
    
}