gera-richarte commited on
Commit
f453698
·
1 Parent(s): 805f17c

feat(app): added google maps link

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -2,6 +2,7 @@ from datasets import load_dataset
2
  from functools import partial
3
  from pandas import DataFrame
4
  import earthview as ev
 
5
  import gradio as gr
6
  import tqdm
7
  import os
@@ -58,8 +59,8 @@ def get_images(batch_size, only_rgb, state):
58
  item = next(state["dsi"])
59
  except StopIteration:
60
  break
61
- metadata = item["metadata"]
62
  item = ev.item_to_images(subset, item)
 
63
 
64
  if subset == "satellogic":
65
  images.extend(item["rgb"])
@@ -72,8 +73,9 @@ def get_images(batch_size, only_rgb, state):
72
  if not only_rgb:
73
  images.extend(item["chm"])
74
  images.extend(item["1m"])
75
-
76
- metadatas.append(item["metadata"])
 
77
 
78
  return images, DataFrame(metadatas)
79
 
@@ -90,7 +92,7 @@ if __name__ == "__main__":
90
  gr.Markdown(f"# Viewer for [{ev.DATASET}](https://huggingface.co/datasets/satellogic/EarthView) Dataset")
91
  batch_size = gr.Number(10, label = "Batch Size", render=False)
92
  shard = gr.Slider(label="Shard", minimum=0, maximum=10000, step=1, render=False)
93
- table = gr.DataFrame(render = False)
94
  # headers=["Index","TimeStamp","Bounds","CRS"],
95
 
96
  gallery = gr.Gallery(
 
2
  from functools import partial
3
  from pandas import DataFrame
4
  import earthview as ev
5
+ import utils
6
  import gradio as gr
7
  import tqdm
8
  import os
 
59
  item = next(state["dsi"])
60
  except StopIteration:
61
  break
 
62
  item = ev.item_to_images(subset, item)
63
+ metadata = item["metadata"]
64
 
65
  if subset == "satellogic":
66
  images.extend(item["rgb"])
 
73
  if not only_rgb:
74
  images.extend(item["chm"])
75
  images.extend(item["1m"])
76
+
77
+ metadata["map"] = f'<a href="{utils.get_google_map_link(item, subset)}" target="about:_blank">🧭</a>'
78
+ metadatas.append(metadata)
79
 
80
  return images, DataFrame(metadatas)
81
 
 
92
  gr.Markdown(f"# Viewer for [{ev.DATASET}](https://huggingface.co/datasets/satellogic/EarthView) Dataset")
93
  batch_size = gr.Number(10, label = "Batch Size", render=False)
94
  shard = gr.Slider(label="Shard", minimum=0, maximum=10000, step=1, render=False)
95
+ table = gr.DataFrame(render = False, datatype="html")
96
  # headers=["Index","TimeStamp","Bounds","CRS"],
97
 
98
  gallery = gr.Gallery(