Spaces:
Running
Running
File size: 6,651 Bytes
55d18b1 bd8db44 55d18b1 bd8db44 55d18b1 bd8db44 55d18b1 bd8db44 55d18b1 bd8db44 55d18b1 5a8836e 55d18b1 5a8836e bd8db44 55d18b1 |
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 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 |
import datetime
import logging
import streamlit as st
from streamlit.delta_generator import DeltaGenerator
import cv2
import numpy as np
from input.input_observation import InputObservation
from input.input_validator import get_image_datetime, is_valid_email, is_valid_number
m_logger = logging.getLogger(__name__)
m_logger.setLevel(logging.INFO)
'''
A module to setup the input handling for the whale observation guidance tool
both the UI elements (setup_input_UI) and the validation functions.
'''
allowed_image_types = ['jpg', 'jpeg', 'png', 'webp']
# an arbitrary set of defaults so testing is less painful...
# ideally we add in some randomization to the defaults
spoof_metadata = {
"latitude": 23.5,
"longitude": 44,
"author_email": "[email protected]",
"date": None,
"time": None,
}
def check_inputs_are_set(empty_ok:bool=False, debug:bool=False) -> bool:
"""
Checks if all expected inputs have been entered
Implementation: via the Streamlit session state.
Args:
empty_ok (bool): If True, returns True if no inputs are set. Default is False.
debug (bool): If True, prints and logs the status of each expected input key. Default is False.
Returns:
bool: True if all expected input keys are set, False otherwise.
"""
filenames = st.session_state.image_filenames
if len(filenames) == 0:
return empty_ok
exp_input_key_stubs = ["input_latitude", "input_longitude"]
#exp_input_key_stubs = ["input_latitude", "input_longitude", "input_author_email", "input_date", "input_time", "input_image_selector"]
vals = []
for image_filename in filenames:
for stub in exp_input_key_stubs:
key = f"{stub}_{image_filename}"
val = None
if key in st.session_state:
val = st.session_state[key]
vals.append(val)
if debug:
msg = f"{key:15}, {(val is not None):8}, {val}"
m_logger.debug(msg)
print(msg)
return all([v is not None for v in vals])
def setup_input(
viewcontainer: DeltaGenerator=None,
_allowed_image_types: list=None, ) -> InputObservation:
"""
Sets up the input interface for uploading an image and entering metadata.
It provides input fields for an image upload, lat/lon, author email, and date-time.
In the ideal case, the image metadata will be used to populate location and datetime.
Parameters:
viewcontainer (DeltaGenerator, optional): The Streamlit container to use for the input interface. Defaults to st.sidebar.
_allowed_image_types (list, optional): List of allowed image file types for upload. Defaults to allowed_image_types.
Returns:
InputObservation: An object containing the uploaded image and entered metadata.
"""
if viewcontainer is None:
viewcontainer = st.sidebar
if _allowed_image_types is None:
_allowed_image_types = allowed_image_types
viewcontainer.title("Input image and data")
# 1. Input the author email
author_email = viewcontainer.text_input("Author Email", spoof_metadata.get('author_email', ""))
if author_email and not is_valid_email(author_email):
viewcontainer.error("Please enter a valid email address.")
# 2. Image Selector
uploaded_files = viewcontainer.file_uploader("Upload an image", type=allowed_image_types, accept_multiple_files=True)
observations = {}
images = {}
image_hashes = []
filenames = []
if uploaded_files is not None:
for file in uploaded_files:
viewcontainer.title(f"Metadata for {file.name}")
# Display the uploaded image
# load image using cv2 format, so it is compatible with the ML models
file_bytes = np.asarray(bytearray(file.read()), dtype=np.uint8)
filename = file.name
filenames.append(filename)
image = cv2.imdecode(file_bytes, 1)
# Extract and display image date-time
image_datetime = None # For storing date-time from image
image_datetime = get_image_datetime(file)
m_logger.debug(f"image date extracted as {image_datetime} (from {uploaded_files})")
# 3. Latitude Entry Box
latitude = viewcontainer.text_input(
"Latitude for "+filename,
spoof_metadata.get('latitude', ""),
key=f"input_latitude_{filename}")
if latitude and not is_valid_number(latitude):
viewcontainer.error("Please enter a valid latitude (numerical only).")
m_logger.error(f"Invalid latitude entered: {latitude}.")
# 4. Longitude Entry Box
longitude = viewcontainer.text_input(
"Longitude for "+filename,
spoof_metadata.get('longitude', ""),
key=f"input_longitude_{filename}")
if longitude and not is_valid_number(longitude):
viewcontainer.error("Please enter a valid longitude (numerical only).")
m_logger.error(f"Invalid latitude entered: {latitude}.")
# 5. Date/time
## first from image metadata
if image_datetime is not None:
time_value = datetime.datetime.strptime(image_datetime, '%Y:%m:%d %H:%M:%S').time()
date_value = datetime.datetime.strptime(image_datetime, '%Y:%m:%d %H:%M:%S').date()
else:
time_value = datetime.datetime.now().time() # Default to current time
date_value = datetime.datetime.now().date()
## if not, give user the option to enter manually
date_option = st.sidebar.date_input("Date for "+filename, value=date_value)
time_option = st.sidebar.time_input("Time for "+filename, time_value)
observation = InputObservation(image=file, latitude=latitude, longitude=longitude,
author_email=author_email, date=image_datetime, time=None,
date_option=date_option, time_option=time_option)
image_hash = observation.to_dict()["image_md5"]
observations[image_hash] = observation
images[image_hash] = image
image_hashes.append(image_hash)
st.session_state.images = images
st.session_state.files = uploaded_files
st.session_state.observations = observations
st.session_state.image_hashes = image_hashes
st.session_state.image_filenames = filenames
|