import nltk
import os


def main():
    nltk.download("wordnet")

    adjectives = {
        lemma.name().capitalize()
        for synset in nltk.corpus.wordnet.all_synsets(pos="a")
        for lemma in synset.lemmas()
        if lemma.name().isalpha()
    }
    nouns = {
        lemma.name().capitalize()
        for synset in nltk.corpus.wordnet.all_synsets(pos="n")
        for lemma in synset.lemmas()
        if lemma.name().isalpha()
    }

    os.makedirs("data", exist_ok=True)
    with open("data/adjectives.txt", "w") as adj_file:
        adj_file.write("\n".join(sorted(adjectives)))
    with open("data/nouns.txt", "w") as noun_file:
        noun_file.write("\n".join(sorted(nouns)))


if __name__ == "__main__":
    main()