Format.

app.py

@@ -5,7 +5,7 @@ from utils.gradio import get_window_url_params
 from utils.indicators import IndexGenerator
 
 # Instantiate outside gradio app to avoid re-initializing GEE, which is slow
-indexgenerator = IndexGenerator(indices=['NDWI','Water','Protected','Air'])
+indexgenerator = IndexGenerator(indices=["NDWI", "Water", "Protected", "Air"])
 
 with gr.Blocks() as demo:
  with gr.Column():

utils/duckdb_queries.py

@@ -28,12 +28,16 @@ def get_project_geometry(project_name):
  "SELECT geometry FROM project WHERE name = ? LIMIT 1", [project_name]
  ).fetchall()
 
+
 def get_project_centroid(project_name):
  # Workaround to get centroid of project
  # To-do: refactor to only use DuckDB spatial extension
  _geom = get_project_geometry(project_name)
- _polygon = json.dumps(json.loads(_geom[0][0])['features'][0]['geometry'])
- return con.sql(f"SELECT ST_X(ST_Centroid(ST_GeomFromGeoJSON('{_polygon}'))) AS longitude, ST_Y(ST_Centroid(ST_GeomFromGeoJSON('{_polygon}'))) AS latitude;").fetchall()[0]
+ _polygon = json.dumps(json.loads(_geom[0][0])["features"][0]["geometry"])
+ return con.sql(
+ f"SELECT ST_X(ST_Centroid(ST_GeomFromGeoJSON('{_polygon}'))) AS longitude, ST_Y(ST_Centroid(ST_GeomFromGeoJSON('{_polygon}'))) AS latitude;"
+ ).fetchall()[0]
+
 
 def get_project_scores(project_name, start_year, end_year):
  return con.execute(

utils/indicators.py

@@ -13,7 +13,9 @@ from utils import duckdb_queries as dq
 
 from . import logging
 
-GEE_SERVICE_ACCOUNT = "climatebase-july-2023@ee-geospatialml-aquarry.iam.gserviceaccount.com"
+GEE_SERVICE_ACCOUNT = (
+ "climatebase-july-2023@ee-geospatialml-aquarry.iam.gserviceaccount.com"
+)
 INDICES_FILE = "indices.yaml"
 
 
@@ -36,7 +38,6 @@ class IndexGenerator:
  self.indices = self._load_indices(INDICES_FILE)
  breakpoint()
 
-
  def _cloudfree(self, gee_path, daterange):
  """
  Internal method to generate a cloud-free composite.
@@ -141,7 +142,6 @@ class IndexGenerator:
  return out
 
  def generate_composite_index_df(self, year, project_geometry, indices=[]):
- 
  data = {
  "metric": indices,
  "year": year,
@@ -175,22 +175,26 @@ class IndexGenerator:
  project_geometry = dq.get_project_geometry(project_name)
  project_centroid = dq.get_project_centroid(project_name)
  # to-do: refactor to involve less transformations
- _polygon = json.dumps(json.loads(project_geometry[0][0])['features'][0]['geometry'])
+ _polygon = json.dumps(
+ json.loads(project_geometry[0][0])["features"][0]["geometry"]
+ )
  # to-do: don't use self.roi and instead pass patameter strategically
- self.roi = ee.Geometry.Polygon(json.loads(_polygon)['coordinates'])
+ self.roi = ee.Geometry.Polygon(json.loads(_polygon)["coordinates"])
 
  # to-do: pararelize?
  for year in years:
  logging.info(year)
  self.project_name = project_name
- df = self.generate_composite_index_df(year, project_geometry, list(self.indices.keys()))
+ df = self.generate_composite_index_df(
+ year, project_geometry, list(self.indices.keys())
+ )
  dfs.append(df)
 
  # Concatenate all dataframes
  df_concat = pd.concat(dfs)
- df_concat['centroid'] = project_centroid
- df_concat['project_name'] = project_name
- df_concat['geojson'] = project_geometry
+ df_concat["centroid"] = project_centroid
+ df_concat["project_name"] = project_name
+ df_concat["geojson"] = project_geometry
  breakpoint()
  return df_concat
 
@@ -299,5 +303,3 @@ class IndexGenerator:
  logging.info("upserted records into motherduck")
  scores = dq.get_project_scores(project_name, start_year, end_year)
  return scores
-
-