import pandas as pd
from algo import Algo
from db.db_utils import get_connection

if __name__ == "__main__":
    db_conn = get_connection()
    algo = Algo(db_conn)
    raw_file_name = 'MFB-2022-raw-data.csv'
    input_file_path = f'raw/{raw_file_name}'
    df_input = pd.read_csv(input_file_path)
    input_words = df_input['description'].astype(str).tolist()
    algo.match_words(input_words)
    # algo.match_words(["Cooked meat"])
    # algo.match_words(["Berry - Strawberry", "Berry - Blackberry", "strawberries, raw pineapples, bananas sliced thin", "milk, chocolate", "potatoes / tomatoes / oranges",])
    db_conn.close()