Spaces:
Sleeping
Sleeping
| import utils | |
| import os | |
| import folium | |
| import pandas as pd | |
| import streamlit as st | |
| import geopandas as gpd | |
| from folium.plugins import Draw | |
| from shapely.geometry import Polygon | |
| from streamlit_folium import st_folium | |
| from authentication import greeting, check_password | |
| import shapely.ops as ops | |
| from functools import partial | |
| import pyproj | |
| def check_authentication(): | |
| if not check_password(): | |
| st.stop() | |
| def display_existing_fields(current_user): | |
| with st.expander("Existing Fields", expanded=False): | |
| if os.path.exists(f"fields_{current_user}.parquet"): | |
| gdf = gpd.read_parquet(f"fields_{current_user}.parquet") | |
| st.write(gdf) | |
| mm = gdf.explore() | |
| st_folium(mm) | |
| else: | |
| st.info("No Fields Added Yet!") | |
| def add_existing_fields_to_map(m, current_user): | |
| if os.path.exists(f"fields_{current_user}.parquet"): | |
| fg = folium.FeatureGroup(name="Existing Fields", control=True).add_to(m) | |
| gdf = gpd.read_parquet(f"fields_{current_user}.parquet") | |
| for i, row in gdf.iterrows(): | |
| edges = row['geometry'].exterior.coords.xy | |
| edges = [[i[1], i[0]] for i in zip(*edges)] | |
| folium.Polygon(edges, color='blue', fill=True, fill_color='blue', fill_opacity=0.6).add_to(fg) | |
| return m | |
| def get_center_of_existing_fields(current_user): | |
| if os.path.exists(f"fields_{current_user}.parquet"): | |
| gdf = gpd.read_parquet(f"fields_{current_user}.parquet") | |
| edges = gdf['geometry'][0].exterior.coords.xy | |
| edges = [[i[1], i[0]] for i in zip(*edges)] | |
| edges_center = [sum([i[0] for i in edges]) / len(edges), sum([i[1] for i in edges]) / len(edges)] | |
| return edges_center | |
| return [15.572363674301132, 32.69167103104079] | |
| def display_map_and_drawing_controls(m, center_start): | |
| zoom_start = 13 | |
| if st.session_state['active_drawing'] is None: | |
| st.info("IMPORTANT: Click on the drawing to confirm the drawn field", icon="🚨") | |
| sat_basemap = utils.basemaps['Google Satellite Hybrid'] # Change this line to use 'Google Satellite Hybrid' | |
| sat_basemap.add_to(m) | |
| folium.LayerControl().add_to(m) | |
| output = st_folium(m, center=center_start, zoom=zoom_start, key="new", width=800) | |
| active_drawing = output['last_active_drawing'] | |
| st.session_state['active_drawing'] = active_drawing | |
| return False | |
| else: | |
| st.info("Drawing Captured! Click on the button below to Clear Drawing and Draw Again") | |
| active_drawing = st.session_state['active_drawing'] | |
| new_map = folium.Map(location=center_start, zoom_start=8) | |
| edges = [[i[1], i[0]] for i in active_drawing['geometry']['coordinates'][0]] | |
| edges_center = [sum([i[0] for i in edges]) / len(edges), sum([i[1] for i in edges]) / len(edges)] | |
| folium.Polygon(edges, color='green', fill=True, fill_color='green', fill_opacity=0.6, name="New Field").add_to(new_map) | |
| sat_basemap = utils.basemaps['Google Satellite'] | |
| sat_basemap.add_to(new_map) | |
| folium.LayerControl().add_to(new_map) | |
| st_folium(new_map, center=edges_center, zoom=zoom_start, key="drawn", width=800) | |
| return True | |
| def handle_user_actions(active_drawing, current_user, intersects, within_area): | |
| draw_again_col, add_field_info_col = st.columns([1, 1]) | |
| with draw_again_col: | |
| draw_again = st.button("Draw Again", key="draw_again", help="Click to Clear Drawing and Draw Again", | |
| type="primary", use_container_width=True, disabled=st.session_state['active_drawing'] is None) | |
| if draw_again: | |
| st.session_state['active_drawing'] = None | |
| st.rerun() | |
| with add_field_info_col: | |
| if st.session_state['active_drawing'] is None: | |
| st.info("Drawing not captured yet!") | |
| else: | |
| field_name = st.text_input("Field Name*", help="Enter a distinct name for the field", key="field_name") | |
| if field_name == "": | |
| st.warning("Field Name cannot be empty!") | |
| if os.path.exists(f"fields_{current_user}.parquet"): | |
| gdf = gpd.read_parquet(f"fields_{current_user}.parquet") | |
| if field_name in gdf['name'].tolist(): | |
| st.warning("Field Name already exists. Please enter a different name!") | |
| submit = st.button("Submit", key="submit", help="Click to Submit Field Information", type="primary", | |
| use_container_width=True,disabled=(st.session_state['active_drawing'] is None or field_name == "") or intersects or not within_area) | |
| if submit: | |
| save_field_information(active_drawing, field_name, current_user) | |
| st.success("Field Information Submitted Successfully!") | |
| st.session_state['active_drawing'] = None | |
| st.rerun() | |
| def save_field_information(active_drawing, field_name, current_user): | |
| edges = [[i[0], i[1]] for i in active_drawing['geometry']['coordinates'][0]] | |
| geom = Polygon(edges) | |
| field_dict = { | |
| "name": field_name, | |
| "geometry": geom | |
| } | |
| gdf = gpd.GeoDataFrame([field_dict], geometry='geometry') | |
| gdf.crs = "EPSG:4326" | |
| if os.path.exists(f"fields_{current_user}.parquet"): | |
| old_gdf = gpd.read_parquet(f"fields_{current_user}.parquet") | |
| gdf = gpd.GeoDataFrame(pd.concat([old_gdf, gdf], ignore_index=True), crs="EPSG:4326") | |
| gdf.to_parquet(f"fields_{current_user}.parquet") | |
| def initialize_active_drawing_state(): | |
| if 'active_drawing' not in st.session_state: | |
| st.session_state['active_drawing'] = None | |
| if 'current_user' not in st.session_state: | |
| st.session_state['current_user'] = None | |
| def check_intersection_with_existing_fields(active_drawing, current_user): | |
| if active_drawing is None: | |
| return False | |
| if os.path.exists(f"fields_{current_user}.parquet"): | |
| gdf = gpd.read_parquet(f"fields_{current_user}.parquet") | |
| edges = [[i[0], i[1]] for i in active_drawing['geometry']['coordinates'][0]] | |
| geom = Polygon(edges) | |
| geom = gpd.GeoSeries([geom]*len(gdf), crs="EPSG:4326") | |
| geom1 = geom.to_crs(gdf.crs) | |
| geom2 = gdf.geometry.to_crs(gdf.crs) | |
| if geom1.overlaps(geom2).any(): | |
| st.warning("Field intersects with existing fields. Please draw again!") | |
| with st.expander("Intersecting Fields", expanded=False): | |
| m = geom1.explore(name= "New Field", color="red") | |
| m = gdf.explore(m=m, name="Existing Fields", color="blue") | |
| st_folium(m) | |
| return True | |
| return False | |
| def check_polygon_area_within_range(active_drawing, min_area_km2=1, max_area_km2=10): | |
| if active_drawing is None: | |
| return | |
| edges = [[i[0], i[1]] for i in active_drawing['geometry']['coordinates'][0]] | |
| geom = Polygon(edges) | |
| geom_area = ops.transform( | |
| partial( | |
| pyproj.transform, | |
| pyproj.Proj(init='EPSG:4326'), | |
| pyproj.Proj(proj='aea',lat_1=geom.bounds[1], lat_2=geom.bounds[3])) | |
| , geom) | |
| geom_area = geom_area.area / 10**6 | |
| if geom_area < min_area_km2: | |
| st.warning(f"Field area is less than {min_area_km2} km2. Please draw again!") | |
| return False | |
| if geom_area > max_area_km2: | |
| st.warning(f"Field area is more than {max_area_km2} km2. Please draw again!") | |
| return False | |
| return True | |
| def add_drawing(): | |
| initialize_active_drawing_state() | |
| current_user = greeting("Drag and Zoom and draw your fields on the map, make sure to name them uniquely") | |
| current_user = st.session_state['current_user'] | |
| display_existing_fields(current_user) | |
| center_start = get_center_of_existing_fields(current_user) | |
| zoom_start = 13 | |
| m = folium.Map(location=center_start, zoom_start=zoom_start) | |
| draw_options = {'polyline': False, 'polygon': True, 'rectangle': True, 'circle': False, 'marker': False, 'circlemarker': False} | |
| Draw(export=True, draw_options=draw_options).add_to(m) | |
| m = add_existing_fields_to_map(m, current_user) | |
| captured = display_map_and_drawing_controls(m, center_start) | |
| if captured: | |
| intersects = check_intersection_with_existing_fields(st.session_state['active_drawing'], current_user) | |
| within_area = check_polygon_area_within_range(st.session_state['active_drawing']) | |
| handle_user_actions(st.session_state['active_drawing'], current_user, intersects, within_area) | |
| if __name__ == '__main__': | |
| check_authentication() | |
| add_drawing() |