diff --git a/main.py b/main.py index 4910fa5..bf9e985 100644 --- a/main.py +++ b/main.py @@ -22,7 +22,7 @@ def find_song(query, songs_dir): song_path = os.path.join(root, file) #song_name = os.path.relpath(song_path, start=songs_dir) song_name = os.path.splitext(file)[0] - confidence = process.extractOne(query, [song_name])[1] + confidence = process.extractOne(query, [song_name.replace("_", " ")])[1] if confidence > best_confidence: best_match = song_path