Spaces:
Sleeping
Sleeping
File size: 3,364 Bytes
4cae98e 33a22a1 6082315 33a22a1 6082315 33a22a1 4cae98e 33a22a1 23e48c8 33a22a1 6082315 33a22a1 6082315 33a22a1 6082315 33a22a1 6082315 33a22a1 6082315 33a22a1 6082315 23e48c8 f9c215a 33a22a1 4cae98e 33a22a1 4cae98e 6082315 33a22a1 6082315 33a22a1 6082315 4cae98e 33a22a1 23e48c8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
import io
import rasterio
from rasterio.mask import mask
from shapely import wkt as load_wkt
import matplotlib.pyplot as plt
import gradio as gr
import leafmap.foliumap as leafmap
import tempfile
import os
# -------------------------------
# Helper: Get raster bounds
# -------------------------------
def get_raster_bounds(file):
try:
with rasterio.open(file.name) as src:
bounds = src.bounds
return f"{bounds.left},{bounds.bottom},{bounds.right},{bounds.top}"
except Exception as e:
return f"❌ Error: {str(e)}"
# -------------------------------
# Clip TIFF (from upload)
# -------------------------------
def clip_tiff_from_upload(file, bbox):
try:
minx, miny, maxx, maxy = map(float, bbox.split(","))
wkt = f"POLYGON(({minx} {miny}, {minx} {maxy}, {maxx} {maxy}, {maxx} {miny}, {minx} {miny}))"
geom = load_wkt.loads(wkt)
with rasterio.open(file.name) as src:
clipped, transform = mask(src, [geom], crop=True)
meta = src.meta.copy()
meta.update({
"driver": "GTiff",
"height": clipped.shape[1],
"width": clipped.shape[2],
"transform": transform
})
# Save clipped file temporarily
out_path = tempfile.mktemp(suffix=".tif")
with rasterio.open(out_path, "w", **meta) as dest:
dest.write(clipped)
# Preview image
fig, ax = plt.subplots(figsize=(5, 5))
ax.imshow(clipped[0], cmap="terrain")
ax.axis("off")
buf = io.BytesIO()
plt.savefig(buf, format="png", bbox_inches="tight")
buf.seek(0)
plt.close(fig)
return buf, out_path, "✅ Clipped image generated successfully"
except Exception as e:
return None, None, f"❌ Error: {str(e)}"
# -------------------------------
# Visualize clipped TIFF with Leafmap
# -------------------------------
def visualize_leafmap(tif_path):
try:
m = leafmap.Map(center=[0, 0], zoom=2)
m.add_raster(tif_path, layer_name="Clipped TIFF", colormap="terrain")
return m.to_html()
except Exception as e:
return f"<p>❌ Leafmap Error: {str(e)}</p>"
# -------------------------------
# Build Gradio UI
# -------------------------------
with gr.Blocks() as demo:
gr.Markdown("## 🌍 GeoTIFF Clipping Tool (Upload + Leafmap)")
with gr.Tab("From Upload"):
raster_file = gr.File(label="Upload GeoTIFF", file_types=[".tif"])
bbox_file = gr.Textbox(label="Bounding Box (minx,miny,maxx,maxy)", value="")
get_bounds_btn = gr.Button("📐 Get Raster Bounds")
get_bounds_btn.click(get_raster_bounds, [raster_file], [bbox_file])
output_image = gr.Image(label="Clipped TIFF Preview")
download_file = gr.File(label="Download Clipped TIFF")
status = gr.Textbox(label="Status")
leafmap_view = gr.HTML(label="Leafmap Visualization")
submit = gr.Button("✂️ Clip & Visualize")
submit.click(
clip_tiff_from_upload,
[raster_file, bbox_file],
[output_image, download_file, status]
).then(
visualize_leafmap,
[download_file],
[leafmap_view]
)
if __name__ == "__main__":
demo.launch()
|