presentation/com.shifthackz.aisdv1.presentation.screen.txt2img/TextToImageViewModel

TextToImageViewModel

class TextToImageViewModel(val dispatchersProvider: DispatchersProvider, val getConfigurationUseCase: GetConfigurationUseCase, val getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase, val getForgeModulesUseCase: GetForgeModulesUseCase, val fetchAndGetArliAiModelsUseCase: FetchAndGetArliAiModelsUseCase, val isADetailerAvailableUseCase: IsADetailerAvailableUseCase, val textToImageUseCase: TextToImageUseCase, val saveGenerationResultUseCase: SaveGenerationResultUseCase, val saveLastResultToCacheUseCase: SaveLastResultToCacheUseCase, val interruptGenerationUseCase: InterruptGenerationUseCase, val observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase, val observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase, val observeStableDiffusionCppProcessStatusUseCase: ObserveStableDiffusionCppProcessStatusUseCase, val observeCoreMlProcessStatusUseCase: ObserveCoreMlProcessStatusUseCase, val observeBonsaiProcessStatusUseCase: ObserveBonsaiProcessStatusUseCase, val preferenceManager: PreferenceManager, val backgroundTaskManager: BackgroundTaskManager, val backgroundWorkObserver: BackgroundWorkObserver, val wakeLockInterActor: WakeLockInterActor, val platformServices: GenerationPlatformServices, val buildInfoProvider: BuildInfoProvider, val generationFormUpdateEvent: GenerationFormUpdateEvent, val dimensionValidator: DimensionValidator, val localGenerationBenchmarkGateProvider: () -> LocalGenerationBenchmarkGate, val imageSaver: ImageSaver, val imageSharer: ImageSharer, val router: TextToImageRouter, val onError: (Throwable) -> Unit = {}) : BaseMviViewModel<TextToImageState, TextToImageIntent, EmptyEffect> (source)

View-model for the txt2img screen.

It loads provider configuration, mirrors the shared generation form state, gates local providers through benchmark recommendations, and delegates actual generation work to the action handler.

Constructors

TextToImageViewModel

constructor(dispatchersProvider: DispatchersProvider, getConfigurationUseCase: GetConfigurationUseCase, getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase, getForgeModulesUseCase: GetForgeModulesUseCase, fetchAndGetArliAiModelsUseCase: FetchAndGetArliAiModelsUseCase, isADetailerAvailableUseCase: IsADetailerAvailableUseCase, textToImageUseCase: TextToImageUseCase, saveGenerationResultUseCase: SaveGenerationResultUseCase, saveLastResultToCacheUseCase: SaveLastResultToCacheUseCase, interruptGenerationUseCase: InterruptGenerationUseCase, observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase, observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase, observeStableDiffusionCppProcessStatusUseCase: ObserveStableDiffusionCppProcessStatusUseCase, observeCoreMlProcessStatusUseCase: ObserveCoreMlProcessStatusUseCase, observeBonsaiProcessStatusUseCase: ObserveBonsaiProcessStatusUseCase, preferenceManager: PreferenceManager, backgroundTaskManager: BackgroundTaskManager, backgroundWorkObserver: BackgroundWorkObserver, wakeLockInterActor: WakeLockInterActor, platformServices: GenerationPlatformServices, buildInfoProvider: BuildInfoProvider, generationFormUpdateEvent: GenerationFormUpdateEvent, dimensionValidator: DimensionValidator, localGenerationBenchmarkGateProvider: () -> LocalGenerationBenchmarkGate, imageSaver: ImageSaver, imageSharer: ImageSharer, router: TextToImageRouter, onError: (Throwable) -> Unit = {})

Properties

private val _effect: Channel<E>

private val _state: MutableStateFlow<S>

private val actionHandler: TextToImageActionHandler

aDetailerAvailabilityKey

private var aDetailerAvailabilityKey: StableDiffusionSamplersKey?

aDetailerAvailable

private var aDetailerAvailable: Boolean?

private var arliAiModels: List<String>?

arliAiModelsKey

private var arliAiModelsKey: String?

backgroundTaskManager

private val backgroundTaskManager: BackgroundTaskManager

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

buildInfoProvider

private val buildInfoProvider: BuildInfoProvider

protected val currentState: TextToImageState

dimensionValidator

private val dimensionValidator: DimensionValidator

dispatchersProvider

private val dispatchersProvider: DispatchersProvider

open override val effect: Flow<EmptyEffect>

effectDispatcher

private val effectDispatcher: CoroutineDispatcher

fetchAndGetArliAiModelsUseCase

private val fetchAndGetArliAiModelsUseCase: FetchAndGetArliAiModelsUseCase

private var forgeModules: List<ForgeModule>?

forgeModulesKey

private var forgeModulesKey: StableDiffusionSamplersKey?

generationFormUpdateEvent

private val generationFormUpdateEvent: GenerationFormUpdateEvent

getConfigurationUseCase

private val getConfigurationUseCase: GetConfigurationUseCase

getForgeModulesUseCase

private val getForgeModulesUseCase: GetForgeModulesUseCase

getStableDiffusionSamplersUseCase

private val getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase

private val imageSaver: ImageSaver

private val imageSharer: ImageSharer

intentProcessor

private val intentProcessor: TextToImageIntentProcessor

interruptGenerationUseCase

private val interruptGenerationUseCase: InterruptGenerationUseCase

isADetailerAvailableUseCase

private val isADetailerAvailableUseCase: IsADetailerAvailableUseCase

localGenerationBenchmarkGateProvider

private val localGenerationBenchmarkGateProvider: () -> LocalGenerationBenchmarkGate

observeBonsaiProcessStatusUseCase

private val observeBonsaiProcessStatusUseCase: ObserveBonsaiProcessStatusUseCase

observeCoreMlProcessStatusUseCase

private val observeCoreMlProcessStatusUseCase: ObserveCoreMlProcessStatusUseCase

observeHordeProcessStatusUseCase

private val observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase

observeLocalDiffusionProcessStatusUseCase

private val observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase

observeStableDiffusionCppProcessStatusUseCase

private val observeStableDiffusionCppProcessStatusUseCase: ObserveStableDiffusionCppProcessStatusUseCase

private val onError: (Throwable) -> Unit

platformServices

private val platformServices: GenerationPlatformServices

preferenceManager

private val preferenceManager: PreferenceManager

private val router: TextToImageRouter

saveGenerationResultUseCase

private val saveGenerationResultUseCase: SaveGenerationResultUseCase

saveLastResultToCacheUseCase

private val saveLastResultToCacheUseCase: SaveLastResultToCacheUseCase

stableDiffusionSamplers

private var stableDiffusionSamplers: List<String>?

stableDiffusionSamplersKey

private var stableDiffusionSamplersKey: StableDiffusionSamplersKey?

open override val state: StateFlow<TextToImageState>

textToImageUseCase

private val textToImageUseCase: TextToImageUseCase

wakeLockInterActor

private val wakeLockInterActor: WakeLockInterActor

Functions

expect open fun addCloseable(closeable: AutoCloseable)

expect fun addCloseable(key: String, closeable: AutoCloseable)

applyGenerationResult

private fun applyGenerationResult(ai: AiGenerationResult)

internal expect fun clear()

consumePendingGenerationFormUpdate

private fun consumePendingGenerationFormUpdate()

protected fun emitEffect(effect: EmptyEffect)

protected fun emitState(state: TextToImageState)

expect fun <T : AutoCloseable> getCloseable(key: String): T?

protected fun launch(dispatcher: CoroutineDispatcher, start: CoroutineStart, block: suspend CoroutineScope.() -> Unit): Job

loadADetailerAvailability

private fun loadADetailerAvailability(force: Boolean = false)

loadArliAiModels

private fun loadArliAiModels()

loadConfiguration

private fun loadConfiguration()

loadForgeModules

private fun loadForgeModules()

private fun loadSamplers()

observeGenerationFormUpdate

private fun observeGenerationFormUpdate()

observeGenerationProgress

private fun observeGenerationProgress()

observeSettings

private fun observeSettings()

protected expect open fun onCleared()

open override fun processIntent(intent: TextToImageIntent)

refreshArliAiModelsIfNeeded

private fun refreshArliAiModelsIfNeeded(settings: Settings)

refreshStableDiffusionMetadataIfNeeded

private fun refreshStableDiffusionMetadataIfNeeded(settings: Settings)

updateLocalGenerationStatus

private fun updateLocalGenerationStatus(status: LocalDiffusionStatus)

protected fun updateState(mutation: (TextToImageState) -> TextToImageState)