bugfixes

algo.py

@@ -5,7 +5,7 @@ from pluralizer import Pluralizer
 from similarity_fast import SimilarityFast
 from food_nonfood import classify_as_food_nonfood, pessimistic_food_nonfood_score
 from utils import clean_word
-from db.db_utils import store_mapping_to_db, get_mapping_from_db
+from db.db_utils import store_mapping_to_db, get_mapping_from_db, get_dictionary_data_from_db
 from ask_gpt import query_gpt
 from multi_food_item_detector import extract_items
 
@@ -175,7 +175,7 @@ class Algo:
 
         # try the singular form of the word
         singular = self.pluralizer.pluralize(input_word_clean, 1)
-        mapping = wrap_mapping_with_dictionary_data(get_mapping_from_db(self.db_cursor, singular))
+        mapping = self.wrap_mapping_with_dictionary_data(get_mapping_from_db(self.db_cursor, singular))
         if mapping:
             print(f" - Found mapping in db: {mapping}")
             return mapping
@@ -204,7 +204,7 @@ class Algo:
                 'food_nonfood_score': food_nonfood[1]
             }
             store_mapping_to_db(self.db_cursor, self.db_conn, mapping)
-            return wrap_mapping_with_dictionary_data(mapping)
+            return self.wrap_mapping_with_dictionary_data(mapping)
 
         mapping = self.perform_mapping(input_word)
 
@@ -217,7 +217,7 @@ class Algo:
         print(f" - Storing new mapping to db: {mapping}")
         store_mapping_to_db(self.db_cursor, self.db_conn, mapping)
 
-        return wrap_mapping_with_dictionary_data(mapping)
+        return self.wrap_mapping_with_dictionary_data(mapping)
 
     def wrap_mapping_with_dictionary_data(self, mapping):
         dictionary_result = get_dictionary_data_from_db(self.db_cursor, mapping['dictionary_word'])