Update main.py

main.py

@@ -485,19 +485,23 @@ def map_worldcover_to_terrain_simplified(code: int, lat: float, lon: float, elev
         return "Tundra"
     
     else:
-        # Unknown codes - SIMPLIFIED logic for water classification
+        # Unknown codes - IMPROVED logic for water classification when WorldCover fails
         if elevation is not None:
             if elevation >= -2:  # Onshore water body threshold
-                return "Water Body"  # Simplified onshore water
-            elif elevation < -2:  # Marine water
-                depth = -elevation
+                return "Water Body"
+            elif elevation < -2:  # Marine water - THIS IS THE KEY FIX
+                depth = -elevation  # Convert negative elevation to positive depth
                 if depth <= 200:
                     return "Continental Shelf"
                 else:
-                    return "Deep Ocean"
-            elif is_coastal and elevation <= 10:
+                    return "Deep Ocean"  # This should now work for deep ocean
+            elif is_coastal_location(lat, lon, elevation) and elevation <= 10:
                 return "Tidal Zone"
         
+        # If no elevation data available, try to infer from location
+        if is_coastal_location(lat, lon, elevation):
+            return "Tidal Zone"
+        
         return "Unknown"
 
 def detect_coastal_area(lat: float, lon: float, elevation: Optional[float], landcover_analysis: Optional[LandCoverAnalysis]) -> bool:
@@ -767,6 +771,22 @@ async def classify_terrain_enhanced_async(lat: float, lon: float, radius_meters:
     
     # SIMPLIFIED terrain classification logic
     final_terrain = terrain
+
+    if terrain == "Unknown" and elev is not None:
+        # Use elevation-only classification for ocean areas
+        if elev < -2:  # Marine water
+            depth = -elev
+            if depth <= 200:
+                terrain = "Continental Shelf"
+            else:
+                terrain = "Deep Ocean"
+        elif elev >= -2 and elev <= 0:
+            terrain = "Water Body"
+        elif is_coastal_location(lat, lon, elev) and elev <= 10:
+            terrain = "Tidal Zone"
+    
+    # Coastal detection
+    is_coastal = detect_coastal_area(lat, lon, elev, landcover_analysis)
     
     # Override terrain classification based on elevation for water bodies
     if elev is not None and terrain in ["Water Body", "Continental Shelf", "Deep Ocean"]: