augustin64/cemantix-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Import readline

Browse Source
This commit is contained in:
augustin64 2024-10-11 16:06:44 +02:00
parent 035bc05637
commit d2a76adde6
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cemantix.py
View File

@ -1,18 +1,20 @@
from gensim.models import KeyedVectors
from colorama import Fore, Back, Style
import argparse
import readline
import random
import time
def cemantix(model, word=None):
if word is None:
rd = random.randint(0, len(model))
word = model.index_to_key[rd]
while "_" in word:
while '_' in word or '-' in word:
rd += 1
word = model.index_to_key[rd]
nearest = [i[0] for i in model.most_similar(word, topn=1000)]
nearest = [word]+[i[0] for i in model.most_similar(word, topn=999)]
def get_rank(guess):
if guess not in nearest:
return None
@ -40,6 +42,7 @@ def cemantix(model, word=None):
while True:
try:
guess = input(Style.BRIGHT+"Your guess > "+Style.RESET_ALL).strip()
readline.add_history(guess)
except (EOFError, KeyboardInterrupt):
print("The word was "+Style.BRIGHT+word+Style.RESET_ALL)
print("Goodbye!")
@ -56,6 +59,7 @@ def cemantix(model, word=None):
print(chr(27) + "[2J")
print(formatted_status(guesses, last=guess))
if guess == word:
time.sleep(1)
print(Fore.GREEN+"Correct!"+Style.RESET_ALL+f" {len(guesses)} tries.")
return len(guesses)