fixed bug

src/sdgen/ui/layout.py

@@ -216,11 +216,11 @@ def build_ui(txt2img_pipes: dict, img2img_pipes: dict) -> gr.Blocks:
         )
 
         txt_controls = build_txt2img_tab(
-            make_txt2img_handler(model_choice, txt2img_pipes),
+            make_txt2img_handler(model_choice.value, txt2img_pipes),
         )
 
         img_controls = build_img2img_tab(
-            make_img2img_handler(model_choice, img2img_pipes),
+            make_img2img_handler(model_choice.value, img2img_pipes),
         )
 
         build_upscaler_tab(

src/sdgen/ui/tabs/img2img_tab.py

@@ -33,8 +33,7 @@ def build_img2img_tab(handler: Callable[..., Tuple[Any, dict]]) -> Img2ImgContro
     """
     with gr.Tab("Image → Image"):
         with gr.Row():
-            # Left: Controls
-            with gr.Column(scale=1):
+            with gr.Column():
                 input_image = gr.Image(
                     label="Input Image",
                     type="numpy",
@@ -93,8 +92,7 @@ the prompt more strictly. "
 
                 generate_button = gr.Button("Generate")
 
-            # Right: Output preview
-            with gr.Column(scale=2):
+            with gr.Column():
                 out_image = gr.Image(
                     label="Output",
                     type="pil",