tavo/tabeats
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
tavo-wasd 2024-03-31 12:10:43 -06:00
parent 551dec6c08
commit 1dfe60c01f
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
main.py
View file

@ -21,7 +21,8 @@ def find_song(query, songs_dir):
for file in files: for file in files:
if file.endswith(".mp3"): if file.endswith(".mp3"):
song_path = os.path.join(root, file) song_path = os.path.join(root, file)
song_name = os.path.relpath(song_path, start=songs_dir) #song_name = os.path.relpath(song_path, start=songs_dir) # full path doesnt return good results
song_name = os.path.splitext(file)[0]
confidence = process.extractOne(query, [song_name.replace("_", " ")])[1] confidence = process.extractOne(query, [song_name.replace("_", " ")])[1]
if confidence > best_confidence: if confidence > best_confidence: