Update app.py

app.py

@@ -352,17 +352,16 @@ 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)
-            full_path = f"output/{saved_data['ui']['images'][0]['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)
                 
             return full_path
         except Exception as e: