diff --git a/plugins/AITagger/ai_tagger.py b/plugins/AITagger/ai_tagger.py index 0b097c4..918e0ce 100644 --- a/plugins/AITagger/ai_tagger.py +++ b/plugins/AITagger/ai_tagger.py @@ -182,6 +182,9 @@ async def __tag_images(images): async def __tag_scene(scene): async with semaphore: scenePath = scene['files'][0]['path'] + mutated_path = scenePath + for key, value in config.path_mutation.items(): + mutated_path = mutated_path.replace(key, value) sceneId = scene['id'] log.debug("files result:" + str(scene['files'][0])) phash = scene['files'][0].get('fingerprint', None) diff --git a/plugins/AITagger/ai_tagger.yml b/plugins/AITagger/ai_tagger.yml index d2c37e7..34166da 100644 --- a/plugins/AITagger/ai_tagger.yml +++ b/plugins/AITagger/ai_tagger.yml @@ -1,6 +1,6 @@ name: AI Tagger description: Tag videos and Images with Locally hosted AI using Skier's Patreon AI models -version: 1.5 +version: 1.6 url: https://github.com/stashapp/CommunityScripts/tree/main/plugins/AITagger exec: - python