Batch generation implementation (#122)

* Batch implementation | Patch 1

* Batch implementation | Patch 2

* Update README.md

* Update README.md

Author: ShiftHackZ

README.md

@@ -16,6 +16,7 @@ Stable Diffusion AI is an easy-to-use app that lets you quickly generate images
 
 - Can use server environment powered by [AI Horde](https://stablehorde.net/) (a crowdsourced distributed cluster of Stable Diffusion workers)
 - Can use server environment powered by [Stable-Diffusion-WebUI](https://github.com/AUTOMATIC1111/stable-diffusion-webui) (AUTOMATIC1111)
+- Can use local environment powered by LocalDiffusion (Beta)
 - Supports original Txt2Img, Img2Img modes
   - **Positive** and **negative** prompt support
   - Support dynamic **size** in range from 64 to 2048 px (for width and height)
@@ -26,7 +27,8 @@ Stable Diffusion AI is an easy-to-use app that lets you quickly generate images
   - **Restore faces** option
   - ( Img2Img ONLY ) : Image selection from device gallery _(requires user permission)_
   - ( Img2Img ONLY ) : Capture input image from camera _(requires user permission)_
-  - ( Img2Img ONLY ) : Fetching random image for the input 
+  - ( Img2Img ONLY ) : Fetching random image for the input
+  - Batch generation with maximum of 20 images (for A1111 and Horde)
 - In-app Gallery, stored locally, contains all AI generated images
   - Displays generated images grid
   - Image detail view: Zoom, Pinch, Generation Info. 
@@ -37,6 +39,7 @@ Stable Diffusion AI is an easy-to-use app that lets you quickly generate images
   - Active SD Model selection
   - Server availability monitoring (http-ping method)
   - Enable/Disable auto-saving of generated images
+  - Enable/Disable saving generated images to `Download/SDAI` android MediaStore folder
   - Clear gallery / app cache
 
 ## Setup instruction
@@ -60,6 +63,12 @@ If for some reason you have no ability to run your server instance, you can togg
 
 AI Horde requires to use API KEY, this mobile app alows to use either default API KEY (which is "0000000000"), or type your own. You can sign up and get your own AI Horde API KEY [here](https://stablehorde.net/register).
 
+### Option 3: Local Diffusion (Beta)
+
+Only **txt2img** mode is supported.
+
+Allows to use phone resources to generate images.
+
 ## Supported languages
 
 App uses the language provided by OS default settings.

app/src/main/java/com/shifthackz/aisdv1/app/AiStableDiffusionClientApp.kt

@@ -20,7 +20,6 @@ import org.koin.android.ext.koin.androidContext
 import org.koin.core.context.startKoin
 import timber.log.Timber
 
-
 class AiStableDiffusionClientApp : Application() {
 
     override fun onCreate() {
@@ -50,8 +49,8 @@ class AiStableDiffusionClientApp : Application() {
 
     private fun initializeLogging() {
         Timber.plant(FileLoggingTree())
-//        if (BuildConfig.DEBUG) {
+        if (BuildConfig.DEBUG) {
             Timber.plant(Timber.DebugTree())
-//        }
+        }
     }
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/entity/ImageToImagePayload.kt

@@ -15,4 +15,5 @@ data class ImageToImagePayload(
     val subSeedStrength: Float,
     val sampler: String,
     val nsfw: Boolean,
+    val batchCount: Int,
 )

domain/src/main/java/com/shifthackz/aisdv1/domain/entity/TextToImagePayload.kt

@@ -13,4 +13,5 @@ data class TextToImagePayload(
     val subSeedStrength: Float,
     val sampler: String,
     val nsfw: Boolean,
+    val batchCount: Int,
 )

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/generation/ImageToImageUseCase.kt

@@ -5,5 +5,5 @@ import com.shifthackz.aisdv1.domain.entity.ImageToImagePayload
 import io.reactivex.rxjava3.core.Single
 
 interface ImageToImageUseCase {
-    operator fun invoke(payload: ImageToImagePayload): Single<AiGenerationResult>
+    operator fun invoke(payload: ImageToImagePayload): Single<List<AiGenerationResult>>
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/generation/ImageToImageUseCaseImpl.kt

@@ -6,6 +6,7 @@ import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.HordeGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionGenerationRepository
+import io.reactivex.rxjava3.core.Observable
 import io.reactivex.rxjava3.core.Single
 
 internal class ImageToImageUseCaseImpl(
@@ -14,9 +15,12 @@ internal class ImageToImageUseCaseImpl(
     private val preferenceManager: PreferenceManager,
 ) : ImageToImageUseCase {
 
-    override fun invoke(payload: ImageToImagePayload) = execute(payload)
+    override fun invoke(payload: ImageToImagePayload) = Observable
+        .range(1, payload.batchCount)
+        .flatMapSingle { generate(payload) }
+        .toList()
 
-    private fun execute(payload: ImageToImagePayload): Single<AiGenerationResult> {
+    private fun generate(payload: ImageToImagePayload): Single<AiGenerationResult> {
         if (preferenceManager.source == ServerSource.HORDE) {
             return hordeGenerationRepository.generateFromImage(payload)
         }

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/generation/TextToImageUseCase.kt

@@ -5,5 +5,5 @@ import com.shifthackz.aisdv1.domain.entity.TextToImagePayload
 import io.reactivex.rxjava3.core.Single
 
 interface TextToImageUseCase {
-    operator fun invoke(payload: TextToImagePayload): Single<AiGenerationResult>
+    operator fun invoke(payload: TextToImagePayload): Single<List<AiGenerationResult>>
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/generation/TextToImageUseCaseImpl.kt

@@ -1,13 +1,12 @@
 package com.shifthackz.aisdv1.domain.usecase.generation
 
-import com.shifthackz.aisdv1.domain.entity.AiGenerationResult
 import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.entity.TextToImagePayload
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.HordeGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.LocalDiffusionGenerationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionGenerationRepository
-import io.reactivex.rxjava3.core.Single
+import io.reactivex.rxjava3.core.Observable
 
 internal class TextToImageUseCaseImpl(
     private val stableDiffusionGenerationRepository: StableDiffusionGenerationRepository,
@@ -16,13 +15,14 @@ internal class TextToImageUseCaseImpl(
     private val preferenceManager: PreferenceManager,
 ) : TextToImageUseCase {
 
-    override operator fun invoke(payload: TextToImagePayload) = execute(payload)
+    override operator fun invoke(payload: TextToImagePayload) = Observable
+        .range(1, payload.batchCount)
+        .flatMapSingle { generate(payload) }
+        .toList()
 
-    private fun execute(payload: TextToImagePayload): Single<AiGenerationResult> {
-        return when (preferenceManager.source) {
-            ServerSource.HORDE -> hordeGenerationRepository.generateFromText(payload)
-            ServerSource.LOCAL -> localDiffusionGenerationRepository.generateFromText(payload)
-            else -> stableDiffusionGenerationRepository.generateFromText(payload)
-        }
+    private fun generate(payload: TextToImagePayload) = when (preferenceManager.source) {
+        ServerSource.HORDE -> hordeGenerationRepository.generateFromText(payload)
+        ServerSource.LOCAL -> localDiffusionGenerationRepository.generateFromText(payload)
+        else -> stableDiffusionGenerationRepository.generateFromText(payload)
     }
 }

presentation/src/main/java/com/shifthackz/aisdv1/presentation/core/GenerationMviEffect.kt

@@ -0,0 +1,7 @@
+package com.shifthackz.aisdv1.presentation.core
+
+import com.shifthackz.aisdv1.core.ui.MviEffect
+
+sealed interface GenerationMviEffect : MviEffect {
+    data class LaunchGalleryDetail(val itemId: Long) : GenerationMviEffect
+}

presentation/src/main/java/com/shifthackz/aisdv1/presentation/core/GenerationMviScreen.kt

@@ -0,0 +1,20 @@
+package com.shifthackz.aisdv1.presentation.core
+
+import androidx.compose.runtime.Composable
+import com.shifthackz.aisdv1.core.ui.MviScreen
+import com.shifthackz.aisdv1.core.ui.MviState
+import com.shifthackz.aisdv1.core.viewmodel.MviViewModel
+
+abstract class GenerationMviScreen<S: MviState, E: GenerationMviEffect>(
+    viewModel: MviViewModel<S, E>,
+    private val launchGalleryDetail: (Long) -> Unit,
+) : MviScreen<S, E>(viewModel) {
+
+    override fun processEffect(effect: E) = when (effect) {
+        is GenerationMviEffect.LaunchGalleryDetail -> launchGalleryDetail(effect.itemId)
+        else -> super.processEffect(effect)
+    }
+
+    @Composable
+    override fun ApplySystemUiColors() = Unit
+}