atlasai-dstoolkit-client 0.0.15__py3-none-any.whl → 0.0.17__py3-none-any.whl

atlasai/toolkit/constants.py

@@ -11,3 +11,10 @@ DISABLE_SSL_VERIFICATION = 'DISABLE_SSL_VERIFICATION'
 
 TOKEN_ENV_VAR = 'VINZ_ENCRYPTED_TOKEN'
 TOKEN_TIMESTAMP_ENV_VAR = 'VINZ_ENCRYPTED_TOKEN_TIMESTAMP'
+
+LEAFMAP_COLORMAP = [
+    "viridis", "plasma", "inferno", "magma", "cividis",
+    "Greys", "Purples", "Blues", "Greens", "Oranges", "Reds",
+    "YlOrBr", "YlOrRd", "OrRd", "PuRd", "RdPu", "BuPu", "GnBu", "PuBu",
+    "YlGnBu", "PuBuGn", "BuGn", "YlGn"
+]

atlasai/toolkit/workflows.py

@@ -12,19 +12,20 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import ast
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from dataclasses import dataclass
+from datetime import datetime
 import logging
 import os
+import random
 import time
 
+import leafmap
 import pandas as pd
-from tqdm import tqdm
-
-from concurrent.futures import ThreadPoolExecutor, as_completed
-from dataclasses import dataclass
-
 import requests
+from tqdm import tqdm
 
-from . import api, output as o
+from . import api, constants, output as o
 
 logger = logging.getLogger('atlasai.toolkit')
 
@@ -46,6 +47,7 @@ class Workflow:
     def __init__(self, id=None):
         # Allow the selection of an existing workflow and check the logs.
         self.id = id
+        self._run_results = []
 
     @property
     def status(self):
@@ -67,6 +69,9 @@ class Workflow:
     def _results(self, *args, **kwargs):
         raise NotImplementedError
 
+    def preview(self, *args, **kwargs):
+        raise NotImplementedError
+
     def get(self):
         if not self.id:
             raise Exception('No valid execution of the workflow. Please use the `run` method first.')
@@ -186,7 +191,10 @@ class Electrification(Workflow):
         if not self.aoi_path and not self.aoi:
             raise Exception('`aoi_path` or `aoi` must be specified.')
 
-    def _results(self, save_to=None):
+    def _results(self, save_to=None, force=False):
+        if self._run_results and not force:
+            return self._run_results
+
         results = []
         if not self.id:
             raise Exception('No valid execution of the workflow. Please use the `run` method first.')
@@ -203,7 +211,7 @@ class Electrification(Workflow):
             for fut in as_completed(fut_map):
                 results.append(fut.result())
                 pbar.update(1)
-
+        self._run_results = results
         return results
 
     def _download_file(self, name, url, to=None):
@@ -224,6 +232,58 @@ class Electrification(Workflow):
 
         return local_path
 
+    def preview(self, from_=None, to_=None, selection=None, limit=None):
+        if not self._run_results:
+            raise Exception('No result to be displayed. Run the workflow first and wait for its completion.')
+
+        _results = []
+        if selection is None:
+            selection = []
+        if selection:
+            selection = ['-'.join(s.split('-')[:2]) for s in selection]
+            from_, to_, limit = None, None, None
+
+        if from_:
+            from_ = datetime.strptime(from_, "%Y-%m-%d")
+        if to_:
+            to_ = datetime.strptime(to_, "%Y-%m-%d")
+
+        for f in sorted(self._run_results):
+            yyyy, mm = f.split('_')[-1].replace('.tif', '').split('-')
+            dt = datetime(year=int(yyyy), month=int(mm), day=1)
+            if selection:
+                if f'{yyyy}-{mm}' not in selection:
+                    continue
+            if from_ and from_ > dt:
+                continue
+            if to_ and to_ < dt:
+                continue
+            if limit and len(_results) >= limit:
+                continue
+            _results.append(dict(
+                layer_name=f'{yyyy}-{mm}',
+                file=f,
+                colormap=constants.LEAFMAP_COLORMAP[random.randint(0, len(constants.LEAFMAP_COLORMAP) - 1)]
+            ))
+        if not _results:
+            raise Exception('No result matches your criteria.')
+        m = leafmap.Map()
+        if len(_results) > 10:
+            logger.warning(f'Too many layers to display. Expect increased loading time. Layers to display: {len(_results)}')
+
+        for idx, f in enumerate(_results):
+            m.add_raster(
+                f['file'],
+                layer_name=f['layer_name'],
+                nodata=0,
+                colormap=f['colormap'],
+                opacity=0.5,
+            )
+        if hasattr(m, "add_layers_control"):
+            m.add_layers_control()
+        elif hasattr(m, "add_layer_control"):
+            m.add_layer_control()
+        return m
 
 def List(search=None, offset=0, limit=100):
     pd.set_option("display.max_colwidth", None)

{atlasai_dstoolkit_client-0.0.15.dist-info → atlasai_dstoolkit_client-0.0.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atlasai-dstoolkit-client
-Version: 0.0.15
+Version: 0.0.17
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: AtlasAI SWE
@@ -13,6 +13,8 @@ Requires-Dist: arrow (<2.0.0,>=1.3.0)
 Requires-Dist: atlasai-mlhub-client
 Requires-Dist: furl (<3.0.0,>=2.1.2)
 Requires-Dist: geopandas (~=1.1.1)
+Requires-Dist: leafmap (~=0.52.0)
+Requires-Dist: localtileserver (~=0.10.6)
 Requires-Dist: pandas (<3.0.0,>=2.2.3)
 Requires-Dist: pyarrow (>=19.0.1)
 Requires-Dist: python-dotenv (<2.0.0,>=1.0.1)

{atlasai_dstoolkit_client-0.0.15.dist-info → atlasai_dstoolkit_client-0.0.17.dist-info}/RECORD

@@ -1,6 +1,6 @@
 atlasai/toolkit/__init__.py,sha256=0d2EYTgFf-RH2z1M83gGfZdrt_PBXbJk6o5ZVAKCPtA,1013
 atlasai/toolkit/api.py,sha256=BvO-gLRmbmkKduwbbADjcLlIkS9blzfM_cbMR4DhQmU,5269
-atlasai/toolkit/constants.py,sha256=Jxozn9tKOvAxyOYOZ6bzFtI9m1YETJF1GURzTl9NNC8,422
+atlasai/toolkit/constants.py,sha256=fCVI3Q6g28uCQJsY10BUsb301TO9_W0OqWdd9QwZrfg,675
 atlasai/toolkit/dataset.py,sha256=fhzhxF9YMzIwEpaJZEPOK2SuLMJhVGI75eahxnH_T2c,3254
 atlasai/toolkit/fabric.py,sha256=6aFR2PGQc9P3Qa07WdBg9eKoUzU8n2y_-gGjYcyMrWY,1921
 atlasai/toolkit/feature.py,sha256=Q5M9zOGafynYuKaELL1kZemYPfKAZh84TgH7jw9J3ZU,2949
@@ -10,9 +10,9 @@ atlasai/toolkit/model.py,sha256=RUe0HbDpzvHOV9A4rzG3PgN9boMWDHQ2tR7IKHXzbx8,4126
 atlasai/toolkit/output.py,sha256=FyDjrpVlbrEyfHfwOpxp8H57jx_qXahDjO1qpHIeuYM,473
 atlasai/toolkit/requests.py,sha256=X86nIo07hAjUlilZcZ1lV8RB7KOsTKbTGtcY_SpFEXY,1223
 atlasai/toolkit/utils.py,sha256=lYh3P2XOshRgHCjFeXJ0FOJWQW64sddgx8c2kL6Wqwc,1566
-atlasai/toolkit/workflows.py,sha256=9NTqPZqoT46tdjMuFApYsGUpbNNDP_Gp2gYMXgVLo_s,7158
-atlasai_dstoolkit_client-0.0.15.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-atlasai_dstoolkit_client-0.0.15.dist-info/METADATA,sha256=Ada-4ks6hKHOq1GmPEJJKnO9TlATP2XVRyAat70spbo,1471
-atlasai_dstoolkit_client-0.0.15.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-atlasai_dstoolkit_client-0.0.15.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
-atlasai_dstoolkit_client-0.0.15.dist-info/RECORD,,
+atlasai/toolkit/workflows.py,sha256=j91nsVTLzv689EE-aBOZyuOV0a7xlZh4DEkgWIzBFiU,9450
+atlasai_dstoolkit_client-0.0.17.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+atlasai_dstoolkit_client-0.0.17.dist-info/METADATA,sha256=hiDgJIBsJ4WO47gGKD0FdUM-oekVgXZn704V7W08lds,1547
+atlasai_dstoolkit_client-0.0.17.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+atlasai_dstoolkit_client-0.0.17.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
+atlasai_dstoolkit_client-0.0.17.dist-info/RECORD,,