[#143] Double check lang detection only for fr

akvo · Dec 10, 2024 · 52834ae · 52834ae
1 parent 863658c
commit 52834ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/assistant/assistant.py b/assistant/assistant.py
@@ -273,8 +273,8 @@ def get_language(user_prompt) -> str:
     except LangDetectException:
         return "en"
 
-    # Double check to make sure not en
-    if detected_language != "en":
+    # Double check with nltk library
+    if detected_language == "fr":
         # Tokenize the text to get individual words
         tokens = nltk.word_tokenize(user_prompt.lower())