feat: add creator search option and update related functionality across models and UI

py/routes/base_model_routes.py

@@ -177,6 +177,7 @@ class BaseModelRoutes(ABC):
             'filename': request.query.get('search_filename', 'true').lower() == 'true',
             'modelname': request.query.get('search_modelname', 'true').lower() == 'true',
             'tags': request.query.get('search_tags', 'false').lower() == 'true',
+            'creator': request.query.get('search_creator', 'false').lower() == 'true',
             'recursive': request.query.get('recursive', 'false').lower() == 'true',
         }
         

py/services/base_model_service.py

@@ -199,6 +199,22 @@ class BaseModelService(ABC):
                        for tag in item['tags']):
                     search_results.append(item)
                     continue
+            
+            # Search by creator
+            civitai = item.get('civitai')
+            creator_username = ''
+            if civitai and isinstance(civitai, dict):
+                creator = civitai.get('creator')
+                if creator and isinstance(creator, dict):
+                    creator_username = creator.get('username', '')
+            if search_options.get('creator', False) and creator_username:
+                if fuzzy_search:
+                    if fuzzy_match(creator_username, search):
+                        search_results.append(item)
+                        continue
+                elif search.lower() in creator_username.lower():
+                    search_results.append(item)
+                    continue
         
         return search_results
     

py/utils/constants.py

@@ -50,7 +50,8 @@ VALID_LORA_TYPES = ['lora', 'locon', 'dora']
 
 # Civitai model tags in priority order for subfolder organization
 CIVITAI_MODEL_TAGS = [
-    'character', 'style', 'concept', 'clothing', 'base model', 
+    'character', 'style', 'concept', 'clothing', 
+    # 'base model', # exclude 'base model'
     'poses', 'background', 'tool', 'vehicle', 'buildings', 
     'objects', 'assets', 'animal', 'action'
 ]

py/utils/utils.py

@@ -47,7 +47,7 @@ def get_lora_info(lora_name):
         # No event loop is running, we can use asyncio.run()
         return asyncio.run(_get_lora_info_async())
 
-def fuzzy_match(text: str, pattern: str, threshold: float = 0.7) -> bool:
+def fuzzy_match(text: str, pattern: str, threshold: float = 0.85) -> bool:
         """
         Check if text matches pattern using fuzzy matching.
         Returns True if similarity ratio is above threshold.

static/js/api/baseModelApi.js

@@ -628,6 +628,9 @@ export class BaseModelApiClient {
                 if (pageState.searchOptions.tags !== undefined) {
                     params.append('search_tags', pageState.searchOptions.tags.toString());
                 }
+                if (pageState.searchOptions.creator !== undefined) {
+                    params.append('search_creator', pageState.searchOptions.creator.toString());
+                }
                 params.append('recursive', (pageState.searchOptions?.recursive ?? false).toString());
             }
         }

static/js/managers/SearchManager.js

@@ -318,6 +318,7 @@ export class SearchManager {
             filename: options.filename || false,
             modelname: options.modelname || false,
             tags: options.tags || false,
+            creator: options.creator || false,
             recursive: recursive
           };
         } else if (this.currentPage === 'checkpoints') {
@@ -325,6 +326,7 @@ export class SearchManager {
             filename: options.filename || false,
             modelname: options.modelname || false,
             tags: options.tags || false,
+            creator: options.creator || false,
             recursive: recursive
           };
         }

static/js/state/index.js

@@ -37,6 +37,7 @@ export const state = {
                 filename: true,
                 modelname: true,
                 tags: false,
+                creator: false,
                 recursive: false
             },
             filters: {
@@ -83,6 +84,7 @@ export const state = {
             searchOptions: {
                 filename: true,
                 modelname: true,
+                creator: false,
                 recursive: false
             },
             filters: {
@@ -110,6 +112,7 @@ export const state = {
                 filename: true,
                 modelname: true,
                 tags: false,
+                creator: false,
                 recursive: false
             },
             filters: {

templates/components/header.html

@@ -86,15 +86,18 @@
       <div class="search-option-tag active" data-option="filename">Filename</div>
       <div class="search-option-tag active" data-option="modelname">Checkpoint Name</div>
       <div class="search-option-tag active" data-option="tags">Tags</div>
+      <div class="search-option-tag" data-option="creator">Creator</div>
       {% elif request.path == '/embeddings' %}
       <div class="search-option-tag active" data-option="filename">Filename</div>
       <div class="search-option-tag active" data-option="modelname">Embedding Name</div>
       <div class="search-option-tag active" data-option="tags">Tags</div>
+      <div class="search-option-tag" data-option="creator">Creator</div>
       {% else %}
       <!-- Default options for LoRAs page -->
       <div class="search-option-tag active" data-option="filename">Filename</div>
       <div class="search-option-tag active" data-option="modelname">Model Name</div>
       <div class="search-option-tag active" data-option="tags">Tags</div>
+      <div class="search-option-tag" data-option="creator">Creator</div>
       {% endif %}
     </div>
   </div>