From f4a8595dacc20ff46b9c6e9a7506ac7870fba12c Mon Sep 17 00:00:00 2001
From: Pabloader <pabloader@pabloader.ru>
Date: Wed, 15 Apr 2026 08:14:47 +0000
Subject: [PATCH] Add sampler to image generation settings.

---
 .../storywriter/components/settings/image.tsx | 41 +++++++++++++++++--
 src/games/storywriter/contexts/state.tsx      |  2 +
 src/games/storywriter/utils/llm.ts            | 18 ++++++++
 src/games/storywriter/utils/tools.ts          |  3 +-
 4 files changed, 59 insertions(+), 5 deletions(-)

diff --git a/src/games/storywriter/components/settings/image.tsx b/src/games/storywriter/components/settings/image.tsx
index 1b8780e..bcbcd29 100644
--- a/src/games/storywriter/components/settings/image.tsx
+++ b/src/games/storywriter/components/settings/image.tsx
@@ -3,11 +3,12 @@ import { useInputCallback } from "@common/hooks/useInputCallback";
 import { useInputState } from "@common/hooks/useInputState";
 import clsx from "clsx";
 import styles from "../../assets/settings-modal.module.css";
+import LLM from "../../utils/llm";
 import { DEFAULT_IMAGE_GENERATION_SETTINGS, useAppState } from "../../contexts/state";
 
 export const ImageSettings = () => {
     const { imageGenerationSettings, dispatch } = useAppState();
-    const { width, height, negative_prompt } = imageGenerationSettings;
+    const { width, height, negative_prompt, sampler_name } = imageGenerationSettings;
 
     const [widthDraft, setWidthDraft] = useInputState(String(width));
     const [heightDraft, setHeightDraft] = useInputState(String(height));
@@ -15,14 +16,24 @@ export const ImageSettings = () => {
     const commitWidth = () => {
         const parsed = parseInt(widthDraft, 10);
         if (!isNaN(parsed) && parsed > 0) {
-            dispatch({ type: 'SET_IMAGE_GENERATION_SETTINGS', settings: { width: parsed } });
+            const snapped = Math.max(64, Math.round(parsed / 64) * 64);
+            dispatch({
+                type: 'SET_IMAGE_GENERATION_SETTINGS',
+                settings: { width: snapped }
+            });
+            setWidthDraft(String(snapped));
         }
     };
 
     const commitHeight = () => {
         const parsed = parseInt(heightDraft, 10);
         if (!isNaN(parsed) && parsed > 0) {
-            dispatch({ type: 'SET_IMAGE_GENERATION_SETTINGS', settings: { height: parsed } });
+            const snapped = Math.max(64, Math.round(parsed / 64) * 64);
+            dispatch({
+                type: 'SET_IMAGE_GENERATION_SETTINGS',
+                settings: { height: snapped }
+            });
+            setHeightDraft(String(snapped));
         }
     };
 
@@ -30,8 +41,18 @@ export const ImageSettings = () => {
         dispatch({ type: 'SET_IMAGE_GENERATION_SETTINGS', settings: { negative_prompt: value } });
     }, []);
 
+    const handleSamplerChange = useInputCallback((value) => {
+        dispatch({
+            type: 'SET_IMAGE_GENERATION_SETTINGS',
+            settings: { sampler_name: value as LLM.ImageSamplerName },
+        });
+    }, []);
+
     const handleReset = () => {
-        dispatch({ type: 'SET_IMAGE_GENERATION_SETTINGS', settings: DEFAULT_IMAGE_GENERATION_SETTINGS });
+        dispatch({
+            type: 'SET_IMAGE_GENERATION_SETTINGS',
+            settings: DEFAULT_IMAGE_GENERATION_SETTINGS,
+        });
         setWidthDraft(String(DEFAULT_IMAGE_GENERATION_SETTINGS.width));
         setHeightDraft(String(DEFAULT_IMAGE_GENERATION_SETTINGS.height));
     };
@@ -70,6 +91,18 @@ export const ImageSettings = () => {
                     onKeyDown={(e) => e.key === 'Enter' && commitHeight()}
                 />
             </div>
+            <div class={styles.formGroup}>
+                <label class={styles.label}>Sampler</label>
+                <select
+                    value={sampler_name}
+                    onChange={handleSamplerChange}
+                    class={styles.select}
+                >
+                    {Object.values(LLM.ImageSamplerName).map(s => (
+                        <option key={s} value={s}>{s}</option>
+                    ))}
+                </select>
+            </div>
             <div class={clsx(styles.formGroup, styles.formGroupFill)}>
                 <label class={styles.label}>Negative Prompt</label>
                 <ContentEditable
diff --git a/src/games/storywriter/contexts/state.tsx b/src/games/storywriter/contexts/state.tsx
index 18cde9d..fc554a1 100644
--- a/src/games/storywriter/contexts/state.tsx
+++ b/src/games/storywriter/contexts/state.tsx
@@ -76,12 +76,14 @@ export interface ImageGenerationSettings {
     width: number;
     height: number;
     negative_prompt: string;
+    sampler_name: LLM.ImageSamplerName;
 }
 
 export const DEFAULT_IMAGE_GENERATION_SETTINGS: ImageGenerationSettings = {
     width: 768,
     height: 768,
     negative_prompt: 'ugly face, bad face, no face, poorly drawn face, greyscale, depth of field, mutated fingers, mutated hands, extra fingers, deformed, ugly, bad anatomy, bad proportions, extra limbs, signature, text, lowres, error, missing fingers, missing limbs, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, watermark, username, blurry, artist name',
+    sampler_name: LLM.ImageSamplerName.KEulerA,
 };
 
 export const DEFAULT_GENERATION_SETTINGS: GenerationSettings = {
diff --git a/src/games/storywriter/utils/llm.ts b/src/games/storywriter/utils/llm.ts
index f32bdef..452399e 100644
--- a/src/games/storywriter/utils/llm.ts
+++ b/src/games/storywriter/utils/llm.ts
@@ -118,10 +118,28 @@ namespace LLM {
         error: string;
     }
 
+    export enum ImageSamplerName {
+        DDIM = 'DDIM',
+        LCM = 'lcm',
+        DPMSolver = 'dpmsolver',
+        KHeun = 'k_heun',
+        KLMS = 'k_lms',
+        KDpmpp2M = 'k_dpmpp_2m',
+        KDpmppSDE = 'k_dpmpp_sde',
+        KDpm2 = 'k_dpm_2',
+        KDpmFast = 'k_dpm_fast',
+        KDpmAdaptive = 'k_dpm_adaptive',
+        KEuler = 'k_euler',
+        KDpmpp2sA = 'k_dpmpp_2s_a',
+        KEulerA = 'k_euler_a',
+        KDpm2A = 'k_dpm_2_a',
+    }
+
     export interface ImageGenerationSettings {
         width?: number;
         height?: number;
         negative_prompt?: string;
+        sampler_name?: ImageSamplerName;
     }
 
     export interface ImageGenerationRequest {
diff --git a/src/games/storywriter/utils/tools.ts b/src/games/storywriter/utils/tools.ts
index cab23f0..bb62720 100644
--- a/src/games/storywriter/utils/tools.ts
+++ b/src/games/storywriter/utils/tools.ts
@@ -478,7 +478,7 @@ export namespace Tools {
                 if (!appState.imageModel) {
                     return 'Error: No image model configured';
                 }
-                const { width: defaultWidth, height: defaultHeight, negative_prompt } = appState.imageGenerationSettings;
+                const { width: defaultWidth, height: defaultHeight, negative_prompt, sampler_name } = appState.imageGenerationSettings;
                 const response = await LLM.generateImage(appState.connection, {
                     model: appState.imageModel.id,
                     prompt: args.prompt,
@@ -487,6 +487,7 @@ export namespace Tools {
                         width: args.width ?? defaultWidth,
                         height: args.height ?? defaultHeight,
                         negative_prompt: negative_prompt || undefined,
+                        sampler_name: sampler_name || undefined,
                     },
                 });
                 if ('error' in response) {