Deduplicate gallery images by base name to avoid duplicate extensions

ui/layout.py

@@ -15,16 +15,25 @@ def build_ui(event_handler_function):
     # times (e.g., due to accidental duplicate saves), the Gallery tab will
     # only display each image once.
     def load_gallery_images():
+        """Return a deduplicated list of image paths.
+        If multiple files share the same stem (e.g., ``img.png`` and ``img.webp``),
+        only the first encountered file (based on sorted order) is kept. This prevents
+        the Gallery tab from displaying duplicate visual entries that arise from
+        different extensions of the same generated image.
+        """
         output_dir = os.path.abspath(OUTPUT_DIR)
         if not os.path.isdir(output_dir):
             return []
-        # Use a set to collect unique absolute paths.
-        image_path_set = set()
+        seen_bases = set()
+        image_paths = []
         for fname in sorted(os.listdir(output_dir)):
-            if fname.lower().endswith((".png", ".jpg", ".jpeg", ".gif", ".webp")):
-                image_path_set.add(os.path.join(output_dir, fname))
-        # Convert back to a sorted list for deterministic ordering.
-        return sorted(image_path_set)
+            low = fname.lower()
+            if low.endswith((".png", ".jpg", ".jpeg", ".gif", ".webp")):
+                base = os.path.splitext(fname)[0]
+                if base not in seen_bases:
+                    seen_bases.add(base)
+                    image_paths.append(os.path.join(output_dir, fname))
+        return image_paths
 
     # All UI components, including tabs, must be created inside the same Gradio Blocks context.
     with gr.Blocks() as demo: