import streamlit as st
import pandas as pd
import numpy as np

DATE_TIME = 'date/time'
DATA_URL = 'https://s3-us-west-2.amazonaws.com/streamlit-demo-data/uber-raw-data-sep14.csv.gz'

lowercase = lambda x: str(x).lower()

@st.cache
def load_data(nrows):
    data = pd.read_csv(DATA_URL, nrows=nrows)
    data.rename(lowercase, axis='columns', inplace=True)
    data[DATE_TIME] = pd.to_datetime(data[DATE_TIME])
    data['hour'] = (data[DATE_TIME].dt.hour + 17) % 24
    return data

data = load_data(100000)
hour = st.slider('hour', 0, 23, 11)
data = data[data[DATE_TIME].dt.hour == hour]

st.write('### Geo Data at %sh' % hour)
midpoint = (np.average(data['lat']), np.average(data['lon']))
st.deck_gl_chart(
    viewport={
        'latitude': midpoint[0],
        'longitude': midpoint[1],
        'zoom': 11,
        'pitch': 50,
    },
    layers=[{
        'type': 'HexagonLayer',
        'data': data,
        'radius': 100,
        'elevationScale': 4,
        'elevationRange': [0, 1000],
        'pickable': True,
        'extruded': True,
    }])

st.write('### Data by Minute at %sh' % hour)
st.bar_chart(np.histogram(data[DATE_TIME].dt.minute, bins=60)[0])

if st.checkbox("Show Raw Data"):
    st.write('### Raw Data at %sh' % hour)
    st.write(data)