atlasai-dstoolkit-client 0.0.14__py3-none-any.whl → 0.0.16__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

@@ -11,19 +11,21 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # 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')
 
@@ -45,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.results = []
 
     @property
     def status(self):
@@ -66,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.')
@@ -74,6 +80,11 @@ class Workflow:
         return result
 
     def logs(self, only_errors=True):
+        def cleanup_logs(logs):
+            if not logs:
+                return "-"
+            return ast.literal_eval(logs).decode('utf-8')
+
         if not self.id:
             raise Exception('No valid execution of the workflow. Please use the `run` method first.')
 
@@ -86,7 +97,6 @@ class Workflow:
         o.print_subtitle(f'Message: {workflow["data"]["message"] or "-"}')
 
         o.print_body('---------------------------------------------------')
-
         if not result['data']:
             if only_errors:
                 o.print_title('No errors detected.')
@@ -96,7 +106,7 @@ class Workflow:
         for pod, data in result['data'].items():
             o.print_title(f'Pod: {pod}')
             o.print_subtitle(f'Message: {data["message"] or "-"}')
-            o.print_subtitle(f'Logs: {data["logs"] or "-"}')
+            o.print_subtitle(f'Logs: {cleanup_logs(data.get("logs"))}')
             o.print_body('---------------------------------------------------')
 
     def run(self, wait_until_complete=False):
@@ -182,6 +192,9 @@ class Electrification(Workflow):
             raise Exception('`aoi_path` or `aoi` must be specified.')
 
     def _results(self, save_to=None):
+        if self.results:
+            return self.results
+
         results = []
         if not self.id:
             raise Exception('No valid execution of the workflow. Please use the `run` method first.')
@@ -198,7 +211,7 @@ class Electrification(Workflow):
             for fut in as_completed(fut_map):
                 results.append(fut.result())
                 pbar.update(1)
-
+        self.results = results
         return results
 
     def _download_file(self, name, url, to=None):
@@ -219,6 +232,57 @@ class Electrification(Workflow):
 
         return local_path
 
+    def preview(self, from_=None, to_=None, selection=None, limit=None):
+        if not self.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.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.14.dist-info → atlasai_dstoolkit_client-0.0.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atlasai-dstoolkit-client
-Version: 0.0.14
+Version: 0.0.16
 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.14.dist-info → atlasai_dstoolkit_client-0.0.16.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=HlmLNma1P4xyd74gzM5moBXPGjVQqziGOtyRTXBXUhw,6994
-atlasai_dstoolkit_client-0.0.14.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-atlasai_dstoolkit_client-0.0.14.dist-info/METADATA,sha256=vWbApYiDX_XA2mFvwXKRWDCV6KW5OwOYL-jqJN4S56A,1471
-atlasai_dstoolkit_client-0.0.14.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-atlasai_dstoolkit_client-0.0.14.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
-atlasai_dstoolkit_client-0.0.14.dist-info/RECORD,,
+atlasai/toolkit/workflows.py,sha256=u17Hl-tIgyoB_-dA23g5h2o900fcsyj3H1uVVbZ4Mao,9392
+atlasai_dstoolkit_client-0.0.16.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+atlasai_dstoolkit_client-0.0.16.dist-info/METADATA,sha256=cC7U97hGZ68zR7Gapl1ZXPR2R2Q2Y4Xyp7--UtfdOnM,1547
+atlasai_dstoolkit_client-0.0.16.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+atlasai_dstoolkit_client-0.0.16.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
+atlasai_dstoolkit_client-0.0.16.dist-info/RECORD,,