From f62dbdd7d511fb474fa58ac577a63fb4530df998 Mon Sep 17 00:00:00 2001
From: ArEnSc <6252325+ArEnSc@users.noreply.github.com>
Date: Wed, 15 May 2024 15:09:31 -0700
Subject: [PATCH] Style Strength button (#383)

Co-authored-by: Storyteller <storyteller@MacBook-Pro-de-Storyteller.local>
---
 .../itemTabs/StylizeTab/StyleButtons.tsx      | 30 +++++++++++++++++++
 app/pages/PageEnigma/js/api_manager.ts        |  4 ++-
 app/pages/PageEnigma/js/editor.ts             |  5 ++--
 .../PageEnigma/models/generationOptions.ts    |  1 +
 4 files changed, 37 insertions(+), 3 deletions(-)

diff --git a/app/pages/PageEnigma/comps/SidePanelTabs/itemTabs/StylizeTab/StyleButtons.tsx b/app/pages/PageEnigma/comps/SidePanelTabs/itemTabs/StylizeTab/StyleButtons.tsx
index 078d0358f4..1fdbc0c277 100644
--- a/app/pages/PageEnigma/comps/SidePanelTabs/itemTabs/StylizeTab/StyleButtons.tsx
+++ b/app/pages/PageEnigma/comps/SidePanelTabs/itemTabs/StylizeTab/StyleButtons.tsx
@@ -15,6 +15,12 @@ export function StyleButtons() {
   const [upscale, setUpscale] = useState(false);
   const [faceDetail, setFaceDetail] = useState(false);
 
+  const [value, setStyleStrength] = useState(0.7);
+
+  const sliderChanged = (event:React.ChangeEvent<HTMLInputElement>) => {
+     setStyleStrength(parseFloat(event.target.value));
+  };
+
   const switchPreview = async () => {
     Queue.publish({
       queueName: QueueNames.TO_ENGINE,
@@ -32,9 +38,11 @@ export function StyleButtons() {
   };
 
   const generateMovie = async () => {
+    
     const options: GenerationOptions = {
       upscale: upscale,
       faceDetail: faceDetail,
+      styleStrength: value,
     };
 
     Queue.publish({
@@ -73,6 +81,27 @@ export function StyleButtons() {
         )}
       </div>
       <div className="w-full">
+
+      <Label>
+          <div className="mb-1 leading-tight">
+            Set the Style Strength, the higher the value the more the style will be applied,
+            the lower the value the closer to source.
+          </div>
+        </Label>
+
+      <input 
+        type="range" 
+        value={value} min="0" 
+        max="1.0"
+        step="0.1"
+        onChange={sliderChanged} 
+        style={{width: "100%"}}
+        ></input>
+        <Label>
+          <div className="mb-1 leading-tight">
+            Currently set to {value}
+          </div>
+        </Label>
         <PremiumLock requiredPlan="any" plural={true}>
           <Switch.Group>
             <div className="mt-1 flex gap-6">
@@ -83,6 +112,7 @@ export function StyleButtons() {
                 <Switch
                   checked={upscale}
                   onChange={setUpscale}
+
                   className={`${
                     upscale
                       ? "bg-brand-primary"
diff --git a/app/pages/PageEnigma/js/api_manager.ts b/app/pages/PageEnigma/js/api_manager.ts
index 7a719d9764..efbddbb243 100644
--- a/app/pages/PageEnigma/js/api_manager.ts
+++ b/app/pages/PageEnigma/js/api_manager.ts
@@ -361,7 +361,8 @@ export class APIManager {
     negative_prompt: string,
     visibility: Visibility,
     use_face_detailer: boolean = false,
-    use_upscaler: boolean = false
+    use_upscaler: boolean = false,
+    style_strength: number = 1.0,
   ) {
 
     const uuid = uuidv4();
@@ -378,6 +379,7 @@ export class APIManager {
       creator_set_visibility: visibility,
       use_face_detailer: use_face_detailer,
       use_upscaler: use_upscaler,
+      style_strength: style_strength,
     };
 
     const json_data = JSON.stringify(data);
diff --git a/app/pages/PageEnigma/js/editor.ts b/app/pages/PageEnigma/js/editor.ts
index ae9d515af1..fe6c6e88ae 100644
--- a/app/pages/PageEnigma/js/editor.ts
+++ b/app/pages/PageEnigma/js/editor.ts
@@ -275,7 +275,7 @@ class Editor {
     this.negative_prompt = "";
     this.art_style = ArtStyle.Anime2DFlat;
 
-    this.generation_options = { faceDetail: false, upscale:false }
+    this.generation_options = { faceDetail: false, upscale:false, styleStrength: 1.0 };
   }
 
   isEmpty(value: string) {
@@ -1243,7 +1243,8 @@ class Editor {
         this.negative_prompt,
         Visibility.Public,
         this.generation_options.faceDetail,
-        this.generation_options.upscale
+        this.generation_options.upscale,
+        this.generation_options.styleStrength,
       )
       .catch((error) => {
         console.log(error);
diff --git a/app/pages/PageEnigma/models/generationOptions.ts b/app/pages/PageEnigma/models/generationOptions.ts
index d30c9ee897..fc62108db2 100644
--- a/app/pages/PageEnigma/models/generationOptions.ts
+++ b/app/pages/PageEnigma/models/generationOptions.ts
@@ -1,4 +1,5 @@
 export interface GenerationOptions {
     upscale: boolean;
     faceDetail: boolean;
+    styleStrength: number;
   }
\ No newline at end of file