keywords

app.py

@@ -329,13 +329,10 @@ def get_ai_response(text: str) -> str:
 HAUSA_WORDS = [
     "aikin", "manoma", "gona", "amfanin", "yanayi", "tsaba", "fasaha", "bisa", "noman", "shuka",
     "daji", "rani", "damina", "amfani", "bidi'a", "noma", "bashi", "manure", "tsiro", "gishiri",
-    "da", "shi", "ta", "su", "mu", "ku", "ni", "kai", "ita", "ina", "yana", "tana", "suna", "muna", "kuna", "na", "ka", "sa",
-    "wani", "wata", "wasu", "wadansu", "wadannan", "wannan", "wancan",
-    "kamar", "kusa", "nisa", "gaba", "baya", "hagu", "dama", "sama", "kasa", "lokaci",
-    "zan", "tsidaya", "jiya", "lafi", "siyo", "ba", "ce", "ya", "wani", "yafi",
-    "kowa", "kome", "koyaushe", "koyaushe", "koyaushe", "koyaushe", "koyaushe", "koyaushe",
-    "saboda", "domin", "don", "saboda", "domin", "don", "saboda", "domin", "don",
-    "kuma", "har", "kadan", "sosai", "daidai", "kwata", "kwata", "kwata"
+    "gonaki", "gonar", "tsirrai", "kayan", "gonar", "tsirrai", "kayan", "gonar",
+    "tsirrai", "kayan", "gonar", "tsirrai", "kayan", "gonar", "tsirrai", "kayan",
+    "gonar", "tsirrai", "kayan", "gonar", "tsirrai", "kayan", "gonar", "tsirrai",
+    "kayan", "gonar", "tsirrai", "kayan", "gonar", "tsirrai", "kayan", "gonar"
 ]
 
 YORUBA_WORDS = [
@@ -353,14 +350,18 @@ YORUBA_WORDS = [
 
 IGBO_WORDS = [
     "ugbo", "akụkọ", "mmiri", "ala", "ọrụ", "ncheta", "ọhụrụ", "ugwu", "nri", "ahụhụ",
-    "n'", "maka", "n'ihi", "n'ime", "n'elu", "n'okpuru",
-    "ya", "anyị", "unu", "ha",
-    "otu", "ọtụtụ", "ebe", "oge",
-    "ụ", "ọ", "ị", "ụ", "ọ", "ị", "ụ", "ọ", "ị", "ụ", "ọ", "ị",
-    "bụ", "nọ", "ga", "na", "na", "na", "na", "na", "na", "na", "na",
-    "n'ihi", "n'ihi", "n'ihi", "n'ihi", "n'ihi", "n'ihi", "n'ihi", "n'ihi",
-    "ma", "ọ", "bụ", "na", "ga", "nọ", "ma", "ọ", "bụ", "na", "ga", "nọ",
-    "dị", "ka", "n'ime", "n'elu", "n'okpuru", "n'ihi", "maka", "n'ihi", "maka"
+    "ọkụkọ", "ewu", "atụrụ", "ehi", "azụ", "osisi", "mkpụrụ", "ubi", "ọka", "ji",
+    "akwụkwọ", "ofe", "azu", "anụ", "nnu", "mmanụ", "ngwọ", "ọgwụ", "ahịhịa", "osisi",
+    "n'", "maka", "n'ihi", "n'ime", "n'elu", "n'okpuru", "ya", "anyị", "unu", "ha",
+    "otu", "ọtụtụ", "ebe", "oge", "ụ", "ọ", "ị", "bụ", "nọ", "ga", "dị", "ka", "ma"
+]
+
+ENGLISH_WORDS = [
+    "farm", "farming", "agriculture", "crop", "crops", "plant", "plants", "seed", "seeds", "soil",
+    "water", "rain", "weather", "harvest", "yield", "field", "fields", "farmer", "farmers", "grow",
+    "growing", "fertilizer", "pesticide", "irrigation", "livestock", "cattle", "chicken", "goat", "sheep",
+    "maize", "corn", "rice", "wheat", "vegetable", "vegetables", "fruit", "fruits", "tree", "trees",
+    "cultivate", "cultivation", "plow", "plowing", "sow", "sowing", "reap", "reaping", "season", "seasons"
 ]
 
 def detect_language_keywords(text: str) -> str:
@@ -372,16 +373,17 @@ def detect_language_keywords(text: str) -> str:
     hausa_count = sum(1 for word in HAUSA_WORDS if word in text_lower)
     yoruba_count = sum(1 for word in YORUBA_WORDS if word in text_lower)
     igbo_count = sum(1 for word in IGBO_WORDS if word in text_lower)
+    english_count = sum(1 for word in ENGLISH_WORDS if word in text_lower)
     
-    logger.info(f"Language detection scores - Hausa: {hausa_count}, Yoruba: {yoruba_count}, Igbo: {igbo_count}")
+    logger.info(f"Language detection scores - Hausa: {hausa_count}, Yoruba: {yoruba_count}, Igbo: {igbo_count}, English: {english_count}")
     
-    if hausa_count > yoruba_count and hausa_count > igbo_count:
+    if hausa_count > yoruba_count and hausa_count > igbo_count and hausa_count > english_count:
         logger.info("Keyword detection: Hausa")
         return "ha"
-    elif yoruba_count > hausa_count and yoruba_count > igbo_count:
+    elif yoruba_count > hausa_count and yoruba_count > igbo_count and yoruba_count > english_count:
         logger.info("Keyword detection: Yoruba")
         return "yo"
-    elif igbo_count > hausa_count and igbo_count > yoruba_count:
+    elif igbo_count > hausa_count and igbo_count > yoruba_count and igbo_count > english_count:
         logger.info("Keyword detection: Igbo")
         return "ig"
     else: