Update app.py

app.py

@@ -328,6 +328,12 @@ def generate_image(prompt, structure_image, style_image, depth_strength=15, styl
         saved_path = f"output/{saved['ui']['images'][0]['filename']}"
         return saved_path
 
+
+
+output_image = gr.Image(label="Virtual Try-On Result")
+
+
+
 # Create Gradio interface
 examples = [
     ["person wearing fashionable clothing", "f1.webp", "f11.webp", 15, 0.6],
@@ -335,8 +341,6 @@ examples = [
     ["person wearing casual outfit", "f3.webp", "f31.webp", 15, 0.5],
 ]
 
-output_image = gr.Image(label="Virtual Try-On Result")
-
 # Gradio 인터페이스 생성
 demo = gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange")
 
@@ -378,34 +382,26 @@ with demo:
                                 value=0.5, 
                                 label="Style Transfer Strength"
                             )
-                    tryon_btn = gr.Button("Generate Try-On")
                     
-                    gr.Examples(
-                        examples=examples,
-                        inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
-                        outputs=output_image,
-                        fn=generate_image,
-                        cache_examples=False
-                    )
-                
                 with gr.Column():
                     output_image = gr.Image(label="Virtual Try-On Result")
-    
-    
-    generate_btn.click(
-        fn=generate_image,
-        inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
-        outputs=[output_image]
-    )
-
-
-
-    tryon_btn.click(
-        fn=generate_image,
-        inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
-        outputs=[output_image]
-    )
+                    
+            generate_button = gr.Button("Generate Try-On")
+            
+            gr.Examples(
+                examples=examples,
+                inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
+                outputs=output_image,
+                fn=generate_image,
+                cache_examples=False
+            )
+            
+            # Connect the button to the generation function
+            generate_button.click(
+                fn=generate_image,
+                inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
+                outputs=output_image
+            )
 
 if __name__ == "__main__":
-    demo.launch(share=True)  # Changed from app.launch to demo.launch
-    
+    demo.launch(share=True)