polysem/polysem.py at master · alex-rusakevich/polysem · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#!/usr/bin/env python3
import sys

import colorama
from colorama import Back, Fore, Style

from polysem.logics import *
from polysem.settings import *

MIN_PYTHON = (3, 8)

if sys.version_info < MIN_PYTHON:
    sys.exit("Python %s.%s or later is required.\n" % MIN_PYTHON)


colorama.init(autoreset=True)


def main():
    print(
        "The program was created by Alexander Rusakevich, 402/1, Chinese faculty (https://github.com/alex-rusakevich/polysem)"
    )
    the_yellow_word = Fore.YELLOW + "'" + THE_WORD + "'" + Style.RESET_ALL
    print(
        f"The core word is {the_yellow_word}. The base has {Fore.GREEN}{len(MEANINGS)}{Style.RESET_ALL} meanings and \
{Fore.GREEN}{len(MEANING_SINGS)}{Style.RESET_ALL} meaning signs"
    )

    if len(sys.argv) == 1:  # REPL
        print("Press Ctrl-Z or Ctrl-C to stop")

        while True:
            stem_seqs = text_to_stem_seqs(
                input(Fore.YELLOW + "\nText: " + Style.RESET_ALL)
            )

            if stem_seqs == []:
                print(Fore.RED + f"No sentence with word '{THE_WORD}', skipping...")
                continue

            for orig_sent, sent_stems in stem_seqs:
                print(Fore.YELLOW + f"\nProcessing sentence '{orig_sent}'...")

                meanings = get_seq_meanings(stems_to_meaning_seq(sent_stems))

                if not meanings:
                    print(
                        Fore.RED
                        + "Couldn't estimate the meaning, please, try another sentence"
                    )
                    continue
                else:
                    for i, (meaning, score) in enumerate(meanings):
                        print(
                            Fore.GREEN
                            + "\n-> "
                            + Style.RESET_ALL
                            + "Meaning with score "
                            + Fore.GREEN
                            + str(score)
                            + Style.RESET_ALL
                            + (
                                ""
                                if i > 0
                                else Fore.YELLOW + " (best)" + Style.RESET_ALL
                            )
                            + ":",
                            meaning.text,
                        )
                        print(
                            Fore.GREEN + "?>" + Style.RESET_ALL,
                            "Example:",
                            meaning.example,
                        )

            print(Fore.GREEN + "\nDone.")

    else:
        raise NotImplementedError("Command-line args are not supported yet")


if __name__ == "__main__":
    main()