Update app.py

app.py

@@ -3,22 +3,31 @@ import rasterio
 import numpy as np
 import matplotlib.pyplot as plt
 import plotly.graph_objs as go
+from gradio.exceptions import Error
+from rasterio.errors import RasterioIOError
 
-def process_dem(dem_path):
-    # --- OPEN DEM ---
-    with rasterio.open(dem_path.name) as src:
-        dem = src.read(1).astype(float)
+def process_dem(dem_file):
+    # dem_file is a tempfile-like object from Gradio
+    path = dem_file.name
+
+    # Validate that it’s a GeoTIFF using rasterio (more robust than extension)
+    try:
+        with rasterio.open(path) as src:
+            if src.driver not in ("GTiff", "COG"):
+                raise Error(f"Unsupported raster driver: {src.driver}. Please upload a GeoTIFF (.tif/.tiff).")
+            dem = src.read(1).astype(float)
+    except RasterioIOError as e:
+        raise Error(f"Could not read the file as a GeoTIFF. {e}")
 
     nrows, ncols = dem.shape
     Z = dem
 
     # --- COMPUTE SLOPE ---
-    # Assuming square pixels (ok for visualization)
     dy, dx = np.gradient(Z)
     slope = np.sqrt(dx**2 + dy**2)
 
     # --- RISK MASK ---
-    threshold = np.percentile(slope, 95)  # Top 5% steepest slopes
+    threshold = np.percentile(slope, 95)
     risk_mask = slope > threshold
 
     # --- 2D RISK MAP ---
@@ -37,17 +46,13 @@ def process_dem(dem_path):
     step = max(1, nrows // 200)
 
     fig3d = go.Figure()
-
-    # Base DEM surface
     fig3d.add_trace(go.Surface(
         z=Z[::step, ::step],
-        colorscale="Earth",  # similar to matplotlib 'terrain'
+        colorscale="Earth",
         showscale=True,
         opacity=0.9,
         contours=dict(z=dict(show=True, usecolormap=True, highlightcolor="black", project_z=True))
     ))
-
-    # Risk overlay (purple)
     fig3d.add_trace(go.Surface(
         z=np.where(risk_mask[::step, ::step], Z[::step, ::step], np.nan),
         surfacecolor=np.ones_like(Z[::step, ::step]),
@@ -55,7 +60,6 @@ def process_dem(dem_path):
         showscale=False,
         opacity=0.6
     ))
-
     fig3d.update_layout(
         title="Interactive 3D DEM with Contours & Steep Slope Highlight",
         scene=dict(
@@ -68,17 +72,15 @@ def process_dem(dem_path):
 
     return risk_map_path, fig3d
 
-
-# --- GRADIO APP ---
 demo = gr.Interface(
     fn=process_dem,
-    inputs=gr.File(label="Upload DEM (.tif)", file_types=[".tif"]),
+    inputs=gr.File(label="Upload DEM (.tif or .tiff)", file_types=[".tif", ".tiff"]),  # <-- allow both
     outputs=[
         gr.Image(type="filepath", label="2D Slope Risk Map"),
         gr.Plot(label="Interactive 3D DEM (with Contours & Risk Zones)")
     ],
     title="3D DEM & Landslide Risk Visualizer",
-    description="Upload a GeoTIFF DEM file to see a 2D slope risk map and an interactive 3D DEM with contours & steep slope zones highlighted."
+    description="Upload a GeoTIFF DEM file to see a 2D slope risk map and an interactive 3D DEM."
 )
 
 if __name__ == "__main__":