Update app.py

app.py

@@ -551,7 +551,7 @@ def process_input(input_image, upscale_factor, **kwargs):
     h = h - h % 8
 
     return input_image.resize((w, h)), w_original, h_original, was_resized
-
+    
 @spaces.GPU
 def infer_upscale(
     seed,
@@ -562,7 +562,11 @@ def infer_upscale(
     controlnet_conditioning_scale,
     progress=gr.Progress(track_tqdm=True),
 ):
+    if input_image is None:
+        return None, seed
+
     try:
+
         if input_image is None:
             return None, seed
 
@@ -616,8 +620,8 @@ def infer_upscale(
 
 def check_upscale_input(input_image, *args):
     if input_image is None:
-        return gr.update(), *args, gr.update(visible=True, value="Please provide an input image for upscaling.")
-    return input_image, *args, gr.update(visible=False)
+        return gr.update(interactive=False), *args, gr.update(visible=True, value="Please upload an image for upscaling.")
+    return gr.update(interactive=True), *args, gr.update(visible=False)
     
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as app:
     loras_state = gr.State(loras)
@@ -692,7 +696,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
 
 # 업스케일 관련 UI 추가
     with gr.Row():
-        upscale_button = gr.Button("Upscale")
+        upscale_button = gr.Button("Upscale", interactive=False)
 
     with gr.Row():
         with gr.Column(scale=4):
@@ -777,15 +781,17 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
     )
 
     upscale_button.click(
-        check_upscale_input,
+        infer_upscale,
         inputs=[
-            upscale_input,
             upscale_seed,
             upscale_randomize_seed,
+            upscale_input,
             upscale_steps,
             upscale_factor,
             controlnet_conditioning_scale,
         ],
+
+
         outputs=[
             upscale_input,
             upscale_seed,
@@ -805,8 +811,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
             upscale_factor,
             controlnet_conditioning_scale,
         ],
-        outputs=[upscale_result, upscale_seed_output],
-        show_progress=False
+        outputs=[upscale_result, upscale_seed_output]
     )