[debug] add debug logs, set os.environ['MPLCONFIGDIR']

src/app.py

@@ -71,6 +71,7 @@ def get_parsed_bbox_points(request_input: RequestBody) -> Dict:
     source_type = request_input["source_type"] if "zoom" in request_input else SOURCE_TYPE
     app_logger.info(f"try to validate input request {request_input}...")
     request_body = RequestBody(ne=request_input["ne"], sw=request_input["sw"], model=model_name, zoom=zoom, source_type=source_type)
+    app_logger.info(f"unpacking {request_body}...")
     return {
         "bbox": [
             request_body.ne.lat, request_body.sw.lat,

src/prediction_api/samgeo_predictors.py

@@ -1,23 +1,33 @@
 # Press the green button in the gutter to run the script.
 import json
+import os
 
 from src import app_logger
 from src.utilities.constants import ROOT, MODEL_NAME, ZOOM, SOURCE_TYPE
-from src.utilities.type_hints import input_floatlist, input_floatlist2
+from src.utilities.type_hints import input_floatlist
+from src.utilities.utilities import get_system_info
 
 
-def samgeo_fast_predict(bbox: input_floatlist, zoom: float = ZOOM, model_name: str = MODEL_NAME, root_folder: str = ROOT, source_type: str = SOURCE_TYPE) -> dict:
+def samgeo_fast_predict(
+        bbox: input_floatlist, zoom: float = ZOOM, model_name: str = MODEL_NAME, root_folder: str = ROOT, source_type: str = SOURCE_TYPE, crs="EPSG:4326"
+) -> dict:
     import tempfile
     from samgeo import tms_to_geotiff
     from samgeo.fast_sam import SamGeo
 
+    try:
+        os.environ['MPLCONFIGDIR'] = root_folder
+        get_system_info()
+    except Exception as e:
+        app_logger.error(f"Error while setting 'MPLCONFIGDIR':{e}.")
+
     with tempfile.NamedTemporaryFile(prefix=f"{source_type}_", suffix=".tif", dir=root_folder) as image_input_tmp:
         app_logger.info(f"start tms_to_geotiff using bbox:{bbox}, type:{type(bbox)}, download image to {image_input_tmp.name} ...")
         for coord in bbox:
             app_logger.info(f"coord:{coord}, type:{type(coord)}.")
 
         # bbox: image input coordinate
-        tms_to_geotiff(output=image_input_tmp.name, bbox=bbox, zoom=zoom, source=source_type, overwrite=True)
+        tms_to_geotiff(output=image_input_tmp.name, bbox=bbox, zoom=zoom, source=source_type, overwrite=True, crs=crs)
         app_logger.info(f"geotiff created, start to initialize SamGeo instance (read model {model_name} from {root_folder})...")
 
         predictor = SamGeo(

src/utilities/utilities.py

@@ -1,4 +1,5 @@
 """Various utilities (logger, time benchmark, args dump, numerical and stats info)"""
+from src import app_logger
 
 
 def is_base64(sb):
@@ -17,6 +18,22 @@ def is_base64(sb):
         return False
 
 
+def get_system_info():
+    import multiprocessing
+    import torch.multiprocessing as mp
+    import os
+    import subprocess
+
+    app_logger.info(f"mp::cpu_count:{mp.cpu_count()}.")
+    app_logger.info(f"multiprocessing::cpu_count:{multiprocessing.cpu_count()}.")
+    app_logger.info(f"os::cpu_count:{os.cpu_count()}")
+    app_logger.info(f"os::sched_getaffinity:{len(os.sched_getaffinity(0))}")
+    lscpu_output = subprocess.run("/usr/bin/lscpu", capture_output=True)
+    app_logger.info(f"lscpu:{lscpu_output.stdout.decode('utf-8')}.")
+    free_mem_output = subprocess.run(["/usr/bin/free", "-m"], capture_output=True)
+    app_logger.info(f"free_mem_output:{free_mem_output.stdout.decode('utf-8')}.")
+
+
 def base64_decode(s):
     import base64