Update modules/visuals.py

modules/visuals.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import plotly.express as px
+import plotly.graph_object as go
 import pandas as pd
 
 def display_dashboard(df, location):
@@ -19,6 +20,14 @@ def display_map_heatmap(df, location):
     if df.empty:
         st.warning("No data available for this location.")
         return
+
+# Example DataFrame with location data
+df = pd.DataFrame({
+    'latitude': [17.385044, 17.444418],
+    'longitude': [78.486671, 78.348397],
+    'alert_level': ['red', 'yellow'],
+    'location': ['Location A', 'Location B']
+})
     
     # Debug: Print DataFrame to verify coordinates
     st.write("Debug: Sample Data", df[["Latitude", "Longitude", "AlertLevel"]].head())  # Temporary debug
@@ -72,4 +81,25 @@ def display_map_heatmap(df, location):
             traceorder="normal"
         )
     )
+fig.update_layout(
+    mapbox=dict(
+        style="open-street-map",  # Base style
+        center=dict(lat=17.385044, lon=78.486671),
+        zoom=12  # Zoom to an appropriate level
+    )
+)
+
+fig.show()
+fig = go.Figure(go.Scattermapbox(
+    lat=df['latitude'],
+    lon=df['longitude'],
+    mode='markers',
+    marker=go.scattermapbox.Marker(
+        size=14,  # Increased size
+        color=df['alert_level'],  # Colors based on alert level
+        colorscale='YlOrRd',  # Set a color scale
+        opacity=0.9  # Increased opacity for visibility
+    ),
+    text=df['location'],
+))
     st.plotly_chart(fig, use_container_width=True)