main.py

import base64
from pathlib import Path
import pandas as pd
import streamlit
import os
from pathlib import Path
import numpy as np
import pydeck as pdk
import random
from send_email import send_message

import streamlit as st
from PIL import Image


file_dir = Path(os.path.dirname(os.path.abspath(__file__)))
DATE_TIME = "date/time"
DATA_URL = file_dir / "data_long.csv"


@st.cache(persist=True)
def load_data(DATA_URL, nrows=None):
    data = pd.read_csv(DATA_URL, nrows=nrows)

    lowercase = lambda x: str(x).lower()
    data.rename(lowercase, axis="columns", inplace=True)

    try:
        data[DATE_TIME] = pd.to_datetime(data[DATE_TIME])
    except KeyError:
        pass

    return data


extra = load_data(file_dir / "data_long.csv")
extra = extra.dropna()

# CREATING FUNCTION FOR MAPS
def map(data, lat, lon, zoom):
    st.write(
        pdk.Deck(
            map_style="mapbox://styles/mapbox/light-v9",
            initial_view_state={
                "latitude": float(lat),
                "longitude": float(lon),
                "zoom": zoom,
                "pitch": 50,
            },
            layers=[
                pdk.Layer(
                    "HexagonLayer",
                    data=data,
                    get_position=["lon", "lat"],
                    radius=500,
                    elevation_scale=50,
                    elevation_range=[0, 3000],
                    pickable=True,
                    extruded=True,
                    coverage=1,
                ),
            ],
        )
    )


def main():
    img = Image.open(file_dir / "Images" / "logo-print.png")
    st.image(img, width=200, use_column_width=200)

    img2 = Image.open(file_dir / "Images" / "background.jpeg")
    st.image(
        img2,
        caption="No task is so important that we can’t take the time to do it safely. A safe company is a successful company.",
    )

    # Title
    st.header("Incidents Reporting")
    # test_data = pd.concat([data]*20, ignore_index=True)

    latitudes = [
        "29.07393266320772",
        "38.83889745592545",
        "67.28150241774922",
        "65.76861506352907",
        "27.850857403798024",
        "31.400790392869073",
        "32.14315679366797",
        "29.7828109",
        "69.5259183676525",
        "-38.41968552700519",
    ]
    longitudes = [
        "-95.74462745592294",
        "-90.07362745592292",
        "8.899795546234275",
        "7.891749990422023",
        "-98.58710217513708",
        "-103.5499547076863",
        "-102.18566929860735",
        "-95.618529",
        "-152.13526519858635",
        "142.6508871470676",
    ]
    basenames = [
        "Sweeny Refinery",
        "Wood River Refinery",
        "SNE-1 well",
        "Warka well",
        "Eagle Ford Well",
        "Permian Basin Well",
        "Midland Well",
        "Houston office",
        "Alphine well",
        "Otway Basin Well",
    ]
    baseIDS = [
        "B02512",
        "B09768",
        "B06567",
        "B08798",
        "B04673",
        "B05767",
        "B06376",
        "B02783",
        "B04237",
        "B06788",
    ]
    csv_location = [0, 0, 0]

    # Added ,clear_on_submit=True to clear the fields once the submit button hass been pressed
    with st.form("my_form", clear_on_submit=True):
        st.write("Please Fill Out the Information Below")
        # used for date in CSV file
        date = st.date_input("Enter Date")
        # Used for incidents csv file, DOES NOT PREVENT OR CHECK FOR IDENTICAL NUMBERS YET
        incident_number = random.randint(100000, 9999999)
        # Used to select incident typer
        incident_type = st.selectbox(
            "Select Type or Incident",
            ("Trip/Fall", "Heavy Equipment Violation", "Other"),
        )
        # Used for

        location = st.selectbox(
            "Select location, If not available select other and add in description below",
            (
                "Sweeny Refinery",
                "Wood River Refinery",
                "SNE-1 well",
                "Warka well",
                "Eagle Ford Well",
                "Permian Basin Well",
                "Midland Well",
                "Houston office",
            ),
        )

        location_index = basenames.index(location)
        csv_location[0] = basenames[location_index]
        csv_location[1] = longitudes[location_index]
        csv_location[2] = latitudes[location_index]

        description = st.text_area(
            "Enter a brief description, Include time, Number of people Involved and if Medical attention was required."
        )
        uploaded_file = st.file_uploader("Choose a file")

        submitted = st.form_submit_button("Submit")
        if submitted:
            # Fetch Data from db
            st.write("Submitted. Thank you for your dedication to safety.")
            # ##### used for the location
            base_ID = csv_location[0]
            long_ID = csv_location[1]
            lat_ID = csv_location[2]
            print(description)
            send_message(description)
            # print(csv_location[0], csv_location[1], csv_location[2])

    # used to display the map
    with st.form("my_map"):

        map_display = st.selectbox(
            "View our current Incidents",
            ("", "All", "Trip/Fall", "Heavy Equipment Violation", "Other"),
        )
        submitted = st.form_submit_button("Submit")

        if submitted:
            if map_display == "All":
                map(extra[["lat", "lon"]], csv_location[2], csv_location[1], 10)
            elif map_display == "Trip/Fall":
                # st.write("slider", slider_val, "checkbox", checkbox_val)
                # map(data, lat, lon, zoom)
                map(
                    extra.loc[extra["incident type"] == "trip"][["lat", "lon"]],
                    csv_location[2],
                    csv_location[1],
                    10,
                )
            elif map_display == "Heavy Equipment Violation":
                map(
                    extra.loc[extra["incident type"] == "too close to heavy equipment"][
                        ["lat", "lon"]
                    ],
                    csv_location[2],
                    csv_location[1],
                    10,
                )

            # st.write("slider", slider_val, "checkbox", checkbox_val)


if __name__ == "__main__":
    main()