Update modules/visuals.py

modules/visuals.py

@@ -22,10 +22,10 @@ def display_map_heatmap(df, location):
     
     # Map AlertLevel to sizes, colors, and styles
     df = df.copy()
-    df["AlertColor"] = df["AlertLevel"].map({"Green": "#00FF00", "Yellow": "#FFFF00", "Red": "#FF0000"})  # Green to Red gradient
+    df["MarkerColor"] = df["AlertLevel"].map({"Green": "lime", "Yellow": "gold", "Red": "red"})
     df["MarkerSize"] = df["AlertLevel"].map({"Green": 10, "Yellow": 15, "Red": 20})
     df["MarkerSymbol"] = df["AlertLevel"].map({"Green": "circle", "Yellow": "circle", "Red": "star"})
-    df["MarkerOpacity"] = df["AlertLevel"].map({"Green": 0.7, "Yellow": 0.8, "Red": 1.0})
+    df["MarkerOpacity"] = 0.9  # Increased opacity for visibility
     
     # Create scatter map with dark theme
     fig = px.scatter_mapbox(
@@ -33,10 +33,10 @@ def display_map_heatmap(df, location):
         lat="Latitude",
         lon="Longitude",
         color="AlertLevel",
-        color_discrete_map={"Green": "#00FF00", "Yellow": "#FFFF00", "Red": "#FF0000"},
+        color_discrete_map={"Green": "lime", "Yellow": "gold", "Red": "red"},
         size="MarkerSize",
         size_max=25,
-        zoom=15 if location == "Hyderabad" else 11,  # Higher zoom for Hyderabad vacant space
+        zoom=15 if location == "Hyderabad" else 11,
         hover_data={
             "PoleID": True,
             "RFID": True,
@@ -52,14 +52,20 @@ def display_map_heatmap(df, location):
     )
     fig.update_traces(
         marker=dict(
+            color=df["MarkerColor"],  # Explicitly set marker color
             symbol=df["MarkerSymbol"],
             opacity=df["MarkerOpacity"],
-            color=df["AlertColor"]
+            line=dict(width=2, color="white")  # Add white outline for contrast
         )
     )
     fig.update_layout(
         mapbox_style="carto-darkmatter",
         margin={"r":0,"t":50,"l":0,"b":0},
-        showlegend=True
+        showlegend=True,
+        legend=dict(
+            itemsizing="constant",
+            bgcolor="rgba(0, 0, 0, 0.7)",
+            font=dict(color="white")
+        )
     )
     st.plotly_chart(fig, use_container_width=True)