Update app.py

app.py

@@ -726,7 +726,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         upscale_result = gr.Image(label="Upscaled Image", type="pil")
         upscale_seed_output = gr.Number(label="Seed Used", precision=0)
 
-    # Event handlers
+
+
     gallery.select(
         update_selection,
         inputs=[selected_indices, loras_state, width, height],
@@ -769,45 +770,39 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         outputs=history_gallery,
     )
 
-
-def check_upscale_input(input_image, *args):
-    if input_image is None:
-        raise gr.Error("Please provide an input image for upscaling.")
-    return input_image, *args
-
-upscale_button.click(
-    check_upscale_input,
-    inputs=[
-        upscale_input,
-        upscale_seed,
-        upscale_randomize_seed,
-        upscale_steps,
-        upscale_factor,
-        controlnet_conditioning_scale,
-    ],
-    outputs=[
-        upscale_input,
-        upscale_seed,
-        upscale_randomize_seed,
-        upscale_steps,
-        upscale_factor,
-        controlnet_conditioning_scale,
-    ]
-).then(
-    infer_upscale,
-    inputs=[
-        upscale_seed,
-        upscale_randomize_seed,
-        upscale_input,
-        upscale_steps,
-        upscale_factor,
-        controlnet_conditioning_scale,
-    ],
-    outputs=[upscale_result, upscale_seed_output],
-)
-
-    
+    # 업스케일 버튼 이벤트 추가 (이 부분을 gr.Blocks 컨텍스트 내부로 이동)
+    upscale_button.click(
+        check_upscale_input,
+        inputs=[
+            upscale_input,
+            upscale_seed,
+            upscale_randomize_seed,
+            upscale_steps,
+            upscale_factor,
+            controlnet_conditioning_scale,
+        ],
+        outputs=[
+            upscale_input,
+            upscale_seed,
+            upscale_randomize_seed,
+            upscale_steps,
+            upscale_factor,
+            controlnet_conditioning_scale,
+        ]
+    ).then(
+        infer_upscale,
+        inputs=[
+            upscale_seed,
+            upscale_randomize_seed,
+            upscale_input,
+            upscale_steps,
+            upscale_factor,
+            controlnet_conditioning_scale,
+        ],
+        outputs=[upscale_result, upscale_seed_output],
+    )
 
 if __name__ == "__main__":
     app.queue(max_size=20)
-    app.launch(debug=True)
+    app.launch(debug=True)
+