[AI Tagger] 2.0 small fixes

plugins/AITagger/ai_tagger.py

@@ -40,6 +40,12 @@ def install(package):
     except ModuleNotFoundError:
         install('pydantic')
         toRaise = True
+
+    try:
+        import cv2
+    except ModuleNotFoundError:
+        install('opencv-python')
+        toRaise = True
 
     if toRaise:
         raise Exception("Installed required packages, please retry the task.")
@@ -53,11 +59,6 @@ def install(package):
     import ai_server
     import utility
     from datetime import datetime
-    try:
-        import cv2
-    except ModuleNotFoundError:
-        install('opencv-python')
-        toRaise = True
 except:
     log.error("Attempted to install required packages, please retry the task.")
     log.error(f"Stack trace {traceback.format_exc()}")
@@ -238,7 +239,7 @@ async def __tag_images(images):
                     else:
                         tags_list = []
                         for _, tags in result.items():
-                            stashtag_ids = media_handler.get_tag_ids(tags)
+                            stashtag_ids = media_handler.get_tag_ids(tags, create=True)
                             stashtag_ids.append(media_handler.ai_tagged_tag_id)
                             tags_list.extend(stashtag_ids)
                         media_handler.add_tags_to_image(id, tags_list)
@@ -375,4 +376,4 @@ def increment_progress():
     global increment
     progress += increment
     log.progress(progress)
-asyncio.run(main())
+asyncio.run(main())

plugins/AITagger/ai_tagger.yml

@@ -1,6 +1,6 @@
 name: AI Tagger
 description: Tag videos and Images with Locally hosted AI using Skier's Free and Patreon AI models
-version: 2.0
+version: 2.1
 url: https://github.com/stashapp/CommunityScripts/tree/main/plugins/AITagger
 exec:
   - python