Update modules/visuals.py

modules/visuals.py

@@ -25,18 +25,18 @@ def display_map_heatmap(df, location):
     
     # Map AlertLevel to sizes, colors, and styles
     df = df.copy()
-    df["MarkerColor"] = df["AlertLevel"].map({"Green": "chartreuse", "Yellow": "gold", "Red": "darkred"})
+    df["MarkerColor"] = df["AlertLevel"].map({"Green": "green", "Yellow": "yellow", "Red": "red"})
     df["MarkerSize"] = df["AlertLevel"].map({"Green": 20, "Yellow": 25, "Red": 35})
     df["MarkerSymbol"] = df["AlertLevel"].map({"Green": "circle", "Yellow": "circle", "Red": "star"})
     df["MarkerOpacity"] = 1.0  # Full opacity for maximum visibility
     
-    # Create scatter map with lighter theme for testing visibility
+    # Create scatter map with lighter theme
     fig = px.scatter_mapbox(
         df,
         lat="Latitude",
         lon="Longitude",
         color="AlertLevel",
-        color_discrete_map={"Green": "chartreuse", "Yellow": "gold", "Red": "darkred"},
+        color_discrete_map={"Green": "green", "Yellow": "yellow", "Red": "red"},
         size="MarkerSize",
         size_max=35,
         zoom=15 if location == "Hyderabad" else 11,
@@ -55,20 +55,21 @@ def display_map_heatmap(df, location):
     )
     fig.update_traces(
         marker=dict(
-            color=df["MarkerColor"],  # Explicitly set marker color to override defaults
+            color=df["MarkerColor"],  # Explicitly set marker color
             symbol=df["MarkerSymbol"],
             opacity=df["MarkerOpacity"],
-            size=df["MarkerSize"]  # Ensure size is applied
+            size=df["MarkerSize"]
         )
     )
     fig.update_layout(
-        mapbox_style="open-street-map",  # Temporarily switched to lighter theme
+        mapbox_style="open-street-map",
         margin={"r": 0, "t": 50, "l": 0, "b": 0},
         showlegend=True,
         legend=dict(
             itemsizing="constant",
             bgcolor="rgba(0, 0, 0, 0.7)",
-            font=dict(color="white")
+            font=dict(color="white"),
+            traceorder="normal"  # Ensure legend order matches data
         )
     )
     st.plotly_chart(fig, use_container_width=True)