Fix Template Image selection mismatch

utils/constants.py

@@ -240,6 +240,9 @@ LORA_DETAILS = {
             "parameters" :{
                 "num_inference_steps": "30"
             }
+        },
+        {
+            "thumbnail": ""
         }
         ],
     "XLabs-AI/flux-lora-collection/anime_lora.safetensors":[
@@ -257,6 +260,9 @@ LORA_DETAILS = {
             "parameters" :{
                 "num_inference_steps": "30"
             }
+        },
+        {
+            "thumbnail": ""
         }
         ],
     "XLabs-AI/flux-lora-collection/scenery_lora.safetensors":[
@@ -274,6 +280,9 @@ LORA_DETAILS = {
             "parameters" :{
                 "num_inference_steps": "30"
             }
+        },
+        {
+            "thumbnail": ""
         }
         ],
     "XLabs-AI/flux-lora-collection/disney_lora.safetensors":[
@@ -291,6 +300,9 @@ LORA_DETAILS = {
             "parameters" :{
                 "num_inference_steps": "30"
             }
+        },
+        {
+            "thumbnail": ""
         }
         ],
     "XLabs-AI/flux-RealismLora":[
@@ -303,6 +315,9 @@ LORA_DETAILS = {
                 "num_inference_steps": "34",
                 "scale": "0.85"
             }
+        },
+        {
+            "thumbnail": ""
         }
         ],
     "Cossale/Frames2-Flex.1": [
@@ -320,6 +335,9 @@ LORA_DETAILS = {
             "parameters" :{
                 "num_inference_steps": "50"
             }
+        },
+        {
+            "thumbnail": ""
         }
     ],
     "Yuanshi/OminiControl": [
@@ -339,6 +357,9 @@ LORA_DETAILS = {
         },
         {
             "notes": "Select an Image as a guide."
+        },
+        {
+            "thumbnail": ""
         }
     ],
     "Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design": [
@@ -357,6 +378,9 @@ LORA_DETAILS = {
             "parameters" :{
                 "num_inference_steps": "38"
             }
+        },
+        {
+            "thumbnail": ""
         }
     ],
     "Borcherding/FLUX.1-dev-LoRA-FractalLand-v0.1": [
@@ -372,6 +396,9 @@ LORA_DETAILS = {
                 "guidance_scale": "3.5",
                 "num_inference_steps": "38"
             }
+        },
+        {
+            "thumbnail": ""
         }
     ]
 }

utils/image_utils.py

@@ -100,21 +100,34 @@ def build_prerendered_images(images_list):
 # Example usage
 # filtered_maps = get_maps_with_quality_less_than(3)
 # print(filtered_maps)
-def build_prerendered_images_by_quality(quality_limit,key='file'):
+def build_prerendered_images_by_quality(quality_limit, key='file'):
     """
-    Retrieve all file paths from PRE_RENDERED_MAPS_JSON_LEVELS where quality is less than the given limit.
-
+    Retrieve and sort file paths from PRE_RENDERED_MAPS_JSON_LEVELS where quality is less than or equal to the given limit.
+    The sorting order matches pre_rendered_maps_paths based on quality and a case-insensitive alphanumeric key.
+    
     Args:
-        limit (int): The quality threshold.
-
+        quality_limit (int): The quality threshold.
+        key (str): The key to extract the file path from each map info (default is 'file').
+    
     Returns:
-        list: A list of file paths meeting the quality criteria.
-    """
+        list: A list of sorted file paths meeting the quality criteria.
+    """
+    # Sort the PRE_RENDERED_MAPS_JSON_LEVELS items by quality and alphanumeric key
+    sorted_maps = sorted(
+        PRE_RENDERED_MAPS_JSON_LEVELS.items(),
+        key=lambda x: (
+            x[1]['quality'],
+            ''.join(char.lower() for char in x[0] if char.isalnum())
+        )
+    )
+    
+    # Filter and extract the file paths that meet the quality limit
     images_list = [
-        map_info[key]
-        for map_info in PRE_RENDERED_MAPS_JSON_LEVELS.values()
+        map_info[key].replace("\\", "/")
+        for _, map_info in sorted_maps
         if map_info['quality'] <= quality_limit
     ]
+    
     return build_prerendered_images(images_list)