Upload app_builder.py

ui/app_builder.py

@@ -1293,6 +1293,48 @@ def build_app() -> gr.Blocks:
                 # Resolution manager popup (hidden by default)
                 with gr.Column(visible=False, elem_id="resolution-popup-container") as res_popup:
                     gr.Markdown("#### Manage Resolutions")
+
+                    # Custom Resolution Section
+                    with gr.Group():
+                        gr.Markdown("##### Custom Resolution")
+                        use_custom_res = gr.Checkbox(
+                            label="Use Custom Resolution",
+                            value=False,
+                            elem_id="naia-use-custom-res"
+                        )
+                        
+                        with gr.Row(visible=False) as custom_res_inputs:
+                            custom_width = gr.Number(
+                                label="Width", 
+                                value=1024, 
+                                step=64, 
+                                minimum=64, 
+                                precision=0,
+                                elem_id="naia-custom-width"
+                            )
+                            custom_height = gr.Number(
+                                label="Height", 
+                                value=1024, 
+                                step=64, 
+                                minimum=64, 
+                                precision=0,
+                                elem_id="naia-custom-height"
+                            )
+                        
+                        custom_res_warning = gr.Markdown(
+                            "", 
+                            visible=False,
+                            elem_id="naia-custom-res-warning"
+                        )
+                        apply_custom_btn = gr.Button(
+                            "Apply Custom Resolution", 
+                            size="sm", 
+                            visible=False,
+                            elem_id="naia-apply-custom-btn"
+                        )
+
+                    gr.HTML("<hr style='margin: 16px 0; border: 0; border-top: 1px solid #eee;'>")
+
                     gr.Markdown("Toggle which resolutions are available for random selection:")
                     res_checkboxes = gr.CheckboxGroup(
                         choices=DEFAULT_RESOLUTIONS,
@@ -2055,6 +2097,69 @@ def build_app() -> gr.Blocks:
             outputs=[enabled_resolutions]
         )
 
+        # ========== Custom Resolution Handlers ==========
+
+        def toggle_custom_inputs(enabled):
+            return gr.update(visible=enabled), gr.update(visible=enabled)
+
+        use_custom_res.change(
+            fn=toggle_custom_inputs,
+            inputs=[use_custom_res],
+            outputs=[custom_res_inputs, apply_custom_btn]
+        )
+
+        def validate_custom_res(w, h):
+            if w is None or h is None:
+                return gr.update(visible=False, value="")
+                
+            msg = ""
+            
+            # Check 64 multiple
+            if w % 64 != 0 or h % 64 != 0:
+                msg += "⚠️ Resolution must be a multiple of 64.<br>"
+            
+            # Check 1MP limit
+            pixels = w * h
+            if pixels > 1024 * 1024:
+                msg += "<span style='color: red; font-weight: bold;'>🔴 Resolution exceeds 1MP (1024x1024). This will incur Anlas cost!</span>"
+            
+            return gr.update(value=msg, visible=bool(msg))
+
+        custom_width.change(
+            fn=validate_custom_res, 
+            inputs=[custom_width, custom_height], 
+            outputs=[custom_res_warning]
+        )
+        custom_height.change(
+            fn=validate_custom_res, 
+            inputs=[custom_width, custom_height], 
+            outputs=[custom_res_warning]
+        )
+
+        def apply_custom_resolution(w, h):
+            # Enforce 64 multiple (round to nearest)
+            w = int(round(w / 64) * 64)
+            h = int(round(h / 64) * 64)
+            
+            res_str = f"{w} x {h}"
+            
+            # Add to choices if not present
+            new_choices = DEFAULT_RESOLUTIONS.copy()
+            if res_str not in new_choices:
+                new_choices.append(res_str)
+            
+            return (
+                gr.update(choices=new_choices, value=res_str), # Update dropdown
+                False, # Uncheck Random
+                gr.update(visible=False) # Close popup
+            )
+
+        apply_custom_btn.click(
+            fn=apply_custom_resolution,
+            inputs=[custom_width, custom_height],
+            outputs=[resolution, random_resolution, res_popup]
+        )
+
         # ========== Character Panel Handlers ==========
 
         # Helper function to build character status text