diff --git a/javascript/tagAutocomplete.js b/javascript/tagAutocomplete.js index 50c11f2..1d8fdef 100644 --- a/javascript/tagAutocomplete.js +++ b/javascript/tagAutocomplete.js @@ -753,7 +753,14 @@ function addResultsToList(textArea, results, tagword, resetList) { // Set link based on selected file let tagFileNameLower = tagFileName.toLowerCase(); - if (tagFileNameLower.startsWith("danbooru")) { + if (tagFileNameLower.startsWith("danbooru_e621_merged")) { + // Use danbooru for categories 0-5, e621 for 6+ + // Based on the merged categories from https://github.com/DraconicDragon/dbr-e621-lists-archive/tree/main/tag-lists/danbooru_e621_merged + // Danbooru is also the fallback if result.category is not set + wikiLink.href = result.category && result.category >= 6 + ? `https://e621.net/wiki_pages/${linkPart}` + : `https://danbooru.donmai.us/wiki_pages/${linkPart}`; + } else if (tagFileNameLower.startsWith("danbooru")) { wikiLink.href = `https://danbooru.donmai.us/wiki_pages/${linkPart}`; } else if (tagFileNameLower.startsWith("e621")) { wikiLink.href = `https://e621.net/wiki_pages/${linkPart}`; diff --git a/scripts/tag_autocomplete_helper.py b/scripts/tag_autocomplete_helper.py index c105de6..438432b 100644 --- a/scripts/tag_autocomplete_helper.py +++ b/scripts/tag_autocomplete_helper.py @@ -686,6 +686,23 @@ def on_ui_settings(): "9": ["#df3647", "#8e1c2b"], "10": ["#c98f2b", "#7b470e"], "11": ["#e87ebe", "#a83583"] + }, + "danbooru_e621_merged": { + "-1": ["red", "maroon"], + "0": ["lightblue", "dodgerblue"], + "1": ["indianred", "firebrick"], + "3": ["violet", "darkorchid"], + "4": ["lightgreen", "darkgreen"], + "5": ["orange", "darkorange"], + "6": ["red", "maroon"], + "7": ["lightblue", "dodgerblue"], + "8": ["gold", "goldenrod"], + "9": ["gold", "goldenrod"], + "10": ["violet", "darkorchid"], + "11": ["lightgreen", "darkgreen"], + "12": ["tomato", "darksalmon"], + "14": ["whitesmoke", "black"], + "15": ["seagreen", "darkseagreen"] } }\ """ @@ -883,5 +900,5 @@ def api_tac(_: gr.Blocks, app: FastAPI): @app.get("/tacapi/v1/get-all-use-counts") async def get_all_tag_counts(): return db_request(lambda: db.get_all_tags(), get=True) - + script_callbacks.on_app_started(api_tac)