fridge_test_data/utils.py at main · FixOurFood/fridge_test_data · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import streamlit as st
import pandas as pd
import numpy as np

data_url = 'https://docs.google.com/spreadsheets/d/e/' \
           + '2PACX-1vQgzBjOTgcTLekn38loPWBIZY82PDkEg349snHFnPVlmGJT'\
           + 'AH1tUZ98RFvvjugG8_sRNYI1nCgP7IOz'\
           + '/pub?gid=493561849&single=true&output=csv'

def get_fridge_data(pulses, alt, chicken, df):
    """Get fridge data based on user inputs."""
    filtered_df = df[
        (df['Eat more beans and lentils'] == f'{pulses}%') &
        (df['Eat more alternative meat and dairy'] == f'{alt}%') &
        (df['Eat more chicken'] == f'{chicken}%')
    ]

    result = {}

    if not filtered_df.empty:
        row = filtered_df.iloc[0].copy()
        row = row.fillna("-")
        result = {
            'fridge top door': row['fridge top door'],
            'fridge top middle': row['fridge top middle'],
            'fridge top left': row['fridge top left'],

            'fridge middle left': row['fridge middle left'],
            'fridge bottom right': row['fridge bottom right'],
            'fridge middle right': row['fridge middle right'],

            'cupboard left': row['cupboard left'],
            'fridge top right': row['fridge top right'],
            'countertop': row['Countertop'],
        }

    return result

def fill_fridge(fridge_dict):

    cols = st.columns(3)

    with cols[0]:
        print_item(fridge_dict.get('fridge top left', 'No data'))
        print_item(fridge_dict.get('fridge middle left', 'No data'))
        print_item(fridge_dict.get('fridge top middle', 'No data'))

    with cols[1]:
        print_item(fridge_dict.get('fridge top right', 'No data'))
        print_item(fridge_dict.get('fridge middle right', 'No data'))
        print_item(fridge_dict.get('fridge bottom right', 'No data'))

    with cols[2]:
        print_item(fridge_dict.get('fridge top door', 'No data'))
        print_item(fridge_dict.get('countertop', 'No data'))
        print_item(fridge_dict.get('cupboard left', 'No data'))

@st.cache_data(ttl=24*3600)
def get_fridge_csv(data_url=data_url):
    data = pd.read_csv(data_url, header=1, index_col=0)
    return data

def print_item(item):

    colour = None

    if item in [
        'Hummus',
        'Bean burgers',
        'Lentil lasagne',
        'Baked beans',
        'Bag lentils']:
        colour = 'green'

    if item in [
        "Eggs",
        "Whole chicken",
        "Chicken pasta bake",
        "Chicken breasts"]:
        colour = 'orange'

    if item in [
        "Vegan cheese",
        "Vegan burgers",
        "Vegetable lasagne",
        "Veggie sausages",
        "Tofu"]:
        colour = 'violet'

    if item in [
        "Vegetable spread",
        "Plant milk",
    ]:
        colour = 'rainbow'

    if colour is not None:
        st.markdown(f""":{colour}[{item}]""")

    else:
        st.markdown(item)