Update app.py

app.py

@@ -39,16 +39,14 @@ def process_dem(dem_path):
     plt.savefig(risk_map_path, dpi=150, bbox_inches="tight")
     plt.close(fig2d)
 
-    # --- INTERACTIVE 3D DEM (Plotly, fixed aspect ratio) ---
+       # --- INTERACTIVE 3D DEM (Simplified, works like Matplotlib) ---
     step = max(1, nrows // 200)
 
     fig3d = go.Figure()
 
-    # Base DEM
+    # Base DEM (terrain)
     fig3d.add_trace(go.Surface(
         z=Z[::step, ::step],
-        x=X[::step, ::step],
-        y=Y[::step, ::step],
         colorscale="Earth",
         showscale=True,
         opacity=0.9,
@@ -58,43 +56,22 @@ def process_dem(dem_path):
     # 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
     ))
 
-    # --- 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
+    # Fix proportions
     fig3d.update_layout(
         title="Interactive 3D DEM with Contours & Steep Slope Highlight",
         scene=dict(
-            xaxis_title="Longitude",
-            yaxis_title="Latitude",
+            xaxis_title="X (grid cols)",
+            yaxis_title="Y (grid rows)",
             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
+            aspectmode="data"
         )
     )
 
-    return risk_map_path, fig3d
-
-
-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 (Correct Aspect)")
-    ],
-    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()