Update app.py

app.py

@@ -260,11 +260,13 @@ def generate_30(
     latent_file_3 = gr.File(),  # Add latents file input
     latent_file_4 = gr.File(),  # Add latents file input
     latent_file_5 = gr.File(),  # Add latents file input
-    latent_file_1_scale: float = 3.8,
-    latent_file_2_scale: float = 3.8,
-    latent_file_3_scale: float = 3.8,
-    latent_file_4_scale: float = 3.8,
-    latent_file_5_scale: float = 3.8,
+    text_scale: float = 1.0,
+    ip_scale: float = 1.0,
+    latent_file_1_scale: float = 1.0,
+    latent_file_2_scale: float = 1.0,
+    latent_file_3_scale: float = 1.0,
+    latent_file_4_scale: float = 1.0,
+    latent_file_5_scale: float = 1.0,
     samples=1,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
@@ -301,6 +303,8 @@ def generate_30(
                 pil_image_5=sd_image_e,
                 prompt=prompt,
                 negative_prompt=negative_prompt,
+                text_scale=text_scale,
+                ip_scale=ip_scale,
                 scale_1=latent_file_1_scale,
                 scale_2=latent_file_2_scale,
                 scale_3=latent_file_3_scale,
@@ -341,11 +345,13 @@ def generate_60(
     latent_file_3 = gr.File(),  # Add latents file input
     latent_file_4 = gr.File(),  # Add latents file input
     latent_file_5 = gr.File(),  # Add latents file input
-    latent_file_1_scale: float = 3.8,
-    latent_file_2_scale: float = 3.8,
-    latent_file_3_scale: float = 3.8,
-    latent_file_4_scale: float = 3.8,
-    latent_file_5_scale: float = 3.8,
+    text_scale: float = 1.0,
+    ip_scale: float = 1.0,
+    latent_file_1_scale: float = 1.0,
+    latent_file_2_scale: float = 1.0,
+    latent_file_3_scale: float = 1.0,
+    latent_file_4_scale: float = 1.0,
+    latent_file_5_scale: float = 1.0,
     samples=1,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
@@ -382,6 +388,8 @@ def generate_60(
                 pil_image_5=sd_image_e,
                 prompt=prompt,
                 negative_prompt=negative_prompt,
+                text_scale=text_scale,
+                ip_scale=ip_scale,
                 scale_1=latent_file_1_scale,
                 scale_2=latent_file_2_scale,
                 scale_3=latent_file_3_scale,
@@ -422,11 +430,13 @@ def generate_90(
     latent_file_3 = gr.File(),  # Add latents file input
     latent_file_4 = gr.File(),  # Add latents file input
     latent_file_5 = gr.File(),  # Add latents file input
-    latent_file_1_scale: float = 3.8,
-    latent_file_2_scale: float = 3.8,
-    latent_file_3_scale: float = 3.8,
-    latent_file_4_scale: float = 3.8,
-    latent_file_5_scale: float = 3.8,
+    text_scale: float = 1.0,
+    ip_scale: float = 1.0,
+    latent_file_1_scale: float = 1.0,
+    latent_file_2_scale: float = 1.0,
+    latent_file_3_scale: float = 1.0,
+    latent_file_4_scale: float = 1.0,
+    latent_file_5_scale: float = 1.0,
     samples=1,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
@@ -463,6 +473,8 @@ def generate_90(
                 pil_image_5=sd_image_e,
                 prompt=prompt,
                 negative_prompt=negative_prompt,
+                text_scale=text_scale,
+                ip_scale=ip_scale,
                 scale_1=latent_file_1_scale,
                 scale_2=latent_file_2_scale,
                 scale_3=latent_file_3_scale,
@@ -530,19 +542,32 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
         run_button_60 = gr.Button("Run 60 Seconds", scale=0)
         run_button_90 = gr.Button("Run 90 Seconds", scale=0)
     result = gr.Gallery(label="Result", columns=1, show_label=False) 
-
+        text_strength =  gr.Slider(
+            label="Text Strength",
+            minimum=0.0,
+            maximum=16.0,
+            step=0.01,
+            value=1.0,
+        )
     with gr.Row():
+        ip_strength =  gr.Slider(
+            label="Image Strength",
+            minimum=0.0,
+            maximum=16.0,
+            step=0.01,
+            value=1.0,
+        )
         latent_file = gr.File(label="Image Prompt (Required)")
         file_1_strength =  gr.Slider(
-            label="Img 1 Str",
+            label="Img 1 %",
             minimum=0.0,
             maximum=16.0,
             step=0.01,
-            value=3.8,
+            value=1.0,
         )
         latent_file_2 = gr.File(label="Image Prompt 2 (Optional)")
         file_2_strength =  gr.Slider(
-            label="Img 2 Str",
+            label="Img 2 %",
             minimum=0.0,
             maximum=16.0,
             step=0.01,
@@ -550,27 +575,27 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
         )
         latent_file_3 = gr.File(label="Image Prompt 3 (Optional)")
         file_3_strength =  gr.Slider(
-            label="Img 3 Str",
+            label="Img 3 %",
             minimum=0.0,
             maximum=16.0,
             step=0.01,
-            value=3.8,
+            value=1.0,
         )
         latent_file_4 = gr.File(label="Image Prompt 4 (Optional)")
         file_4_strength =  gr.Slider(
-            label="Img 4 Str",
+            label="Img 4 %",
             minimum=0.0,
             maximum=16.0,
             step=0.01,
-            value=3.8,
+            value=1.0,
         )
         latent_file_5 = gr.File(label="Image Prompt 5 (Optional)")
         file_5_strength =  gr.Slider(
-            label="Img 5 Str",
+            label="Img 5 %",
             minimum=0.0,
             maximum=16.0,
             step=0.01,
-            value=3.8,
+            value=1.0,
         )
         style_selection = gr.Radio(
             show_label=True,
@@ -663,6 +688,8 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             latent_file_3,
             latent_file_4,
             latent_file_5,
+            text_strength,
+            ip_strength,
             file_1_strength,
             file_2_strength,
             file_3_strength,
@@ -693,6 +720,8 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             latent_file_3,
             latent_file_4,
             latent_file_5,
+            text_strength,
+            ip_strength,
             file_1_strength,
             file_2_strength,
             file_3_strength,
@@ -723,6 +752,8 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             latent_file_3,
             latent_file_4,
             latent_file_5,
+            text_strength,
+            ip_strength,
             file_1_strength,
             file_2_strength,
             file_3_strength,