(A1111) Lora & HyperNet & Text Inversion implementation (#130)

* Lora & HyperNet implementation | Patch 1

* Text inversion implementation | Patch 1

* Text inversion implementation | Patch 2

* Design Patch

* Finalization

* Translations

* Fix build

* Fix states ui

* Fix local storage dao/datasource cache issue

* Update

Author: ShiftHackZ

core/ui/src/main/java/com/shifthackz/aisdv1/core/extensions/ShimmerExtensions.kt

@@ -21,16 +21,17 @@ import androidx.compose.ui.unit.IntSize
 
 fun Modifier.shimmer() = composed {
     var size by remember { mutableStateOf(IntSize.Zero) }
-    val transition = rememberInfiniteTransition()
+    val transition = rememberInfiniteTransition(label = "shimmer")
     val startOffsetX by transition.animateFloat(
         initialValue = -2 * size.width.toFloat(),
         targetValue = 2 * size.width.toFloat(),
-        animationSpec = infiniteRepeatable(animation = tween(durationMillis = 1000))
+        animationSpec = infiniteRepeatable(animation = tween(durationMillis = 1000)),
+        label = "shimmer"
     )
-    val colorBg = MaterialTheme.colorScheme.secondaryContainer
-        .copy(alpha = 0.5f)
-        .compositeOver(Color(0xFFCACACA))
-    val colorEffect = MaterialTheme.colorScheme.secondary
+    val colorBg = MaterialTheme.colorScheme.surfaceVariant
+        .copy(alpha = 0.75f)
+        .compositeOver(MaterialTheme.colorScheme.background)
+    val colorEffect = MaterialTheme.colorScheme.surfaceTint
     background(
         brush = Brush.linearGradient(
             colors = listOf(

core/ui/src/main/java/com/shifthackz/aisdv1/core/viewmodel/MviViewModel.kt

@@ -5,6 +5,7 @@ package com.shifthackz.aisdv1.core.viewmodel
 import android.util.Log
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
+import com.shifthackz.aisdv1.core.common.log.debugLog
 import com.shifthackz.aisdv1.core.ui.BuildConfig
 import com.shifthackz.aisdv1.core.ui.MviEffect
 import com.shifthackz.aisdv1.core.ui.MviState
@@ -16,6 +17,7 @@ import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.flow.receiveAsFlow
+import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
 
 abstract class MviViewModel<S : MviState, E : MviEffect> : ViewModel() {
@@ -35,13 +37,18 @@ abstract class MviViewModel<S : MviState, E : MviEffect> : ViewModel() {
 
     abstract val emptyState: S
 
-    open fun setState(state: S) {
+    fun setState(state: S) {
         if (BuildConfig.DEBUG) {
             Log.d(this::class.simpleName, "NEW STATE : $state")
         }
         mutableState.tryEmit(state)
     }
 
+    open fun updateState(mutation: (S) -> S) = mutableState.update {
+        if (BuildConfig.DEBUG) debugLog("NEW STATE : $state")
+        mutation(it)
+    }
+
     fun emitEffect(effect: E) {
         viewModelScope.launch(Dispatchers.Main.immediate + exceptionHandler) {
             effectChannel.send(effect)

core/validation/src/main/java/com/shifthackz/aisdv1/core/validation/dimension/DimensionValidator.kt

@@ -7,8 +7,8 @@ fun interface DimensionValidator {
     operator fun invoke(input: String?): ValidationResult<Error>
 
     sealed interface Error {
-        object Empty : Error
-        object Unexpected : Error
+        data object Empty : Error
+        data object Unexpected : Error
         data class LessThanMinimum(val min: Int) : Error
         data class BiggerThanMaximum(val max: Int) : Error
     }

core/validation/src/main/java/com/shifthackz/aisdv1/core/validation/horde/CommonStringValidator.kt

@@ -7,6 +7,6 @@ interface CommonStringValidator {
     operator fun invoke(input: String?) : ValidationResult<Error>
 
     sealed interface Error {
-        object Empty : Error
+        data object Empty : Error
     }
 }

core/validation/src/main/java/com/shifthackz/aisdv1/core/validation/url/UrlValidator.kt

@@ -7,9 +7,9 @@ interface UrlValidator {
     operator fun invoke(input: String?): ValidationResult<Error>
 
     sealed interface Error {
-        object Empty : Error
-        object BadScheme : Error
-        object Invalid : Error
-        object Localhost : Error
+        data object Empty : Error
+        data object BadScheme : Error
+        data object Invalid : Error
+        data object Localhost : Error
     }
 }

data/src/main/java/com/shifthackz/aisdv1/data/di/LocalDataSourceModule.kt

@@ -5,11 +5,17 @@ import com.shifthackz.aisdv1.data.gateway.mediastore.MediaStoreGatewayFactory
 import com.shifthackz.aisdv1.data.local.DownloadableModelLocalDataSource
 import com.shifthackz.aisdv1.data.local.GenerationResultLocalDataSource
 import com.shifthackz.aisdv1.data.local.ServerConfigurationLocalDataSource
+import com.shifthackz.aisdv1.data.local.StableDiffusionEmbeddingsLocalDataSource
+import com.shifthackz.aisdv1.data.local.StableDiffusionHyperNetworksLocalDataSource
+import com.shifthackz.aisdv1.data.local.StableDiffusionLorasLocalDataSource
 import com.shifthackz.aisdv1.data.local.StableDiffusionModelsLocalDataSource
 import com.shifthackz.aisdv1.data.local.StableDiffusionSamplersLocalDataSource
 import com.shifthackz.aisdv1.domain.datasource.DownloadableModelDataSource
 import com.shifthackz.aisdv1.domain.datasource.GenerationResultDataSource
 import com.shifthackz.aisdv1.domain.datasource.ServerConfigurationDataSource
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionEmbeddingsDataSource
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionHyperNetworksDataSource
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionLorasDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionModelsDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionSamplersDataSource
 import com.shifthackz.aisdv1.domain.gateway.DatabaseClearGateway
@@ -23,6 +29,9 @@ val localDataSourceModule = module {
     singleOf(::DatabaseClearGatewayImpl) bind DatabaseClearGateway::class
     factoryOf(::StableDiffusionModelsLocalDataSource) bind StableDiffusionModelsDataSource.Local::class
     factoryOf(::StableDiffusionSamplersLocalDataSource) bind StableDiffusionSamplersDataSource.Local::class
+    factoryOf(::StableDiffusionLorasLocalDataSource) bind StableDiffusionLorasDataSource.Local::class
+    factoryOf(::StableDiffusionHyperNetworksLocalDataSource) bind StableDiffusionHyperNetworksDataSource.Local::class
+    factoryOf(::StableDiffusionEmbeddingsLocalDataSource) bind StableDiffusionEmbeddingsDataSource.Local::class
     factoryOf(::ServerConfigurationLocalDataSource) bind ServerConfigurationDataSource.Local::class
     factoryOf(::GenerationResultLocalDataSource) bind GenerationResultDataSource.Local::class
     factoryOf(::DownloadableModelLocalDataSource) bind DownloadableModelDataSource.Local::class

data/src/main/java/com/shifthackz/aisdv1/data/di/RemoteDataSourceModule.kt

@@ -8,14 +8,20 @@ import com.shifthackz.aisdv1.data.remote.HordeGenerationRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.HordeStatusSource
 import com.shifthackz.aisdv1.data.remote.RandomImageRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.ServerConfigurationRemoteDataSource
+import com.shifthackz.aisdv1.data.remote.StableDiffusionEmbeddingsRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.StableDiffusionGenerationRemoteDataSource
+import com.shifthackz.aisdv1.data.remote.StableDiffusionHyperNetworksRemoteDataSource
+import com.shifthackz.aisdv1.data.remote.StableDiffusionLorasRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.StableDiffusionModelsRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.StableDiffusionSamplersRemoteDataSource
 import com.shifthackz.aisdv1.domain.datasource.DownloadableModelDataSource
 import com.shifthackz.aisdv1.domain.datasource.HordeGenerationDataSource
 import com.shifthackz.aisdv1.domain.datasource.RandomImageDataSource
 import com.shifthackz.aisdv1.domain.datasource.ServerConfigurationDataSource
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionEmbeddingsDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionGenerationDataSource
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionHyperNetworksDataSource
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionLorasDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionModelsDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionSamplersDataSource
 import com.shifthackz.aisdv1.domain.entity.ServerSource
@@ -44,6 +50,9 @@ val remoteDataSourceModule = module {
     factoryOf(::StableDiffusionGenerationRemoteDataSource) bind StableDiffusionGenerationDataSource.Remote::class
     factoryOf(::StableDiffusionSamplersRemoteDataSource) bind StableDiffusionSamplersDataSource.Remote::class
     factoryOf(::StableDiffusionModelsRemoteDataSource) bind StableDiffusionModelsDataSource.Remote::class
+    factoryOf(::StableDiffusionLorasRemoteDataSource) bind StableDiffusionLorasDataSource.Remote::class
+    factoryOf(::StableDiffusionHyperNetworksRemoteDataSource) bind StableDiffusionHyperNetworksDataSource.Remote::class
+    factoryOf(::StableDiffusionEmbeddingsRemoteDataSource) bind StableDiffusionEmbeddingsDataSource.Remote::class
     factoryOf(::ServerConfigurationRemoteDataSource) bind ServerConfigurationDataSource.Remote::class
     factoryOf(::RandomImageRemoteDataSource) bind RandomImageDataSource.Remote::class
     factoryOf(::DownloadableModelRemoteDataSource) bind DownloadableModelDataSource.Remote::class

data/src/main/java/com/shifthackz/aisdv1/data/di/RepositoryModule.kt

@@ -8,7 +8,10 @@ import com.shifthackz.aisdv1.data.repository.HordeGenerationRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.LocalDiffusionGenerationRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.RandomImageRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.ServerConfigurationRepositoryImpl
+import com.shifthackz.aisdv1.data.repository.StableDiffusionEmbeddingsRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.StableDiffusionGenerationRepositoryImpl
+import com.shifthackz.aisdv1.data.repository.StableDiffusionHyperNetworksRepositoryImpl
+import com.shifthackz.aisdv1.data.repository.StableDiffusionLorasRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.StableDiffusionModelsRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.StableDiffusionSamplersRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.TemporaryGenerationResultRepositoryImpl
@@ -19,7 +22,10 @@ import com.shifthackz.aisdv1.domain.repository.HordeGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.LocalDiffusionGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.RandomImageRepository
 import com.shifthackz.aisdv1.domain.repository.ServerConfigurationRepository
+import com.shifthackz.aisdv1.domain.repository.StableDiffusionEmbeddingsRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionGenerationRepository
+import com.shifthackz.aisdv1.domain.repository.StableDiffusionHyperNetworksRepository
+import com.shifthackz.aisdv1.domain.repository.StableDiffusionLorasRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionModelsRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionSamplersRepository
 import com.shifthackz.aisdv1.domain.repository.TemporaryGenerationResultRepository
@@ -43,6 +49,9 @@ val repositoryModule = module {
     factoryOf(::StableDiffusionGenerationRepositoryImpl) bind StableDiffusionGenerationRepository::class
     factoryOf(::StableDiffusionModelsRepositoryImpl) bind StableDiffusionModelsRepository::class
     factoryOf(::StableDiffusionSamplersRepositoryImpl) bind StableDiffusionSamplersRepository::class
+    factoryOf(::StableDiffusionLorasRepositoryImpl) bind StableDiffusionLorasRepository::class
+    factoryOf(::StableDiffusionHyperNetworksRepositoryImpl) bind StableDiffusionHyperNetworksRepository::class
+    factoryOf(::StableDiffusionEmbeddingsRepositoryImpl) bind StableDiffusionEmbeddingsRepository::class
     factoryOf(::ServerConfigurationRepositoryImpl) bind ServerConfigurationRepository::class
     factoryOf(::GenerationResultRepositoryImpl) bind GenerationResultRepository::class
     factoryOf(::RandomImageRepositoryImpl) bind RandomImageRepository::class

data/src/main/java/com/shifthackz/aisdv1/data/local/StableDiffusionEmbeddingsLocalDataSource.kt

@@ -0,0 +1,21 @@
+package com.shifthackz.aisdv1.data.local
+
+import com.shifthackz.aisdv1.data.mappers.mapDomainToEntity
+import com.shifthackz.aisdv1.data.mappers.mapEntityToDomain
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionEmbeddingsDataSource
+import com.shifthackz.aisdv1.domain.entity.StableDiffusionEmbedding
+import com.shifthackz.aisdv1.storage.db.cache.dao.StableDiffusionEmbeddingDao
+import com.shifthackz.aisdv1.storage.db.cache.entity.StableDiffusionEmbeddingEntity
+
+internal class StableDiffusionEmbeddingsLocalDataSource(
+    private val dao: StableDiffusionEmbeddingDao,
+) : StableDiffusionEmbeddingsDataSource.Local {
+
+    override fun getEmbeddings() = dao
+        .queryAll()
+        .map(List<StableDiffusionEmbeddingEntity>::mapEntityToDomain)
+
+    override fun insertEmbeddings(list: List<StableDiffusionEmbedding>) = dao
+        .deleteAll()
+        .andThen(dao.insertList(list.mapDomainToEntity()))
+}

data/src/main/java/com/shifthackz/aisdv1/data/local/StableDiffusionHyperNetworksLocalDataSource.kt

@@ -0,0 +1,21 @@
+package com.shifthackz.aisdv1.data.local
+
+import com.shifthackz.aisdv1.data.mappers.mapDomainToEntity
+import com.shifthackz.aisdv1.data.mappers.mapEntityToDomain
+import com.shifthackz.aisdv1.domain.datasource.StableDiffusionHyperNetworksDataSource
+import com.shifthackz.aisdv1.domain.entity.StableDiffusionHyperNetwork
+import com.shifthackz.aisdv1.storage.db.cache.dao.StableDiffusionHyperNetworkDao
+import com.shifthackz.aisdv1.storage.db.cache.entity.StableDiffusionHyperNetworkEntity
+
+internal class StableDiffusionHyperNetworksLocalDataSource(
+    private val dao: StableDiffusionHyperNetworkDao,
+) : StableDiffusionHyperNetworksDataSource.Local {
+
+    override fun getHyperNetworks() = dao
+        .queryAll()
+        .map(List<StableDiffusionHyperNetworkEntity>::mapEntityToDomain)
+
+    override fun insertHyperNetworks(list: List<StableDiffusionHyperNetwork>) = dao
+        .deleteAll()
+        .andThen(dao.insertList(list.mapDomainToEntity()))
+}