Update app.py

app.py

@@ -8,18 +8,13 @@ def process_dem(dem_path):
     # --- OPEN DEM ---
     with rasterio.open(dem_path.name) as src:
         dem = src.read(1).astype(float)
-        transform = src.transform
-        res_x, res_y = transform.a, -transform.e  # pixel size
 
-    # --- BUILD GRID ---
     nrows, ncols = dem.shape
-    x = np.arange(ncols) * res_x + transform.c
-    y = np.arange(nrows) * res_y + transform.f
-    X, Y = np.meshgrid(x, y)
     Z = dem
 
     # --- COMPUTE SLOPE ---
-    dy, dx = np.gradient(Z, res_y, res_x)
+    # Assuming square pixels (ok for visualization)
+    dy, dx = np.gradient(Z)
     slope = np.sqrt(dx**2 + dy**2)
 
     # --- RISK MASK ---
@@ -28,26 +23,25 @@ def process_dem(dem_path):
 
     # --- 2D RISK MAP ---
     fig2d, ax = plt.subplots(figsize=(8, 6))
-    c = ax.imshow(slope, cmap="hot", extent=[x.min(), x.max(), y.min(), y.max()], origin="upper")
-    ax.contour(risk_mask, levels=[0.5], colors="blue", linewidths=0.8,
-               extent=[x.min(), x.max(), y.min(), y.max()])
+    c = ax.imshow(slope, cmap="hot", origin="upper")
+    ax.contour(risk_mask, levels=[0.5], colors="blue", linewidths=0.8)
     plt.colorbar(c, ax=ax, label="Slope (steepness)")
     ax.set_title("Slope Risk Map (Hot = Steep, Blue = Risk zones)")
-    ax.set_xlabel("Longitude")
-    ax.set_ylabel("Latitude")
+    ax.set_xlabel("Column Index (X)")
+    ax.set_ylabel("Row Index (Y)")
     risk_map_path = "risk_map.png"
     plt.savefig(risk_map_path, dpi=150, bbox_inches="tight")
     plt.close(fig2d)
 
-       # --- INTERACTIVE 3D DEM (Simplified, works like Matplotlib) ---
+    # --- INTERACTIVE 3D DEM (Plotly) ---
     step = max(1, nrows // 200)
 
     fig3d = go.Figure()
 
-    # Base DEM (terrain)
+    # Base DEM surface
     fig3d.add_trace(go.Surface(
         z=Z[::step, ::step],
-        colorscale="Earth",
+        colorscale="Earth",  # similar to matplotlib 'terrain'
         showscale=True,
         opacity=0.9,
         contours=dict(z=dict(show=True, usecolormap=True, highlightcolor="black", project_z=True))
@@ -62,7 +56,6 @@ def process_dem(dem_path):
         opacity=0.6
     ))
 
-    # Fix proportions
     fig3d.update_layout(
         title="Interactive 3D DEM with Contours & Steep Slope Highlight",
         scene=dict(
@@ -73,5 +66,20 @@ 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"]),
+    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."
+)
+
 if __name__ == "__main__":
     demo.launch()