Update all result galleries to use filepath type for proper image handling

ui/shared/hires_fix_ui.py

@@ -1,104 +1,104 @@
-import gradio as gr
-from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
-from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
-from .ui_components import (
-    create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_ipadapter_ui, create_embedding_ui,
-    create_conditioning_ui, create_vae_override_ui,
-    create_model_architecture_filter_ui, create_category_filter_ui,
-    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui, create_hidream_o1_reference_ui
-)
-
-default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
-DEFAULT_STEPS = default_vals.get('steps', 20)
-DEFAULT_CFG = default_vals.get('cfg', 5.0)
-DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
-DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
-DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
-DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
-
-def create_ui():
-    prefix = "hires_fix"
-    components = {}
-    
-    with gr.Column():
-        components.update(create_model_architecture_filter_ui(prefix))
-        
-        with gr.Row():
-            components.update(create_category_filter_ui(prefix))
-            components[f'base_model_{prefix}'] = gr.Dropdown(
-                label="Base Model", 
-                choices=list(MODEL_MAP_CHECKPOINT.keys()), 
-                value=list(MODEL_MAP_CHECKPOINT.keys())[0], 
-                scale=3,
-                allow_custom_value=True
-            )
-            with gr.Column(scale=1): 
-                components[f'run_{prefix}'] = gr.Button("Run Hires. Fix", variant="primary")
-        
-        with gr.Row():
-            with gr.Column(scale=1):
-                components[f'input_image_{prefix}'] = gr.Image(type="pil", label="Input Image", height=255)
-            with gr.Column(scale=2):
-                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3, value=DEFAULT_POS_PROMPT)
-                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=3, value=DEFAULT_NEG_PROMPT)
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                with gr.Row():
-                    components[f'hires_upscaler_{prefix}'] = gr.Dropdown(
-                        label="Upscaler", 
-                        choices=["nearest-exact", "bilinear", "area", "bicubic", "bislerp"], 
-                        value="nearest-exact"
-                    )
-                    components[f'hires_scale_by_{prefix}'] = gr.Slider(
-                        label="Upscale by", minimum=1.0, maximum=4.0, step=0.1, value=1.5
-                    )
-                
-                with gr.Row():
-                    components[f'denoise_{prefix}'] = gr.Slider(label="Denoise Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.55)
-                
-                with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(
-                        label="Sampler", 
-                        choices=SAMPLER_CHOICES, 
-                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
-                    )
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(
-                        label="Scheduler", 
-                        choices=SCHEDULER_CHOICES, 
-                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
-                    )
-                with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
-                with gr.Row():
-                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
-                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
-                with gr.Row():
-                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
-                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
-                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU.")
-
-                components[f'width_{prefix}'] = gr.State(value=512)
-                components[f'height_{prefix}'] = gr.State(value=512)
-
-            with gr.Column(scale=1):
-                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, object_fit="contain", height=610)
-
-
-        components.update(create_lora_settings_ui(prefix))
-        components.update(create_controlnet_ui(prefix))
-        components.update(create_anima_controlnet_lllite_ui(prefix))
-        components.update(create_ipadapter_ui(prefix))
-        components.update(create_flux1_ipadapter_ui(prefix))
-        components.update(create_sd3_ipadapter_ui(prefix))
-        components.update(create_style_ui(prefix))
-        components.update(create_embedding_ui(prefix))
-        components.update(create_conditioning_ui(prefix))
-        components.update(create_reference_latent_ui(prefix))
-        components.update(create_hidream_o1_reference_ui(prefix))
-        components.update(create_vae_override_ui(prefix))
-            
+import gradio as gr
+from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
+from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
+from .ui_components import (
+    create_lora_settings_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_ipadapter_ui, create_embedding_ui,
+    create_conditioning_ui, create_vae_override_ui,
+    create_model_architecture_filter_ui, create_category_filter_ui,
+    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
+    create_reference_latent_ui, create_hidream_o1_reference_ui
+)
+
+default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
+DEFAULT_STEPS = default_vals.get('steps', 20)
+DEFAULT_CFG = default_vals.get('cfg', 5.0)
+DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
+DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
+DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
+DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
+
+def create_ui():
+    prefix = "hires_fix"
+    components = {}
+    
+    with gr.Column():
+        components.update(create_model_architecture_filter_ui(prefix))
+        
+        with gr.Row():
+            components.update(create_category_filter_ui(prefix))
+            components[f'base_model_{prefix}'] = gr.Dropdown(
+                label="Base Model", 
+                choices=list(MODEL_MAP_CHECKPOINT.keys()), 
+                value=list(MODEL_MAP_CHECKPOINT.keys())[0], 
+                scale=3,
+                allow_custom_value=True
+            )
+            with gr.Column(scale=1): 
+                components[f'run_{prefix}'] = gr.Button("Run Hires. Fix", variant="primary")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                components[f'input_image_{prefix}'] = gr.Image(type="pil", label="Input Image", height=255)
+            with gr.Column(scale=2):
+                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3, value=DEFAULT_POS_PROMPT)
+                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=3, value=DEFAULT_NEG_PROMPT)
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                with gr.Row():
+                    components[f'hires_upscaler_{prefix}'] = gr.Dropdown(
+                        label="Upscaler", 
+                        choices=["nearest-exact", "bilinear", "area", "bicubic", "bislerp"], 
+                        value="nearest-exact"
+                    )
+                    components[f'hires_scale_by_{prefix}'] = gr.Slider(
+                        label="Upscale by", minimum=1.0, maximum=4.0, step=0.1, value=1.5
+                    )
+                
+                with gr.Row():
+                    components[f'denoise_{prefix}'] = gr.Slider(label="Denoise Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.55)
+                
+                with gr.Row():
+                    components[f'sampler_{prefix}'] = gr.Dropdown(
+                        label="Sampler", 
+                        choices=SAMPLER_CHOICES, 
+                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
+                    )
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(
+                        label="Scheduler", 
+                        choices=SCHEDULER_CHOICES, 
+                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
+                    )
+                with gr.Row():
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
+                with gr.Row():
+                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
+                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
+                with gr.Row():
+                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
+                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
+                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU.")
+
+                components[f'width_{prefix}'] = gr.State(value=512)
+                components[f'height_{prefix}'] = gr.State(value=512)
+
+            with gr.Column(scale=1):
+                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, type="filepath", height=610)
+
+
+        components.update(create_lora_settings_ui(prefix))
+        components.update(create_controlnet_ui(prefix))
+        components.update(create_anima_controlnet_lllite_ui(prefix))
+        components.update(create_ipadapter_ui(prefix))
+        components.update(create_flux1_ipadapter_ui(prefix))
+        components.update(create_sd3_ipadapter_ui(prefix))
+        components.update(create_style_ui(prefix))
+        components.update(create_embedding_ui(prefix))
+        components.update(create_conditioning_ui(prefix))
+        components.update(create_reference_latent_ui(prefix))
+        components.update(create_hidream_o1_reference_ui(prefix))
+        components.update(create_vae_override_ui(prefix))
+            
     return components

ui/shared/img2img_ui.py

@@ -1,86 +1,86 @@
-import gradio as gr
-from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
-from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
-from .ui_components import (
-    create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
-    create_conditioning_ui, create_vae_override_ui,
-    create_model_architecture_filter_ui, create_category_filter_ui,
-    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui, create_hidream_o1_reference_ui
-)
-
-default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
-DEFAULT_STEPS = default_vals.get('steps', 20)
-DEFAULT_CFG = default_vals.get('cfg', 5.0)
-DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
-DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
-DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
-DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
-
-def create_ui():
-    prefix = "img2img"
-    components = {}
-    
-    with gr.Column():
-        components.update(create_model_architecture_filter_ui(prefix))
-        
-        with gr.Row():
-            components.update(create_category_filter_ui(prefix))
-            components[f'base_model_{prefix}'] = gr.Dropdown(label="Base Model", choices=list(MODEL_MAP_CHECKPOINT.keys()), value=list(MODEL_MAP_CHECKPOINT.keys())[0], scale=3, allow_custom_value=True)
-            with gr.Column(scale=1): 
-                components[f'run_{prefix}'] = gr.Button("Run", variant="primary")
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                components[f'input_image_{prefix}'] = gr.Image(type="pil", label="Input Image", height=255)
-
-            with gr.Column(scale=2):
-                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3, value=DEFAULT_POS_PROMPT)
-                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=3, value=DEFAULT_NEG_PROMPT)
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                components[f'denoise_{prefix}'] = gr.Slider(label="Denoise Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.7)
-                
-                with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(
-                        label="Sampler", 
-                        choices=SAMPLER_CHOICES, 
-                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
-                    )
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(
-                        label="Scheduler", 
-                        choices=SCHEDULER_CHOICES, 
-                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
-                    )
-                with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
-                with gr.Row():
-                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
-                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
-                with gr.Row():
-                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
-                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
-                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU. Longer jobs may need more time.")
-                
-            with gr.Column(scale=1):
-                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, object_fit="contain", height=505)
-        
-
-        components.update(create_lora_settings_ui(prefix))
-        components.update(create_controlnet_ui(prefix))
-        components.update(create_anima_controlnet_lllite_ui(prefix))
-        components.update(create_diffsynth_controlnet_ui(prefix))
-        components.update(create_ipadapter_ui(prefix))
-        components.update(create_flux1_ipadapter_ui(prefix))
-        components.update(create_sd3_ipadapter_ui(prefix))
-        components.update(create_embedding_ui(prefix))
-        components.update(create_style_ui(prefix))
-        components.update(create_conditioning_ui(prefix))
-        components.update(create_reference_latent_ui(prefix))
-        components.update(create_hidream_o1_reference_ui(prefix))
-        components.update(create_vae_override_ui(prefix))
-            
+import gradio as gr
+from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
+from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
+from .ui_components import (
+    create_lora_settings_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
+    create_conditioning_ui, create_vae_override_ui,
+    create_model_architecture_filter_ui, create_category_filter_ui,
+    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
+    create_reference_latent_ui, create_hidream_o1_reference_ui
+)
+
+default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
+DEFAULT_STEPS = default_vals.get('steps', 20)
+DEFAULT_CFG = default_vals.get('cfg', 5.0)
+DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
+DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
+DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
+DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
+
+def create_ui():
+    prefix = "img2img"
+    components = {}
+    
+    with gr.Column():
+        components.update(create_model_architecture_filter_ui(prefix))
+        
+        with gr.Row():
+            components.update(create_category_filter_ui(prefix))
+            components[f'base_model_{prefix}'] = gr.Dropdown(label="Base Model", choices=list(MODEL_MAP_CHECKPOINT.keys()), value=list(MODEL_MAP_CHECKPOINT.keys())[0], scale=3, allow_custom_value=True)
+            with gr.Column(scale=1): 
+                components[f'run_{prefix}'] = gr.Button("Run", variant="primary")
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                components[f'input_image_{prefix}'] = gr.Image(type="pil", label="Input Image", height=255)
+
+            with gr.Column(scale=2):
+                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3, value=DEFAULT_POS_PROMPT)
+                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=3, value=DEFAULT_NEG_PROMPT)
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                components[f'denoise_{prefix}'] = gr.Slider(label="Denoise Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.7)
+                
+                with gr.Row():
+                    components[f'sampler_{prefix}'] = gr.Dropdown(
+                        label="Sampler", 
+                        choices=SAMPLER_CHOICES, 
+                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
+                    )
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(
+                        label="Scheduler", 
+                        choices=SCHEDULER_CHOICES, 
+                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
+                    )
+                with gr.Row():
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
+                with gr.Row():
+                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
+                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
+                with gr.Row():
+                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
+                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
+                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU. Longer jobs may need more time.")
+                
+            with gr.Column(scale=1):
+                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, type="filepath", height=505)
+        
+
+        components.update(create_lora_settings_ui(prefix))
+        components.update(create_controlnet_ui(prefix))
+        components.update(create_anima_controlnet_lllite_ui(prefix))
+        components.update(create_diffsynth_controlnet_ui(prefix))
+        components.update(create_ipadapter_ui(prefix))
+        components.update(create_flux1_ipadapter_ui(prefix))
+        components.update(create_sd3_ipadapter_ui(prefix))
+        components.update(create_embedding_ui(prefix))
+        components.update(create_style_ui(prefix))
+        components.update(create_conditioning_ui(prefix))
+        components.update(create_reference_latent_ui(prefix))
+        components.update(create_hidream_o1_reference_ui(prefix))
+        components.update(create_vae_override_ui(prefix))
+            
     return components

ui/shared/inpaint_ui.py

@@ -1,119 +1,119 @@
-import gradio as gr
-from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
-from .ui_components import (
-    create_base_parameter_ui, create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
-    create_conditioning_ui, create_vae_override_ui,
-    create_model_architecture_filter_ui, create_category_filter_ui,
-    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui, create_hidream_o1_reference_ui
-)
-
-default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
-DEFAULT_STEPS = default_vals.get('steps', 20)
-DEFAULT_CFG = default_vals.get('cfg', 5.0)
-DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
-DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
-DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
-DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
-
-def create_ui():
-    prefix = "inpaint"
-    components = {}
-    
-    with gr.Column():
-        with gr.Row() as arch_row:
-            components.update(create_model_architecture_filter_ui(prefix))
-
-        with gr.Row() as model_and_run_row:
-            components.update(create_category_filter_ui(prefix))
-            components[f'base_model_{prefix}'] = gr.Dropdown(
-                label="Base Model", 
-                choices=list(MODEL_MAP_CHECKPOINT.keys()), 
-                value=list(MODEL_MAP_CHECKPOINT.keys())[0], 
-                scale=3,
-                allow_custom_value=True
-            )
-            with gr.Column(scale=1): 
-                components[f'run_{prefix}'] = gr.Button("Run Inpaint", variant="primary")
-        
-        components[f'model_and_run_row_{prefix}'] = [arch_row, model_and_run_row]
-
-        with gr.Row() as main_content_row:
-            with gr.Column(scale=1) as editor_column:
-                components[f'view_mode_{prefix}'] = gr.Radio(
-                    ["Normal View", "Fullscreen View"], 
-                    label="Editor View", 
-                    value="Normal View", 
-                    interactive=True
-                )
-                components[f'input_image_dict_{prefix}'] = gr.ImageEditor(
-                    type="pil", 
-                    label="Image & Mask", 
-                    height=272
-                )
-            components[f'editor_column_{prefix}'] = editor_column
-
-            with gr.Column(scale=2) as prompts_column:
-                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=6, value=DEFAULT_POS_PROMPT)
-                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=6, value=DEFAULT_NEG_PROMPT)
-            components[f'prompts_column_{prefix}'] = prompts_column
-
-        with gr.Row() as params_and_gallery_row:
-            with gr.Column(scale=1):
-                from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
-                with gr.Row():
-                    components[f'denoise_{prefix}'] = gr.Slider(
-                        label="Denoise", minimum=0.0, maximum=1.0, step=0.05, value=1.0
-                    )
-                    components[f'grow_mask_by_{prefix}'] = gr.Slider(
-                        label="Grow Mask By", minimum=0, maximum=64, step=1, value=6
-                    )
-                with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(
-                        label="Sampler", 
-                        choices=SAMPLER_CHOICES, 
-                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
-                    )
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(
-                        label="Scheduler", 
-                        choices=SCHEDULER_CHOICES, 
-                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
-                    )
-                with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
-                with gr.Row():
-                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
-                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
-                with gr.Row():
-                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
-                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
-                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU.")
-
-                components[f'width_{prefix}'] = gr.State(value=512)
-                components[f'height_{prefix}'] = gr.State(value=512)
-
-            with gr.Column(scale=1):
-                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, object_fit="contain", height=510)
-        
-        components[f'params_and_gallery_row_{prefix}'] = params_and_gallery_row
-            
-        with gr.Column() as accordion_wrapper:
-
-            components.update(create_lora_settings_ui(prefix))
-            components.update(create_controlnet_ui(prefix))
-            components.update(create_anima_controlnet_lllite_ui(prefix))
-            components.update(create_diffsynth_controlnet_ui(prefix))
-            components.update(create_ipadapter_ui(prefix))
-            components.update(create_flux1_ipadapter_ui(prefix))
-            components.update(create_sd3_ipadapter_ui(prefix))
-            components.update(create_style_ui(prefix))
-            components.update(create_embedding_ui(prefix))
-            components.update(create_conditioning_ui(prefix))
-            components.update(create_reference_latent_ui(prefix))
-            components.update(create_hidream_o1_reference_ui(prefix))
-            components.update(create_vae_override_ui(prefix))
-        components[f'accordion_wrapper_{prefix}'] = accordion_wrapper
-            
+import gradio as gr
+from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
+from .ui_components import (
+    create_base_parameter_ui, create_lora_settings_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
+    create_conditioning_ui, create_vae_override_ui,
+    create_model_architecture_filter_ui, create_category_filter_ui,
+    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
+    create_reference_latent_ui, create_hidream_o1_reference_ui
+)
+
+default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
+DEFAULT_STEPS = default_vals.get('steps', 20)
+DEFAULT_CFG = default_vals.get('cfg', 5.0)
+DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
+DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
+DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
+DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
+
+def create_ui():
+    prefix = "inpaint"
+    components = {}
+    
+    with gr.Column():
+        with gr.Row() as arch_row:
+            components.update(create_model_architecture_filter_ui(prefix))
+
+        with gr.Row() as model_and_run_row:
+            components.update(create_category_filter_ui(prefix))
+            components[f'base_model_{prefix}'] = gr.Dropdown(
+                label="Base Model", 
+                choices=list(MODEL_MAP_CHECKPOINT.keys()), 
+                value=list(MODEL_MAP_CHECKPOINT.keys())[0], 
+                scale=3,
+                allow_custom_value=True
+            )
+            with gr.Column(scale=1): 
+                components[f'run_{prefix}'] = gr.Button("Run Inpaint", variant="primary")
+        
+        components[f'model_and_run_row_{prefix}'] = [arch_row, model_and_run_row]
+
+        with gr.Row() as main_content_row:
+            with gr.Column(scale=1) as editor_column:
+                components[f'view_mode_{prefix}'] = gr.Radio(
+                    ["Normal View", "Fullscreen View"], 
+                    label="Editor View", 
+                    value="Normal View", 
+                    interactive=True
+                )
+                components[f'input_image_dict_{prefix}'] = gr.ImageEditor(
+                    type="pil", 
+                    label="Image & Mask", 
+                    height=272
+                )
+            components[f'editor_column_{prefix}'] = editor_column
+
+            with gr.Column(scale=2) as prompts_column:
+                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=6, value=DEFAULT_POS_PROMPT)
+                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=6, value=DEFAULT_NEG_PROMPT)
+            components[f'prompts_column_{prefix}'] = prompts_column
+
+        with gr.Row() as params_and_gallery_row:
+            with gr.Column(scale=1):
+                from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
+                with gr.Row():
+                    components[f'denoise_{prefix}'] = gr.Slider(
+                        label="Denoise", minimum=0.0, maximum=1.0, step=0.05, value=1.0
+                    )
+                    components[f'grow_mask_by_{prefix}'] = gr.Slider(
+                        label="Grow Mask By", minimum=0, maximum=64, step=1, value=6
+                    )
+                with gr.Row():
+                    components[f'sampler_{prefix}'] = gr.Dropdown(
+                        label="Sampler", 
+                        choices=SAMPLER_CHOICES, 
+                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
+                    )
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(
+                        label="Scheduler", 
+                        choices=SCHEDULER_CHOICES, 
+                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
+                    )
+                with gr.Row():
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
+                with gr.Row():
+                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
+                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
+                with gr.Row():
+                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
+                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
+                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU.")
+
+                components[f'width_{prefix}'] = gr.State(value=512)
+                components[f'height_{prefix}'] = gr.State(value=512)
+
+            with gr.Column(scale=1):
+                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, type="filepath", height=510)
+        
+        components[f'params_and_gallery_row_{prefix}'] = params_and_gallery_row
+            
+        with gr.Column() as accordion_wrapper:
+
+            components.update(create_lora_settings_ui(prefix))
+            components.update(create_controlnet_ui(prefix))
+            components.update(create_anima_controlnet_lllite_ui(prefix))
+            components.update(create_diffsynth_controlnet_ui(prefix))
+            components.update(create_ipadapter_ui(prefix))
+            components.update(create_flux1_ipadapter_ui(prefix))
+            components.update(create_sd3_ipadapter_ui(prefix))
+            components.update(create_style_ui(prefix))
+            components.update(create_embedding_ui(prefix))
+            components.update(create_conditioning_ui(prefix))
+            components.update(create_reference_latent_ui(prefix))
+            components.update(create_hidream_o1_reference_ui(prefix))
+            components.update(create_vae_override_ui(prefix))
+        components[f'accordion_wrapper_{prefix}'] = accordion_wrapper
+            
     return components

ui/shared/outpaint_ui.py

@@ -1,101 +1,101 @@
-import gradio as gr
-from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
-from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
-from .ui_components import (
-    create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
-    create_conditioning_ui, create_vae_override_ui,
-    create_model_architecture_filter_ui, create_category_filter_ui,
-    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui, create_hidream_o1_reference_ui
-)
-
-default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
-DEFAULT_STEPS = default_vals.get('steps', 20)
-DEFAULT_CFG = default_vals.get('cfg', 5.0)
-DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
-DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
-DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
-DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
-
-def create_ui():
-    prefix = "outpaint"
-    components = {}
-    
-    with gr.Column():
-        components.update(create_model_architecture_filter_ui(prefix))
-        
-        with gr.Row():
-            components.update(create_category_filter_ui(prefix))
-            components[f'base_model_{prefix}'] = gr.Dropdown(
-                label="Base Model", 
-                choices=list(MODEL_MAP_CHECKPOINT.keys()), 
-                value=list(MODEL_MAP_CHECKPOINT.keys())[0], 
-                scale=3,
-                allow_custom_value=True
-            )
-            with gr.Column(scale=1): 
-                components[f'run_{prefix}'] = gr.Button("Run Outpaint", variant="primary")
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                components[f'input_image_{prefix}'] = gr.Image(type="pil", label="Input Image", height=255)
-            with gr.Column(scale=2):
-                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3, value=DEFAULT_POS_PROMPT)
-                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=3, value=DEFAULT_NEG_PROMPT)
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                with gr.Row():
-                    components[f'left_{prefix}'] = gr.Slider(label="Pad Left", minimum=0, maximum=512, step=64, value=64)
-                    components[f'right_{prefix}'] = gr.Slider(label="Pad Right", minimum=0, maximum=512, step=64, value=64)
-                with gr.Row():
-                    components[f'top_{prefix}'] = gr.Slider(label="Pad Top", minimum=0, maximum=512, step=64, value=64)
-                    components[f'bottom_{prefix}'] = gr.Slider(label="Pad Bottom", minimum=0, maximum=512, step=64, value=64)
-                
-                components[f'feathering_{prefix}'] = gr.Slider(label="Feathering / Grow Mask", minimum=0, maximum=100, step=1, value=10)
-                
-                with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(
-                        label="Sampler", 
-                        choices=SAMPLER_CHOICES, 
-                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
-                    )
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(
-                        label="Scheduler", 
-                        choices=SCHEDULER_CHOICES, 
-                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
-                    )
-                with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
-                with gr.Row():
-                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
-                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
-                with gr.Row():
-                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
-                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
-                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU.")
-
-                components[f'width_{prefix}'] = gr.State(value=512)
-                components[f'height_{prefix}'] = gr.State(value=512)
-
-            with gr.Column(scale=1):
-                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, object_fit="contain", height=685)
-        
-
-        components.update(create_lora_settings_ui(prefix))
-        components.update(create_controlnet_ui(prefix))
-        components.update(create_anima_controlnet_lllite_ui(prefix))
-        components.update(create_diffsynth_controlnet_ui(prefix))
-        components.update(create_ipadapter_ui(prefix))
-        components.update(create_flux1_ipadapter_ui(prefix))
-        components.update(create_sd3_ipadapter_ui(prefix))
-        components.update(create_style_ui(prefix))
-        components.update(create_embedding_ui(prefix))
-        components.update(create_conditioning_ui(prefix))
-        components.update(create_reference_latent_ui(prefix))
-        components.update(create_hidream_o1_reference_ui(prefix))
-        components.update(create_vae_override_ui(prefix))
-            
+import gradio as gr
+from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
+from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
+from .ui_components import (
+    create_lora_settings_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
+    create_conditioning_ui, create_vae_override_ui,
+    create_model_architecture_filter_ui, create_category_filter_ui,
+    create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
+    create_reference_latent_ui, create_hidream_o1_reference_ui
+)
+
+default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
+DEFAULT_STEPS = default_vals.get('steps', 20)
+DEFAULT_CFG = default_vals.get('cfg', 5.0)
+DEFAULT_SAMPLER = default_vals.get('sampler_name', 'euler')
+DEFAULT_SCHEDULER = default_vals.get('scheduler', 'simple')
+DEFAULT_POS_PROMPT = default_vals.get('positive_prompt', '')
+DEFAULT_NEG_PROMPT = default_vals.get('negative_prompt', '')
+
+def create_ui():
+    prefix = "outpaint"
+    components = {}
+    
+    with gr.Column():
+        components.update(create_model_architecture_filter_ui(prefix))
+        
+        with gr.Row():
+            components.update(create_category_filter_ui(prefix))
+            components[f'base_model_{prefix}'] = gr.Dropdown(
+                label="Base Model", 
+                choices=list(MODEL_MAP_CHECKPOINT.keys()), 
+                value=list(MODEL_MAP_CHECKPOINT.keys())[0], 
+                scale=3,
+                allow_custom_value=True
+            )
+            with gr.Column(scale=1): 
+                components[f'run_{prefix}'] = gr.Button("Run Outpaint", variant="primary")
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                components[f'input_image_{prefix}'] = gr.Image(type="pil", label="Input Image", height=255)
+            with gr.Column(scale=2):
+                components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3, value=DEFAULT_POS_PROMPT)
+                components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=3, value=DEFAULT_NEG_PROMPT)
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                with gr.Row():
+                    components[f'left_{prefix}'] = gr.Slider(label="Pad Left", minimum=0, maximum=512, step=64, value=64)
+                    components[f'right_{prefix}'] = gr.Slider(label="Pad Right", minimum=0, maximum=512, step=64, value=64)
+                with gr.Row():
+                    components[f'top_{prefix}'] = gr.Slider(label="Pad Top", minimum=0, maximum=512, step=64, value=64)
+                    components[f'bottom_{prefix}'] = gr.Slider(label="Pad Bottom", minimum=0, maximum=512, step=64, value=64)
+                
+                components[f'feathering_{prefix}'] = gr.Slider(label="Feathering / Grow Mask", minimum=0, maximum=100, step=1, value=10)
+                
+                with gr.Row():
+                    components[f'sampler_{prefix}'] = gr.Dropdown(
+                        label="Sampler", 
+                        choices=SAMPLER_CHOICES, 
+                        value=DEFAULT_SAMPLER if DEFAULT_SAMPLER in SAMPLER_CHOICES else (SAMPLER_CHOICES[0] if SAMPLER_CHOICES else 'euler')
+                    )
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(
+                        label="Scheduler", 
+                        choices=SCHEDULER_CHOICES, 
+                        value=DEFAULT_SCHEDULER if DEFAULT_SCHEDULER in SCHEDULER_CHOICES else (SCHEDULER_CHOICES[0] if SCHEDULER_CHOICES else 'simple')
+                    )
+                with gr.Row():
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=DEFAULT_STEPS)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=DEFAULT_CFG)
+                with gr.Row():
+                    components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
+                    components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)
+                with gr.Row():
+                    components[f'clip_skip_{prefix}'] = gr.Slider(label="Clip Skip", minimum=1, maximum=2, step=1, value=1, visible=False, interactive=True)
+                    components[f'guidance_{prefix}'] = gr.Slider(label="Guidance (FLUX)", minimum=1.0, maximum=10.0, step=0.1, value=3.5, visible=False, interactive=True)
+                    components[f'zero_gpu_{prefix}'] = gr.Number(label="ZeroGPU Duration (s)", value=None, placeholder="Default: 60s, Max: 120s", info="Optional: Set how long to reserve the GPU.")
+
+                components[f'width_{prefix}'] = gr.State(value=512)
+                components[f'height_{prefix}'] = gr.State(value=512)
+
+            with gr.Column(scale=1):
+                components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=1, type="filepath", height=685)
+        
+
+        components.update(create_lora_settings_ui(prefix))
+        components.update(create_controlnet_ui(prefix))
+        components.update(create_anima_controlnet_lllite_ui(prefix))
+        components.update(create_diffsynth_controlnet_ui(prefix))
+        components.update(create_ipadapter_ui(prefix))
+        components.update(create_flux1_ipadapter_ui(prefix))
+        components.update(create_sd3_ipadapter_ui(prefix))
+        components.update(create_style_ui(prefix))
+        components.update(create_embedding_ui(prefix))
+        components.update(create_conditioning_ui(prefix))
+        components.update(create_reference_latent_ui(prefix))
+        components.update(create_hidream_o1_reference_ui(prefix))
+        components.update(create_vae_override_ui(prefix))
+            
     return components