Update app.py

app.py

@@ -572,7 +572,7 @@ def infer_upscale(
         
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
-        true_input_image = input_image
+        
         input_image, w_original, h_original, was_resized = process_input(
             input_image, upscale_factor
         )
@@ -607,11 +607,11 @@ def infer_upscale(
         # resize to target desired size
         image = image.resize((w_original * upscale_factor, h_original * upscale_factor))
         image.save("output.jpg")
-        # convert to numpy
-        return [true_input_image, image, seed]
+        
+        return image, seed
     except Exception as e:
         print(f"Error in infer_upscale: {str(e)}")
-        return [None, None, seed]
+        return None, seed
 
 
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as app:
@@ -724,7 +724,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
             upscale_randomize_seed = gr.Checkbox(label="Randomize seed for Upscaling", value=True)
 
     with gr.Row():
-        upscale_result = gr.Image(label="Input / Output for Upscaling", type="pil", interactive=True)
+        upscale_result = gr.Image(label="Upscaled Image", type="pil")
+        upscale_seed_output = gr.Number(label="Seed Used", precision=0)
 
     # Event handlers
     gallery.select(
@@ -757,7 +758,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         inputs=[selected_indices, loras_state],
         outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
     )
-# 이벤트 핸들러 수정
+    # 이벤트 핸들러 수정
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
@@ -768,7 +769,6 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         inputs=[result, history_gallery],
         outputs=history_gallery,
     )
-
     
     # 업스케일 버튼 이벤트 추가
     upscale_button.click(
@@ -781,7 +781,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
             upscale_factor,
             controlnet_conditioning_scale,
         ],
-        outputs=upscale_result,
+        outputs=[upscale_result, upscale_seed_output],
     )
 
 if __name__ == "__main__":