Update app.py

app.py

@@ -114,12 +114,12 @@ def infer_with_lora(input_image, prompt, selected_index, seed=0, guidance_scale=
         ).images[0]
         
         # The seed value is no longer returned, as it's not being changed.
-        return image, gr.update(visible=True), lora_scale
+        return image, lora_scale
     
     except Exception as e:
         print(f"Error during inference: {e}")
         # Return an error state for all outputs
-        return None, gr.update(visible=False), lora_scale
+        return None, lora_scale
 
 # CSS styling
 css = """
@@ -218,8 +218,7 @@ with gr.Blocks(css=css, theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend
                 )
                 run_button = gr.Button("Generate", elem_id="run_button")
             
-            result = gr.Image(label="Generated Image", interactive=False)
-            reuse_button = gr.Button("Reuse this image", visible=False)
+            result = gr.Image(label="Generated Image", interactive=False, height=512)
             
             with gr.Accordion("Advanced Settings", open=False):
                 lora_scale = gr.Slider(
@@ -273,13 +272,7 @@ with gr.Blocks(css=css, theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend
         triggers=[run_button.click, prompt.submit],
         fn=infer_with_lora_wrapper,
         inputs=[input_image, prompt, selected_state, lora_state, custom_loaded_lora, seed, guidance_scale, num_inference_steps, lora_scale, gr_flux_loras],
-        outputs=[result, reuse_button, lora_state]
-    )
-    
-    reuse_button.click(
-        fn=lambda image: image,
-        inputs=[result],
-        outputs=[input_image]
+        outputs=[result, lora_state]
     )
     
     # Initialize gallery