Initital commit

Author: mepavv

README.md

@@ -0,0 +1,18 @@
+# anki_ENG-PL
+### Skrypt CLI pozwalający na automatyczne tworzenie fiszek z tłumaczeniami wyrazów, przykładami zdań i wymową, w formie transkrypcji fonetycznej i TTS.
+\
+Do tłumaczenia wyrazów jest używany słownik [diki.pl](https://diki.pl) i [Bab.la](https://bab.la).\
+Przykłady zdań są wzięte ze słownika [Bab.la](https://bab.la).\
+A wymowa zostaje wygenerowana przez amerykański Google TTS.
+
+## Obsługa
+```
+pip install requirements.txt
+```
+
+Przygotuj dokument tekstowy z wyrazami do przetłumaczenia (każdy w kolejnej linii), a następnie uruchom skrypt:
+
+```
+python anki.py words.txt
+```
+

anki.py

@@ -0,0 +1,167 @@
+from diki_translate import Diki
+from babla_translate import Babla
+from gtts import gTTS
+import genanki
+import pandas as pd
+import eng_to_ipa as p
+import contextlib
+import random
+import datetime
+import argparse
+import itertools
+import os
+import shutil
+
+
+#adds note to deck
+def noteAdder(note_word, phonetic, phonetic_media, meaning, example):
+  with contextlib.suppress(TypeError):
+    my_note = genanki.Note(
+    model=my_model,
+    fields=[note_word, phonetic, "[sound:"+ phonetic_media[6:] +"]", meaning, example])
+    my_deck.add_note(my_note)
+  
+  
+#generates tts from word
+def ttsAdder(word):
+  with contextlib.suppress(AssertionError):
+    tts = gTTS(word, lang='en')
+    tts_word = f"media/{str(word_index)}_" + word[:3] + ".mp3"
+    tts.save(tts_word)
+    my_package.media_files.append(tts_word)
+    return tts_word
+  
+  
+#generates meaning, example and phonetic of word
+def meaningAdder(word):
+  diki_list = list(itertools.islice(diki.translation(word), 5))
+  babla_list = list(itertools.islice(babla.translation(word), 5))
+
+
+  num = [len(diki_list), len(babla_list)]
+  meaning_index = max(num)
+  trans_list = []
+
+  for i in range(meaning_index):
+    with contextlib.suppress(IndexError):
+        trans_list.append(diki_list[i])    
+
+    with contextlib.suppress(IndexError):
+        trans_list.append(babla_list[i])    
+
+
+  uniqueList = []
+  duplicateList = []
+  
+  for i in trans_list:
+      if i not in uniqueList:
+          uniqueList.append(i)
+      elif i not in duplicateList:
+          duplicateList.append(i)  
+
+
+  if uniqueList:
+    meaning_string = ', '.join(uniqueList[:5])
+  else:
+    meaning_string = " "
+
+
+  dataset.at[word_index,'meaning'] = meaning_string
+  dataset.at[word_index,'phonetic'] = p.convert(word)
+
+
+  example = list(itertools.islice(babla.example(word), 1))
+  with contextlib.suppress(IndexError):
+    if example:
+      dataset.at[word_index,'example'] = example[0]
+    else:
+      example = " "
+      dataset.at[word_index,'example'] = example
+
+
+  print(dataset.at[word_index,'phonetic'])
+  print(dataset.at[word_index,'meaning'])
+  print(dataset.at[word_index,'example'])
+
+
+
+#cli and pd init
+parser = argparse.ArgumentParser(description='English <-> Polish anki generator')
+parser.add_argument('file')
+args = parser.parse_args()
+columns = ['word_column', 'phonetic', 'meaning', 'example']
+dataset=pd.read_csv(args.file,header=None, index_col=False,names=columns,sep='\t',dtype=object)
+
+
+# diki and babla module init
+diki = Diki("english")
+babla = Babla("english", "polish")
+
+
+#genanki card model 
+my_model = genanki.Model(
+  random.randrange(1 << 30, 1 << 31),
+  'Translation_model',
+  fields=[
+    {'name': 'word'},
+    {'name': 'phonetic'},
+    {'name': 'phonetic_media'},
+    {'name': 'meaning'},
+    {'name': 'example'},
+  ],
+  templates=[
+    {
+      'name': 'Card 1',
+      'qfmt': '{{word}} <br> {{phonetic}} \t {{phonetic_media}}',
+
+      'afmt': '{{FrontSide}}<hr id="answer">{{meaning}} <br><br> {{example}}',
+    },
+  ],
+  css='.card {\n font-family: arial;\n font-size: 20px;\n text-align: center;\n color: black;\n background-color: white;\n}\n',
+  )
+
+#creates media folder to store tts and checks if it exist
+try:
+  folder_path = 'media'
+  os.mkdir(folder_path)
+  print('Media folder created')
+except Exception:
+  print('Media folder already exists')
+  
+  
+current_day = datetime.datetime.now()
+title = f'Translation_{str(current_day.strftime("%d.%m"))}'
+
+
+my_deck = genanki.Deck(random.randrange(1 << 30, 1 << 31),title)
+my_package = genanki.Package(my_deck)
+my_package.media_files = []
+not_translated = []
+
+
+#main loop
+for word_index in range(len(dataset)):
+    current_word = dataset._get_value(word_index,'word_column')
+    print(word_index, current_word)
+    meaningAdder(current_word)
+
+    if dataset.at[word_index,'meaning'] == " ":
+      not_translated.append(current_word)
+      dataset = dataset.drop(word_index)
+    else:
+      noteAdder(dataset.at[word_index,'word_column'],dataset.at[word_index,'phonetic'], ttsAdder(current_word),dataset.at[word_index,'meaning'], dataset.at[word_index,'example'])
+
+
+#export
+print(dataset)
+dataset.to_csv(title + '.csv',index=False)
+my_package.write_to_file(title + '.apkg')
+
+
+#removes media folder
+try:
+  shutil.rmtree(folder_path)
+  print('Media folder and its content removed')
+except Exception:
+  print('Media folder not deleted')
+print(f"Words not translated : {not_translated}")

requirements.txt

@@ -0,0 +1,6 @@
+diki_translate
+babla_translate
+gtts
+genanki
+pandas
+eng_to_ipa