ai tags: save ai tags on upload.

Save the AI tags when a media asset is uploaded.

app/models/ai_tag.rb

@@ -1,11 +1,13 @@
 # frozen_string_literal: true
 
 class AITag < ApplicationRecord
+  self.primary_keys = :media_asset_id, :tag_id
+
   belongs_to :tag
   belongs_to :media_asset
   has_one :post, through: :media_asset
 
-  validates :score, inclusion: { in: (0.0..1.0) }
+  validates :score, inclusion: { in: (0..100) }
 
   def self.named(name)
     name = $1.downcase if name =~ /\A(rating:.)/i

app/models/media_asset.rb

@@ -27,6 +27,7 @@ class MediaAsset < ApplicationRecord
 
   scope :public_only, -> { where(is_public: true) }
   scope :private_only, -> { where(is_public: false) }
+  scope :without_ai_tags, -> { where.not(AITag.where("ai_tags.media_asset_id = media_assets.id").select(1).arel.exists) }
 
   # Processing: The asset's files are currently being resized and distributed to the backend servers.
   # Active: The asset has been successfully uploaded and is ready to use.
@@ -279,6 +280,14 @@ class MediaAsset < ApplicationRecord
       self.duration = media_file.duration
       self.media_metadata = MediaMetadata.new(file: media_file)
       self.pixiv_ugoira_frame_data = PixivUgoiraFrameData.new(data: media_file.frame_data, content_type: "image/jpeg") if is_ugoira?
+      self.ai_tags = media_file.preview(360, 360).ai_tags # XXX should do this in parallel with thumbnail generation.
+    end
+
+    def regenerate_ai_tags!
+      with_lock do
+        ai_tags.each(&:destroy!)
+        update!(ai_tags: variant(:"360x360").open_file.ai_tags)
+      end
     end
 
     def expunge!