Commit 03d7b394 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

added an option to filter out deepbooru tags

parent ccaacb18

modules/deepbooru.py

+3 −1
Original line number Diff line number Diff line
@@ -79,7 +79,9 @@ class DeepDanbooru: 


        res = []



        for tag in tags:

        filtertags = set([x.strip().replace(' ', '_') for x in shared.opts.deepbooru_filter_tags.split(",")])



        for tag in [x for x in tags if x not in filtertags]:

            probability = probability_dict[tag]

            tag_outformat = tag

            if use_spaces:

modules/shared.py

+1 −0
Original line number Diff line number Diff line
@@ -386,6 +386,7 @@ options_templates.update(options_section(('interrogate', "Interrogate Options"), 
    "deepbooru_sort_alpha": OptionInfo(True, "Interrogate: deepbooru sort alphabetically"),

    "deepbooru_use_spaces": OptionInfo(False, "use spaces for tags in deepbooru"),

    "deepbooru_escape": OptionInfo(True, "escape (\\) brackets in deepbooru (so they are used as literal brackets and not for emphasis)"),

    "deepbooru_filter_tags": OptionInfo("", "filter out those tags from deepbooru output (separated by comma)"),

}))



options_templates.update(options_section(('ui', "User interface"), {

CRA Git | Maintained and supported by SUSTech CRA and CCSE