requirements.txt

plyer >= 2.1.0  # for notification, can help to install dbus-python too
pyfiglet >= 1.0.2   # for banner
prompt_toolkit >= 3.0.43  # for autocomplete prompt
rapidfuzz >= 3.6.1  # to correct for field mapping and note type spelling errors
beepy>=1.0.7  # make a beeping sound
ankipandas>=0.3.15  # to load the anki collection
py_ankiconnect  # DIY to talk to ankiconnect
ftfy>=6.1.1
fire>=0.5.0
platformdirs >= 4.2.2

# datascience and computation
pandas == 1.5.3
joblib >= 1.3.2
umap-learn >= 0.5.5

numpy == 1.23.5
scikit-learn>=1.4.0
tqdm >= 4.65.0

nltk >= 3.8.1
tokenizers >= 0.13.2
transformers >= 4.37.2
sentence_transformers >= 2.7.0
flash_attn  # can speed up some models

# 2D plot
matplotlib>=3.7.1
networkx>=3.0.0
pydot>=1.4.2

bertopic>=0.15.0
hdbscan>=0.8.29
plotly>=5.13.1