Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import rasterio
 import numpy as np
 import matplotlib.pyplot as plt
+import plotly.graph_objs as go
 
 def process_dem(dem_path):
     # --- OPEN DEM ---
@@ -25,7 +26,7 @@ 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,
@@ -38,36 +39,50 @@ def process_dem(dem_path):
     plt.savefig(risk_map_path, dpi=150, bbox_inches="tight")
     plt.close(fig2d)
 
-    # --- 3D DEM (Matplotlib as original) ---
-    fig3d = plt.figure(figsize=(12, 9))
-    ax = fig3d.add_subplot(111, projection="3d")
+    # --- INTERACTIVE 3D DEM (Plotly, fixed aspect ratio) ---
+    step = max(1, nrows // 200)
+
+    fig3d = go.Figure()
+
+    # Base DEM
+    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))
+    ))
 
-    step = max(1, nrows // 300)  # downsample for speed
-    # Base DEM surface
-    surf = ax.plot_surface(
-        X[::step, ::step], Y[::step, ::step], Z[::step, ::step],
-        cmap="terrain", linewidth=0, alpha=0.9
-    )
-    # Contours
-    ax.contour(
-        X[::step, ::step], Y[::step, ::step], Z[::step, ::step],
-        levels=20, colors="k", linewidths=0.4, offset=np.nanmin(Z)
-    )
     # Risk overlay (purple)
-    ax.plot_surface(
-        X[::step, ::step], Y[::step, ::step],
-        np.where(risk_mask[::step, ::step], Z[::step, ::step], np.nan),
-        color="purple", alpha=0.6, zorder=10
+    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
+    ))
+
+    # --- FIX DISTORTION: Set aspect ratio ---
+    xrange = X.max() - X.min()
+    yrange = Y.max() - Y.min()
+    zrange = Z.max() - Z.min()
+    # Scale so Z looks correct compared to XY
+    fig3d.update_layout(
+        title="Interactive 3D DEM with Contours & Steep Slope Highlight",
+        scene=dict(
+            xaxis_title="Longitude",
+            yaxis_title="Latitude",
+            zaxis_title="Elevation (m)",
+            aspectmode="manual",
+            aspectratio=dict(x=xrange/yrange, y=1, z=zrange/yrange*2)  # tweak multiplier if Z looks too flat/steep
+        )
     )
-    ax.set_xlabel("Longitude")
-    ax.set_ylabel("Latitude")
-    ax.set_zlabel("Elevation (m)")
-    plt.title("3D DEM with Contours & Steep Slope Highlight")
-    dem3d_path = "dem3d.png"
-    plt.savefig(dem3d_path, dpi=150, bbox_inches="tight")
-    plt.close(fig3d)
 
-    return risk_map_path, dem3d_path
+    return risk_map_path, fig3d
 
 
 demo = gr.Interface(
@@ -75,10 +90,10 @@ demo = gr.Interface(
     inputs=gr.File(label="Upload DEM (.tif)", file_types=[".tif"]),
     outputs=[
         gr.Image(type="filepath", label="2D Slope Risk Map"),
-        gr.Image(type="filepath", label="3D DEM Visualization")
+        gr.Plot(label="Interactive 3D DEM (Correct Aspect)")
     ],
     title="3D DEM & Landslide Risk Visualizer",
-    description="Upload a GeoTIFF DEM file to see a 2D slope risk map and a static 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 with contours & steep slope zones highlighted."
 )
 
 if __name__ == "__main__":