DSatishchandra commited on
Commit
53d29fc
·
verified ·
1 Parent(s): bb65cb9

Update modules/visuals.py

Browse files
Files changed (1) hide show
  1. modules/visuals.py +9 -5
modules/visuals.py CHANGED
@@ -20,10 +20,13 @@ def display_map_heatmap(df, location):
20
  st.warning("No data available for this location.")
21
  return
22
 
 
 
 
23
  # Map AlertLevel to sizes, colors, and styles
24
  df = df.copy()
25
  df["MarkerColor"] = df["AlertLevel"].map({"Green": "chartreuse", "Yellow": "gold", "Red": "darkred"})
26
- df["MarkerSize"] = df["AlertLevel"].map({"Green": 15, "Yellow": 20, "Red": 30})
27
  df["MarkerSymbol"] = df["AlertLevel"].map({"Green": "circle", "Yellow": "circle", "Red": "star"})
28
  df["MarkerOpacity"] = 1.0 # Full opacity for maximum visibility
29
 
@@ -35,7 +38,7 @@ def display_map_heatmap(df, location):
35
  color="AlertLevel",
36
  color_discrete_map={"Green": "chartreuse", "Yellow": "gold", "Red": "darkred"},
37
  size="MarkerSize",
38
- size_max=30,
39
  zoom=15 if location == "Hyderabad" else 11,
40
  hover_data={
41
  "PoleID": True,
@@ -52,14 +55,15 @@ def display_map_heatmap(df, location):
52
  )
53
  fig.update_traces(
54
  marker=dict(
55
- color=df["MarkerColor"], # Explicitly set marker color
56
  symbol=df["MarkerSymbol"],
57
- opacity=df["MarkerOpacity"]
 
58
  )
59
  )
60
  fig.update_layout(
61
  mapbox_style="open-street-map", # Temporarily switched to lighter theme
62
- margin={"r": 0, "t": 50, "l": 0, "b": 0}, # Fixed formatting with proper spacing
63
  showlegend=True,
64
  legend=dict(
65
  itemsizing="constant",
 
20
  st.warning("No data available for this location.")
21
  return
22
 
23
+ # Debug: Print DataFrame to verify coordinates
24
+ st.write("Debug: Sample Data", df[["Latitude", "Longitude", "AlertLevel"]].head()) # Temporary debug
25
+
26
  # Map AlertLevel to sizes, colors, and styles
27
  df = df.copy()
28
  df["MarkerColor"] = df["AlertLevel"].map({"Green": "chartreuse", "Yellow": "gold", "Red": "darkred"})
29
+ df["MarkerSize"] = df["AlertLevel"].map({"Green": 20, "Yellow": 25, "Red": 35})
30
  df["MarkerSymbol"] = df["AlertLevel"].map({"Green": "circle", "Yellow": "circle", "Red": "star"})
31
  df["MarkerOpacity"] = 1.0 # Full opacity for maximum visibility
32
 
 
38
  color="AlertLevel",
39
  color_discrete_map={"Green": "chartreuse", "Yellow": "gold", "Red": "darkred"},
40
  size="MarkerSize",
41
+ size_max=35,
42
  zoom=15 if location == "Hyderabad" else 11,
43
  hover_data={
44
  "PoleID": True,
 
55
  )
56
  fig.update_traces(
57
  marker=dict(
58
+ color=df["MarkerColor"], # Explicitly set marker color to override defaults
59
  symbol=df["MarkerSymbol"],
60
+ opacity=df["MarkerOpacity"],
61
+ size=df["MarkerSize"] # Ensure size is applied
62
  )
63
  )
64
  fig.update_layout(
65
  mapbox_style="open-street-map", # Temporarily switched to lighter theme
66
+ margin={"r": 0, "t": 50, "l": 0, "b": 0},
67
  showlegend=True,
68
  legend=dict(
69
  itemsizing="constant",