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

ai tags: add ability to order /ai_tags page by confidence.

Browse Source
This commit is contained in:
evazion
2022-06-25 21:28:38 -05:00
parent d3267a5ffa
commit 97ba21e615
2 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/models/ai_tag.rb
View File

@@ -20,7 +20,19 @@ class AITag < ApplicationRecord
q = q.where.missing(:post)
end
q = q.apply_default_order(params)
case params[:order]
when "score", "score_desc"
q = q.order(score: :desc, media_asset_id: :desc, tag_id: :desc)
when "score_asc"
q = q.order(score: :asc, media_asset_id: :asc, tag_id: :asc)
when "media_asset_id", "media_asset_id_desc"
q = q.order(media_asset_id: :desc, tag_id: :desc)
when "media_asset_id_asc"
q = q.order(media_asset_id: :asc, tag_id: :asc)
else
q = q.apply_default_order(params)
end
q
end

1
app/views/ai_tags/index.html.erb
View File

@@ -9,6 +9,7 @@
<%= f.input :post_tags_match, label: "Post Search", input_html: { value: params.dig(:search, :post_tags_match), data: { autocomplete: "tag-query" } } %>
<%= f.input :score, label: "Confidence", input_html: { value: params.dig(:search, :score) } %>
<%= f.input :is_posted, as: :hidden, input_html: { value: params.dig(:search, :is_posted) } %>
<%= f.input :order, collection: [%w[Newest media_asset_id], %w[Oldest media_asset_id_asc], %w[Confidence score_desc]], include_blank: true, selected: params[:search][:order] %>
<%= f.submit "Search" %>
<% end %>