translate_article/example_usage.py at main · stanislavlyalin/translate_article · 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
# coding: utf-8
import webbrowser

from translator_py import Translator

if __name__ == '__main__':

    email = input('Enter email: ')
    password = input('Enter password: ')

    translator = Translator()

    # log in service
    logged_in = translator.login(email, password)

    if logged_in:
        known, unknown = [], []
        url = 'https://en.wikipedia.org/wiki/Blockchain'

        # get article tokens never seen before
        tokens = translator.tokens(url)

        # mark tokens as known/unknown
        for word in tokens:
            ans = input(f"Do you know word '{word}'?: ").lower()

            if ans == 'stop':
                break
            elif ans == 'y':
                known.append(word)
            elif ans == 'n':
                unknown.append(word)

        # update dictionaries
        translator.append_known(known)
        translator.append_unknown(unknown)

        # translate page
        page = translator.translate(transcriptions=True,
                                    translate_unlabeled=False)

        # save page to disk and open in browser
        translated_filepath = 'page.html'
        with open(translated_filepath, encoding='utf-8', mode='w') as f:
            f.write(page)
            webbrowser.open(translated_filepath)

        # mark learned words as known
        print('Type words what you know')
        while True:
            ans = input('I know word: ')
            if ans == 'stop':
                break
            translator.append_known([ans])

        # get anki file
        translator.anki('anki.txt')