TSAIListPhotoGeneratorModel.swift 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. //
  2. // TSAIListPhotoGeneratorModel.swift
  3. // TSLiveWallpaper
  4. //
  5. // Created by 100Years on 2025/6/19.
  6. //
  7. class TSAIListPhotoGeneratorModel {
  8. var upLoadImage:UIImage
  9. var generatorStyle:TSGeneratorImageStyle
  10. var expandEdge:UIEdgeInsets
  11. var expandViewSizes:(CGSize,CGSize)?
  12. var additionalPrompt:String//追加的提示词
  13. //预测宝宝
  14. var upLoadImages:[UIImage]?
  15. init(upLoadImage: UIImage,
  16. generatorStyle: TSGeneratorImageStyle,
  17. expandEdge:UIEdgeInsets = UIEdgeInsets(top: 0, left: 0, bottom: 0, right: 0),
  18. expandViewSizes:(CGSize,CGSize)? = nil,
  19. additionalPrompt:String = "",
  20. upLoadImages:[UIImage]? = nil
  21. ) {
  22. self.upLoadImage = upLoadImage
  23. self.generatorStyle = generatorStyle
  24. self.expandEdge = expandEdge
  25. self.expandViewSizes = expandViewSizes
  26. self.additionalPrompt = additionalPrompt
  27. self.upLoadImages = upLoadImages
  28. }
  29. var prompt:String = ""
  30. var inputText:String = ""
  31. var upLoadImageUrl:String?
  32. var model:String = "" //决定生图的模型
  33. }