Update app.py

app.py

@@ -268,7 +268,10 @@ with gr.Blocks(css='style.css') as demo:
 
     def show_reconstruction_button():
         return reconstruct_button.update(visible=True)
-            
+
+    def hide_hide_reconstruction_button():
+        return reconstruct_button.update(visible=True),hide_reconstruct_button.update(visible=False)
+        
     def hide_reconstruction_buttons():
         return reconstruct_button.update(visible=False), hide_reconstruct_button.update(visible=False)
 
@@ -480,10 +483,9 @@ with gr.Blocks(css='style.css') as demo:
 
     hide_reconstruct_button.click(fn = hide_reconstruction, 
         outputs=[ddpm_edited_image], 
-        queue=False).then(fn = hide_reconstruction_buttons, 
+        queue=False).then(fn = hide_hide_reconstruction_button, 
                           outputs=[reconstruct_button, 
-                                   hide_reconstruct_button]).then(fn = show_reconstruction_button, 
-                                                                  outputs=[reconstruct_button])
+                                   hide_reconstruct_button])
 
     
     # Repeat inversion when these params are changed: