Update app.py

app.py

@@ -352,18 +352,9 @@ 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()
-            
-            # 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)
+            saved_path = f"output/{saved_data['ui']['images'][0]['filename']}"
                 
-            return full_path
+            return saved_path
         except Exception as e:
             print(f"Error getting saved image path: {e}")
             # Fall back to the expected path