Update app_v4.py

app_v4.py

@@ -26,6 +26,30 @@ logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 #############################
 
+presets = {
+    "Strict Upscale": {
+        "scale": 1.0,
+        "steps": 8,
+        "controlnet_conditioning_scale": 0.75,
+        "guidance_scale": 4.0,
+        "guidance_end": 0.9
+    },
+    "Creative Upscale": {
+        "scale": 2.0,
+        "steps": 6,
+        "controlnet_conditioning_scale": 0.42,
+        "guidance_scale": 3.0,
+        "guidance_end": 0.5
+    },
+    "High Detail Upscale": {
+        "scale": 1.25,
+        "steps": 10,
+        "controlnet_conditioning_scale": 0.9,
+        "guidance_scale": 10.0,
+        "guidance_end": 0.9
+    }
+}
+
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = 1000000
 
@@ -157,6 +181,20 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
         # print("Type: " + str(type(image)))
     return image
 
+def update_parameters(preset):
+    if preset in presets:
+        params = presets[preset]
+        return (
+            params["scale"],
+            params["steps"],
+            params["controlnet_conditioning_scale"],
+            params["guidance_scale"],
+            params["guidance_end"]
+        )
+    else:
+        # Default values if preset is not found
+        return 1.0, 8, 0.6, 3.5, 1.0
+
 def process_image(control_image, user_prompt, system_prompt, scale, steps, 
                 controlnet_conditioning_scale, guidance_scale, seed, 
                 guidance_end, temperature, max_new_tokens, log_prompt):
@@ -210,11 +248,14 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as demo:
         with gr.Column(scale=1):
             prompt = gr.Textbox(lines=4, info="Enter your prompt here or wait for auto-generation...", label="Image Description")
             focus = gr.Textbox(label="Area(s) of Focus", info="e.g. 'face', 'eyes', 'hair', 'clothes', 'background', etc.", value="clothing material, textures, ethnicity")
-            scale = gr.Slider(1, 3, value=1, label="Scale (Upscale Factor)", step=0.25)
+            scale = gr.Slider(1, 3, value=1, label="Scale (Upscale Factor)", step=0.1)
             with gr.Row():
                 generate_button = gr.Button("Generate Image", variant="primary")
                 caption_button = gr.Button("Generate Caption", variant="secondary")
         with gr.Column(scale=1):
+            with gr.Row():
+                preset_choices = list(presets.keys())
+                preset_radio = gr.Radio(choices=preset_choices, label="Select Preset", value=preset_choices[0])
             seed = gr.Slider(0, MAX_SEED, value=42, label="Seed", step=1)
             steps = gr.Slider(2, 16, value=8, label="Steps", step=1)
             controlnet_conditioning_scale = gr.Slider(0, 1, value=0.6, label="ControlNet Scale")
@@ -296,6 +337,11 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as demo:
         inputs=[caption_state, focus_state],
         outputs=[prompt]
     )
+    preset_radio.change(
+        fn=update_parameters,
+        inputs=[preset_radio],
+        outputs=[scale, steps, controlnet_conditioning_scale, guidance_scale, guidance_end]
+    )
     def hello(profile: gr.OAuthProfile | None) -> str:
         if profile is None:
             return "Hello guest! There is a bug with HF ZeroGPUs that are afffecting some usage on certain spaces. Testing out some possible solutions."