Stability AI provider (#153)

* Stability AI | Patch 1

* Stability AI | Patch 2

* Stability AI | Patch 3

* Stability AI | Patch 4

* Stability AI | Patch 5

* Resolve conflicts

* Update translations

Author: ShiftHackZ

app/build.gradle

@@ -22,10 +22,12 @@ android {
         buildConfigField "String", "HUGGING_FACE_INFERENCE_URL", "\"https://api-inference.huggingface.co/\""
         buildConfigField "String", "HORDE_AI_URL", "\"https://stablehorde.net/\""
         buildConfigField "String", "OPEN_AI_URL", "\"https://api.openai.com/\""
+        buildConfigField "String", "STABILITY_AI_URL", "\"https://api.stability.ai/\""
 
         buildConfigField "String", "HORDE_AI_SIGN_UP_URL", "\"https://stablehorde.net/register\""
         buildConfigField "String", "HUGGING_FACE_INFO_URL", "\"https://huggingface.co/docs/api-inference/index\""
         buildConfigField "String", "OPEN_AI_INFO_URL", "\"https://platform.openai.com/api-keys\""
+        buildConfigField "String", "STABILITY_AI_INFO_URL", "\"https://platform.stability.ai/\""
         buildConfigField "String", "UPDATE_API_URL", "\"https://sdai.moroz.cc\""
         buildConfigField "String", "DEMO_MODE_API_URL", "\"https://sdai.moroz.cc\""
         buildConfigField "String", "POLICY_URL", "\"https://sdai.moroz.cc/policy.html\""

app/src/main/java/com/shifthackz/aisdv1/app/di/ProvidersModule.kt

@@ -44,6 +44,7 @@ val providersModule = module {
             override val huggingFaceApiUrl: String = BuildConfig.HUGGING_FACE_URL
             override val huggingFaceInferenceApiUrl = BuildConfig.HUGGING_FACE_INFERENCE_URL
             override val openAiApiUrl: String = BuildConfig.OPEN_AI_URL
+            override val stabilityAiApiUrl = BuildConfig.STABILITY_AI_URL
         }
     }
 
@@ -67,6 +68,11 @@ val providersModule = module {
                     NetworkHeaders.AUTHORIZATION to key
                 }
 
+                ServerSource.STABILITY_AI -> {
+                    val key = "${NetworkPrefixes.BEARER} ${preference.stabilityAiApiKey}"
+                    NetworkHeaders.AUTHORIZATION to key
+                }
+
                 else -> null
             }
         }
@@ -94,6 +100,7 @@ val providersModule = module {
             override val hordeSignUpUrl: String = BuildConfig.HORDE_AI_SIGN_UP_URL
             override val huggingFaceUrl: String = BuildConfig.HUGGING_FACE_INFO_URL
             override val openAiInfoUrl: String = BuildConfig.OPEN_AI_INFO_URL
+            override val stabilityAiInfoUrl: String = BuildConfig.STABILITY_AI_INFO_URL
             override val privacyPolicyUrl: String = BuildConfig.POLICY_URL
             override val donateUrl: String = BuildConfig.DONATE_URL
             override val gitHubSourceUrl: String = BuildConfig.GITHUB_SOURCE_URL

core/common/src/main/java/com/shifthackz/aisdv1/core/common/links/LinksProvider.kt

@@ -5,6 +5,7 @@ interface LinksProvider {
     val hordeSignUpUrl: String
     val huggingFaceUrl: String
     val openAiInfoUrl: String
+    val stabilityAiInfoUrl: String
     val privacyPolicyUrl: String
     val donateUrl: String
     val gitHubSourceUrl: String

core/common/src/main/java/com/shifthackz/aisdv1/core/common/model/Quadruple.kt

@@ -0,0 +1,14 @@
+package com.shifthackz.aisdv1.core.common.model
+
+import java.io.Serializable
+
+
+data class Quadruple<out A, out B, out C, out D>(
+    val first: A,
+    val second: B,
+    val third: C,
+    val fourth: D,
+) : Serializable {
+
+    override fun toString(): String = "($first, $second, $third, $fourth)"
+}

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

@@ -6,6 +6,7 @@ import com.shifthackz.aisdv1.data.local.DownloadableModelLocalDataSource
 import com.shifthackz.aisdv1.data.local.GenerationResultLocalDataSource
 import com.shifthackz.aisdv1.data.local.HuggingFaceModelsLocalDataSource
 import com.shifthackz.aisdv1.data.local.ServerConfigurationLocalDataSource
+import com.shifthackz.aisdv1.data.local.StabilityAiCreditsLocalDataSource
 import com.shifthackz.aisdv1.data.local.StableDiffusionEmbeddingsLocalDataSource
 import com.shifthackz.aisdv1.data.local.StableDiffusionHyperNetworksLocalDataSource
 import com.shifthackz.aisdv1.data.local.StableDiffusionLorasLocalDataSource
@@ -15,6 +16,7 @@ import com.shifthackz.aisdv1.domain.datasource.DownloadableModelDataSource
 import com.shifthackz.aisdv1.domain.datasource.GenerationResultDataSource
 import com.shifthackz.aisdv1.domain.datasource.HuggingFaceModelsDataSource
 import com.shifthackz.aisdv1.domain.datasource.ServerConfigurationDataSource
+import com.shifthackz.aisdv1.domain.datasource.StabilityAiCreditsDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionEmbeddingsDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionHyperNetworksDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionLorasDataSource
@@ -29,6 +31,7 @@ import org.koin.dsl.module
 
 val localDataSourceModule = module {
     singleOf(::DatabaseClearGatewayImpl) bind DatabaseClearGateway::class
+    singleOf(::StabilityAiCreditsLocalDataSource) bind StabilityAiCreditsDataSource.Local::class
     factoryOf(::StableDiffusionModelsLocalDataSource) bind StableDiffusionModelsDataSource.Local::class
     factoryOf(::StableDiffusionSamplersLocalDataSource) bind StableDiffusionSamplersDataSource.Local::class
     factoryOf(::StableDiffusionLorasLocalDataSource) bind StableDiffusionLorasDataSource.Local::class

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

@@ -11,6 +11,9 @@ import com.shifthackz.aisdv1.data.remote.HuggingFaceModelsRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.OpenAiGenerationRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.RandomImageRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.ServerConfigurationRemoteDataSource
+import com.shifthackz.aisdv1.data.remote.StabilityAiCreditsRemoteDataSource
+import com.shifthackz.aisdv1.data.remote.StabilityAiEnginesRemoteDataSource
+import com.shifthackz.aisdv1.data.remote.StabilityAiGenerationRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.StableDiffusionEmbeddingsRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.StableDiffusionGenerationRemoteDataSource
 import com.shifthackz.aisdv1.data.remote.StableDiffusionHyperNetworksRemoteDataSource
@@ -24,6 +27,9 @@ import com.shifthackz.aisdv1.domain.datasource.HuggingFaceModelsDataSource
 import com.shifthackz.aisdv1.domain.datasource.OpenAiGenerationDataSource
 import com.shifthackz.aisdv1.domain.datasource.RandomImageDataSource
 import com.shifthackz.aisdv1.domain.datasource.ServerConfigurationDataSource
+import com.shifthackz.aisdv1.domain.datasource.StabilityAiCreditsDataSource
+import com.shifthackz.aisdv1.domain.datasource.StabilityAiEnginesDataSource
+import com.shifthackz.aisdv1.domain.datasource.StabilityAiGenerationDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionEmbeddingsDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionGenerationDataSource
 import com.shifthackz.aisdv1.domain.datasource.StableDiffusionHyperNetworksDataSource
@@ -65,6 +71,9 @@ val remoteDataSourceModule = module {
     factoryOf(::RandomImageRemoteDataSource) bind RandomImageDataSource.Remote::class
     factoryOf(::DownloadableModelRemoteDataSource) bind DownloadableModelDataSource.Remote::class
     factoryOf(::HuggingFaceModelsRemoteDataSource) bind HuggingFaceModelsDataSource.Remote::class
+    factoryOf(::StabilityAiGenerationRemoteDataSource) bind StabilityAiGenerationDataSource.Remote::class
+    factoryOf(::StabilityAiCreditsRemoteDataSource) bind StabilityAiCreditsDataSource.Remote::class
+    factoryOf(::StabilityAiEnginesRemoteDataSource) bind StabilityAiEnginesDataSource.Remote::class
 
     factory<ServerConnectivityGateway> {
         val lambda: () -> Boolean = {

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

@@ -11,6 +11,9 @@ import com.shifthackz.aisdv1.data.repository.LocalDiffusionGenerationRepositoryI
 import com.shifthackz.aisdv1.data.repository.OpenAiGenerationRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.RandomImageRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.ServerConfigurationRepositoryImpl
+import com.shifthackz.aisdv1.data.repository.StabilityAiCreditsRepositoryImpl
+import com.shifthackz.aisdv1.data.repository.StabilityAiEnginesRepositoryImpl
+import com.shifthackz.aisdv1.data.repository.StabilityAiGenerationRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.StableDiffusionEmbeddingsRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.StableDiffusionGenerationRepositoryImpl
 import com.shifthackz.aisdv1.data.repository.StableDiffusionHyperNetworksRepositoryImpl
@@ -28,6 +31,9 @@ import com.shifthackz.aisdv1.domain.repository.LocalDiffusionGenerationRepositor
 import com.shifthackz.aisdv1.domain.repository.OpenAiGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.RandomImageRepository
 import com.shifthackz.aisdv1.domain.repository.ServerConfigurationRepository
+import com.shifthackz.aisdv1.domain.repository.StabilityAiCreditsRepository
+import com.shifthackz.aisdv1.domain.repository.StabilityAiEnginesRepository
+import com.shifthackz.aisdv1.domain.repository.StabilityAiGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionEmbeddingsRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionHyperNetworksRepository
@@ -54,6 +60,9 @@ val repositoryModule = module {
     factoryOf(::HordeGenerationRepositoryImpl) bind HordeGenerationRepository::class
     factoryOf(::HuggingFaceGenerationRepositoryImpl) bind HuggingFaceGenerationRepository::class
     factoryOf(::OpenAiGenerationRepositoryImpl) bind OpenAiGenerationRepository::class
+    factoryOf(::StabilityAiGenerationRepositoryImpl) bind StabilityAiGenerationRepository::class
+    factoryOf(::StabilityAiCreditsRepositoryImpl) bind StabilityAiCreditsRepository::class
+    factoryOf(::StabilityAiEnginesRepositoryImpl) bind StabilityAiEnginesRepository::class
     factoryOf(::StableDiffusionGenerationRepositoryImpl) bind StableDiffusionGenerationRepository::class
     factoryOf(::StableDiffusionModelsRepositoryImpl) bind StableDiffusionModelsRepository::class
     factoryOf(::StableDiffusionSamplersRepositoryImpl) bind StableDiffusionSamplersRepository::class

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

@@ -0,0 +1,21 @@
+package com.shifthackz.aisdv1.data.local
+
+import com.shifthackz.aisdv1.domain.datasource.StabilityAiCreditsDataSource
+import io.reactivex.rxjava3.core.BackpressureStrategy
+import io.reactivex.rxjava3.core.Completable
+import io.reactivex.rxjava3.subjects.BehaviorSubject
+
+internal class StabilityAiCreditsLocalDataSource : StabilityAiCreditsDataSource.Local {
+
+    private val creditsSubject: BehaviorSubject<Float> = BehaviorSubject.createDefault(0f)
+
+    override fun get() = creditsSubject
+        .firstOrError()
+        .onErrorReturn { 0f }
+
+    override fun save(value: Float) = Completable.fromAction {
+        creditsSubject.onNext(value)
+    }
+
+    override fun observe() = creditsSubject.toFlowable(BackpressureStrategy.LATEST)
+}

data/src/main/java/com/shifthackz/aisdv1/data/mappers/ImageToImagePayloadMappers.kt

@@ -2,6 +2,8 @@ package com.shifthackz.aisdv1.data.mappers
 
 import com.shifthackz.aisdv1.domain.entity.AiGenerationResult
 import com.shifthackz.aisdv1.domain.entity.ImageToImagePayload
+import com.shifthackz.aisdv1.domain.entity.StabilityAiClipGuidance
+import com.shifthackz.aisdv1.domain.entity.StabilityAiStylePreset
 import com.shifthackz.aisdv1.network.request.HordeGenerationAsyncRequest
 import com.shifthackz.aisdv1.network.request.HuggingFaceGenerationRequest
 import com.shifthackz.aisdv1.network.request.ImageToImageRequest
@@ -71,6 +73,26 @@ fun ImageToImagePayload.mapToHuggingFaceRequest(): HuggingFaceGenerationRequest
     )
 }
 
+fun ImageToImagePayload.mapToStabilityAiRequest() = with(this) {
+    buildMap {
+        buildList {
+            addAll(prompt.mapToStabilityPrompt(1.0))
+            addAll(negativePrompt.mapToStabilityPrompt(-1.0))
+        }.forEachIndexed { index, stpRaw ->
+            this["text_prompts[$index][text]"] = stpRaw.text
+            this["text_prompts[$index][weight]"] = stpRaw.weight.toString()
+        }
+        this["image_strength"] = "$denoisingStrength"
+        this["cfg_scale"] = "$cfgScale"
+        this["clip_guidance_preset"] = (stabilityAiClipGuidance ?: StabilityAiClipGuidance.NONE).toString()
+        this["seed"] = (seed.toLongOrNull()?.coerceIn(0L .. 4294967295L) ?: 0L).toString()
+        this["steps"] = "$samplingSteps"
+        stabilityAiStylePreset?.takeIf { it != StabilityAiStylePreset.NONE }?.key?.let {
+            this["style_preset"] = it
+        }
+    }
+}
+
 fun Pair<ImageToImagePayload, SdGenerationResponse>.mapToAiGenResult(): AiGenerationResult =
     let { (payload, response) ->
         AiGenerationResult(

data/src/main/java/com/shifthackz/aisdv1/data/mappers/StabilityAiEngineMappers.kt

@@ -0,0 +1,13 @@
+package com.shifthackz.aisdv1.data.mappers
+
+import com.shifthackz.aisdv1.domain.entity.StabilityAiEngine
+import com.shifthackz.aisdv1.network.model.StabilityAiEngineRaw
+
+//region RAW --> DOMAIN
+fun List<StabilityAiEngineRaw>.mapRawToDomain(): List<StabilityAiEngine> =
+    map(StabilityAiEngineRaw::mapRawToDomain)
+
+fun StabilityAiEngineRaw.mapRawToDomain(): StabilityAiEngine = with(this) {
+    StabilityAiEngine(id ?: "", name ?: "")
+}
+//endregion