fix kml multi-polygon issue

app.py

@@ -6,6 +6,9 @@ import geopandas as gpd
 import leafmap.foliumap as leafmap
 from optree import tree_map
 from shapely.ops import transform
+import kml2geojson
+from io import BytesIO, StringIO
+import requests
 
 
 def shape_3d_to_2d(shape):
@@ -66,6 +69,7 @@ if ("file_url" in st.session_state) and ("input_gdf" in st.session_state) and (s
     input_gdf = st.session_state.input_gdf
 else:
     st.session_state.file_url = file_url
+    print(file_url, type(file_url))
     if isinstance(file_url, str):
         if file_url.startswith("https://drive.google.com/file/d/"):
             ID = file_url.replace("https://drive.google.com/file/d/", "").split("/")[0]
@@ -74,7 +78,22 @@ else:
             ID = file_url.replace("https://drive.google.com/open?id=", "")
             file_url = f"https://drive.google.com/uc?id={ID}"
 
-    input_gdf = preprocess_gdf(gpd.read_file(file_url))
+        response = requests.get(file_url)
+        bytes_data = BytesIO(response.content)
+        string_data = response.text
+    else:
+        bytes_data = BytesIO(file_url.getvalue())
+        string_data = file_url.getvalue().decode("utf-8")
+
+    if string_data.startswith("<?xml"):
+        geojson = kml2geojson.convert(bytes_data)
+        features = geojson[0]["features"]
+        epsg = 4326
+        input_gdf = gpd.GeoDataFrame.from_features(features, crs=f"EPSG:{epsg}")
+    else:
+        input_gdf = gpd.read_file(bytes_data)
+
+    input_gdf = preprocess_gdf(input_gdf)
     if len(input_gdf) > 1:
         st.warning(f"Only the first polygon in the KML will be processed; all other geometries will be ignored.")
 

requirements.txt

@@ -1,3 +1,4 @@
 leafmap
 geopandas
 optree
+kml2geojson

sandbox.ipynb

@@ -0,0 +1,100 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import pandas as pd\n",
+    "import geopandas as gpd\n",
+    "from tqdm.notebook import tqdm\n",
+    "from joblib import Parallel, delayed"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "e90c443ceec74b3e9bf223587db76898",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "  0%|          | 0/15 [00:00<?, ?it/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "https://drive.google.com/open?id=1skBV8miaZN3gE2lPVnHQQG-n-RwbMH2d has 5 geometries\n",
+      "https://drive.google.com/open?id=12OJzdi0RE5BDdMHzBovnYK_QhC5jIiSy has 6 geometries\n",
+      "https://drive.google.com/open?id=1lge4nZKw7gxuDbQtM_v7D8eBHzUsLoBI has 3 geometries\n"
+     ]
+    }
+   ],
+   "source": [
+    "files = \"\"\"https://drive.google.com/open?id=1QHbcxHRw3EQWsw5-UN6AUnVI-bhmhCQ5\n",
+    "https://drive.google.com/open?id=1skBV8miaZN3gE2lPVnHQQG-n-RwbMH2d\n",
+    "https://drive.google.com/open?id=1jyG_D6UudyAgicZjsvEfNLT_vNK1G3EB\n",
+    "https://drive.google.com/open?id=1NxO399MvzufeT6-xigq3qi79psA40rkQ\n",
+    "https://drive.google.com/open?id=1P2ZPMsxX82P12m460EBjpYMrdXUkZ05l\n",
+    "https://drive.google.com/open?id=1Pc4sIYbRcu761fuXV5HVDG3EaqQC-uiz\n",
+    "https://drive.google.com/open?id=1xskOyh888HrxCrB0Ue9bNGG6r1CvlH3J\n",
+    "https://drive.google.com/open?id=12OJzdi0RE5BDdMHzBovnYK_QhC5jIiSy\n",
+    "https://drive.google.com/open?id=18bDqdNIZOlp97GHpEOaYsCLSTML--aEA\n",
+    "https://drive.google.com/open?id=1lge4nZKw7gxuDbQtM_v7D8eBHzUsLoBI\n",
+    "https://drive.google.com/open?id=1XzzPh1q0uU3eMPoSC5stCLC7r8oMtgok\n",
+    "https://drive.google.com/open?id=1JGmGCVYjdcUeFVmcoa7LcJE36HH0MGoF\n",
+    "https://drive.google.com/open?id=1EPp6DZUGFcUESQSeMjQ9famygnWM-3BM\n",
+    "https://drive.google.com/open?id=1XY7HbmY0fakAaZaxvUFF5UYk7Qovnit2\n",
+    "https://drive.google.com/open?id=1VSVOpw9TFAxPKClHGPFXI_anBSHZOIi-\"\"\"\n",
+    "\n",
+    "df = pd.DataFrame(columns=['Failed URL', '# of geometries'])\n",
+    "i = 0\n",
+    "def get_gdf(link):\n",
+    "    link = link.strip()\n",
+    "    original_link = link\n",
+    "    if link.startswith('https://drive.google.com/open?id='):\n",
+    "        link = 'https://drive.google.com/uc?id=' + link.split('=')[-1]\n",
+    "    gdf = gpd.read_file(link)\n",
+    "    if len(gdf) != 1:\n",
+    "        print(f\"{original_link} has {len(gdf)} geometries\")\n",
+    "        # df.loc[i, ['Failed URL', '# of geometries']] = [original_link, len(gdf)]\n",
+    "        # i += 1\n",
+    "        \n",
+    "# df.to_csv('failed_links.csv', index=False)\n",
+    "_ = Parallel(n_jobs=32)(delayed(get_gdf)(link) for link in tqdm(files.split('\\n')))"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "zeel_py310",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}