Update app.py

app.py

@@ -352,16 +352,17 @@ def generate_image(prompt, structure_image, style_image, depth_strength=15, cann
         )
         
         try:
-            # Get the saved file path
-            saved_filename = saved_data["ui"]["images"][0]["filename"]
-            saved_subfolder = saved_data["ui"]["images"][0]["subfolder"]
-            output_dir = folder_paths.get_output_directory()
+            # # Get the saved file path
+            # saved_filename = saved_data["ui"]["images"][0]["filename"]
+            # saved_subfolder = saved_data["ui"]["images"][0]["subfolder"]
+            # output_dir = folder_paths.get_output_directory()
             
-            # Construct the full path
-            if saved_subfolder:
-                full_path = os.path.join(output_dir, saved_subfolder, saved_filename)
-            else:
-                full_path = os.path.join(output_dir, saved_filename)
+            # # Construct the full path
+            # if saved_subfolder:
+            #     full_path = os.path.join(output_dir, saved_subfolder, saved_filename)
+            # else:
+            #     full_path = os.path.join(output_dir, saved_filename)
+            full_path = f"output/{saved_data['ui']['images'][0]['filename']}"
                 
             return full_path
         except Exception as e:
@@ -428,16 +429,16 @@ with gr.Blocks(css="footer {visibility: hidden}") as app:
         with gr.Column(scale=1):
             output_image = gr.Image(label="Generated Image")
             
-    gr.Examples(
-        examples=[
-            ["A beautiful landscape with mountains and a lake", "examples/structure1.jpg", "examples/style1.jpg", 20, 0.4, 0.6, 30],
-            ["A cyberpunk cityscape at night", "examples/structure2.jpg", "examples/style2.jpg", 15, 0.35, 0.7, 28],
-        ],
-        inputs=[prompt_input, structure_image, style_image, depth_strength, canny_strength, style_strength, steps],
-        outputs=output_image,
-        fn=generate_image,
-        cache_examples=True
-    )
+    # gr.Examples(
+    #     examples=[
+    #         ["A beautiful landscape with mountains and a lake", "examples/structure1.jpg", "examples/style1.jpg", 20, 0.4, 0.6, 30],
+    #         ["A cyberpunk cityscape at night", "examples/structure2.jpg", "examples/style2.jpg", 15, 0.35, 0.7, 28],
+    #     ],
+    #     inputs=[prompt_input, structure_image, style_image, depth_strength, canny_strength, style_strength, steps],
+    #     outputs=output_image,
+    #     fn=generate_image,
+    #     cache_examples=True
+    # )
 
     generate_btn.click(
         fn=generate_image,