Update app.py

app.py

@@ -26,42 +26,50 @@ def process_dem(dem_path):
     threshold = np.percentile(slope, 95)  # Top 5% steepest slopes
     risk_mask = slope > threshold
 
-    # --- RISK MAP (2D) ---
+    # --- 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()])
+    ax.contour(risk_mask, levels=[0.5], colors="blue", linewidths=0.8,
+               extent=[x.min(), x.max(), y.min(), y.max()])
     plt.colorbar(c, ax=ax, label="Slope (steepness)")
-    ax.set_title("Slope Risk Map (Hot colors = Steeper slopes, Blue = Risk zones)")
+    ax.set_title("Slope Risk Map (Hot = Steep, Blue = Risk zones)")
     ax.set_xlabel("Longitude")
     ax.set_ylabel("Latitude")
     risk_map_path = "risk_map.png"
     plt.savefig(risk_map_path, dpi=150, bbox_inches="tight")
     plt.close(fig2d)
 
-    # --- INTERACTIVE 3D MODEL (Plotly) ---
-    step = max(1, nrows // 200)  # downsample for speed
-    fig3d = go.Figure(data=[
-        go.Surface(
-            z=Z[::step, ::step],
-            x=X[::step, ::step],
-            y=Y[::step, ::step],
-            colorscale="Earth",
-            showscale=True,
-            opacity=0.9
-        ),
-        go.Surface(
-            z=np.where(risk_mask[::step, ::step], Z[::step, ::step], np.nan),
-            x=X[::step, ::step],
-            y=Y[::step, ::step],
-            surfacecolor=np.ones_like(Z[::step, ::step]),
-            colorscale=[[0, "purple"], [1, "purple"]],
-            showscale=False,
-            opacity=0.6
+    # --- INTERACTIVE 3D DEM (Plotly version of Matplotlib style) ---
+    step = max(1, nrows // 200)
+
+    fig3d = go.Figure()
+
+    # Base DEM surface
+    fig3d.add_trace(go.Surface(
+        z=Z[::step, ::step],
+        x=X[::step, ::step],
+        y=Y[::step, ::step],
+        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),
+        x=X[::step, ::step],
+        y=Y[::step, ::step],
+        surfacecolor=np.ones_like(Z[::step, ::step]),
+        colorscale=[[0, "purple"], [1, "purple"]],
+        showscale=False,
+        opacity=0.6
+    ))
 
     fig3d.update_layout(
-        title="Interactive 3D DEM with Landslide Risk Zones",
+        title="Interactive 3D DEM with Contours & Steep Slope Highlight",
         scene=dict(
             xaxis_title="Longitude",
             yaxis_title="Latitude",
@@ -78,11 +86,12 @@ demo = gr.Interface(
     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")
+        gr.Plot(label="Interactive 3D DEM (Matplotlib Style)")
     ],
     title="3D DEM & Landslide Risk Visualizer",
-    description="Upload a GeoTIFF DEM file to see a 2D slope risk map and an interactive 3D terrain with steep slope zones highlighted."
+    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()
+