Register category options in a custom position.

AUTOMATIC1111 · Oct 3, 2024 · 89cba88 · 89cba88
1 parent 82a973c
commit 89cba88
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/modules/options.py b/modules/options.py
@@ -326,11 +326,20 @@ class OptionsCategories:
     def __init__(self):
         self.mapping = {}
 
-    def register_category(self, category_id, label):
+    def register_category(self, category_id, label, position=None):
         if category_id in self.mapping:
             return category_id
 
-        self.mapping[category_id] = OptionsCategory(category_id, label)
+        # Use a dictionary to hold the categories temporarily
+        temp_mapping = self.mapping.copy()
 
+        # Insert the new category at the specified position
+        if position is not None and position == 0:
+            temp_mapping = {category_id: OptionsCategory(category_id, label), **temp_mapping}
+        else:
+            temp_mapping[category_id] = OptionsCategory(category_id, label)
+
+        # Update the mapping
+        self.mapping = temp_mapping
 
 categories = OptionsCategories()