Fix: unintended removal of radio vocabs from query

OpenVoiceOS · Jun 19, 2024 · ff1c70b · ff1c70b
1 parent 93b03bb
commit ff1c70b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/__init__.py b/__init__.py
@@ -114,16 +114,22 @@ def search_pyradios(self, phrase: str, media_type: MediaType) -> Generator[dict,
         else:
             base_score -= 30
 
+        queries = []
         for voc_filename in ["pyradios", "radio"]:
             if self.voc_match(phrase, voc_filename):
                 base_score += 50  # explicit request
                 phrase = self.remove_voc(phrase, voc_filename).strip()
+            if phrase not in queries:
+                # Save queries with and without radio vocabulary
+                queries.append(phrase)
 
-        queries = []
+        # if "radio" still in phrase, also add the query without it
         if "radio" in phrase:
-            phrase_without_radio = ' '.join(phrase.replace('radio', '').split())
+            phrase_without_radio = ' '.join(
+                phrase.replace('radio', '').split()
+            )
             queries.append(phrase_without_radio)
-        queries.append(phrase)
+
         for query in queries:
             for ch in self.search(query=query):
                 score = base_score + int(DamerauLevenshtein.normalized_similarity(ch["name"], query) * 100)

diff --git a/version.py b/version.py
@@ -2,5 +2,5 @@
 VERSION_MAJOR = 0
 VERSION_MINOR = 0
 VERSION_BUILD = 0
-VERSION_ALPHA = 4
+VERSION_ALPHA = 5
 # END_VERSION_BLOCK