-
Notifications
You must be signed in to change notification settings - Fork 0
/
streamlit_app.py
74 lines (61 loc) · 2.79 KB
/
streamlit_app.py
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
import streamlit
import pandas
import requests
import snowflake.connector
from urllib.error import URLError
streamlit.title("My Parents New Healthy Diner")
streamlit.header("Breakfast Favorites")
streamlit.text("🥣 Omega 3 & Blueberry Oatmeal")
streamlit.text("🥗 Kale, Spinach & Rocket Smoothie")
streamlit.text("🐔 Hard-Bolied Free-Range Egg")
streamlit.text("🥑🍞 Avacoda Toast")
streamlit.header("🍌🥭 Build Your Own Fruit Smoothie 🥝🍇")
my_fruit_list = pandas.read_csv("https://uni-lab-files.s3.us-west-2.amazonaws.com/dabw/fruit_macros.txt")
my_fruit_list = my_fruit_list.set_index("Fruit")
fruits_selected = streamlit.multiselect("Pick some fruits:", list(my_fruit_list.index),['Avocado', 'Strawberries'])
fruits_to_show = my_fruit_list.loc[fruits_selected]
streamlit.dataframe(fruits_to_show)
def get_fruityvice_data(this_fruit_choice):
fruityvice_response = requests.get("https://fruityvice.com/api/fruit/" + fruit_choice)
# just writes the data to the screen
# streamlit.text(fruityvice_response.json())
# take the json version of the response and normalize it
fruityvice_normalized = pandas.json_normalize(fruityvice_response.json())
return fruityvice_normalized
# New Section to display fruityvice api response
streamlit.header("Fruityvice Fruit Advice!")
try:
fruit_choice = streamlit.text_input('What fruit would you like information about?')
if not fruit_choice:
streamlit.error("Please select a fruit to get information")
else:
# output it on the screen as a table
# streamlit.dataframe(fruityvice_normalized)
back_from_function = get_fruityvice_data(fruit_choice)
streamlit.dataframe(back_from_function)
except URLError as e:
streamlit.error()
# streamlit.write('The user entered ', fruit_choice)
streamlit.header("View Our Fruit List - Add Your Favorites!")
# Snowflake-related functions
def get_fruit_load_list():
with my_cnx.cursor() as my_cur:
my_cur.execute("select * from fruit_load_list")
return my_cur.fetchall()
# Add a button to load the fruit
if streamlit.button('Get Fruit List'):
my_cnx = snowflake.connector.connect(**streamlit.secrets["snowflake"])
my_data_rows = get_fruit_load_list()
my_cnx.close()
streamlit.dataframe(my_data_rows)
# Allow the end user to add a fruit to the list
def insert_row_snowflake(new_fruit):
with my_cnx.cursor() as my_cur:
my_cur.execute("insert into fruit_load_list values ('" + new_fruit + "')")
return "Thanks for adding " + new_fruit
fruit_choice2 = streamlit.text_input('What fruit would you like to add?')
if streamlit.button('Add a Fruit to the List'):
my_cnx = snowflake.connector.connect(**streamlit.secrets["snowflake"])
back_from_function = insert_row_snowflake(fruit_choice2)
streamlit.text(back_from_function)
# streamlit.write('Thanks for adding', fruit_choice2)