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

test for full path and only song name

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

9
main.py
View file

@ -21,10 +21,15 @@ def find_song(query, songs_dir):
for file in files:
if file.endswith(".mp3"):
song_path = os.path.join(root, file)
#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]
full_song_name = os.path.relpath(song_path, start=songs_dir)
confidence = process.extractOne(query, [song_name.replace("_", " ")])[1]
if confidence > best_confidence:
best_match = song_path
best_confidence = confidence
confidence = process.extractOne(query, [full_song_name.replace("_", " ")])[1]
if confidence > best_confidence:
best_match = song_path
best_confidence = confidence