jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

discord: fix /tagme command to work with html page URLs.

Browse Source
This commit is contained in:
evazion
2022-11-15 15:20:30 -06:00
parent aa21f002be
commit 21a779455f
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
app/logical/discord_slash_command/tagme_command.rb
View File

@@ -31,7 +31,10 @@ class DiscordSlashCommand
end end
def tagme(url, confidence, limit: 50, size: 500) def tagme(url, confidence, limit: 50, size: 500)
response, file = http.download_media(url) extractor = Source::Extractor.find(url)
image_url = extractor.image_urls.first
file = extractor.download_file!(image_url)
preview = file.preview(size, size) preview = file.preview(size, size)
tags = autotagger.evaluate(preview, limit: limit, confidence: confidence).to_a tags = autotagger.evaluate(preview, limit: limit, confidence: confidence).to_a
tags = tags.sort_by { |tag, confidence| [TagCategory.split_header_list.index(tag.category_name.downcase), -confidence] }.to_h tags = tags.sort_by { |tag, confidence| [TagCategory.split_header_list.index(tag.category_name.downcase), -confidence] }.to_h
@@ -45,7 +48,7 @@ class DiscordSlashCommand
icon_url: "https://danbooru.donmai.us/images/danbooru-logo-96x96.png", icon_url: "https://danbooru.donmai.us/images/danbooru-logo-96x96.png",
}, },
image: { image: {
url: url, url: image_url,
}, },
}] }]
} }