Update app.py

app.py

@@ -1,29 +1,29 @@
 import dash
-from dash import dcc, html, dash_table
+from dash import dcc, html
 import pandas as pd
 import numpy as np
 import plotly.graph_objects as go
 import random
 
-# Constants
 SITES = ['Hyderabad', 'Kurnool', 'Bangalore', 'Warangal']
 POLES_PER_SITE = 12
-ALERT_THRESHOLD = 80  # Poles above this are red alert
+ALERT_THRESHOLD = 80
 
-# Generate dummy data
 def generate_data():
-    data = []
-    for site in SITES:
-        for pole in range(1, POLES_PER_SITE + 1):
-            status = random.randint(0, 100)
-            data.append({'Site': site, 'Pole': f'P{pole}', 'Status': status})
-    return pd.DataFrame(data)
+    return pd.DataFrame([
+        {
+            'Site': site,
+            'Pole': f'P{pole}',
+            'Status': random.randint(0, 100)
+        }
+        for site in SITES
+        for pole in range(1, POLES_PER_SITE + 1)
+    ])
 
 df = generate_data()
 
-# Create heatmap figure
-def create_heatmap(dataframe):
-    pivot_df = dataframe.pivot(index="Site", columns="Pole", values="Status")
+def create_heatmap(df):
+    pivot_df = df.pivot(index="Site", columns="Pole", values="Status")
     fig = go.Figure(data=go.Heatmap(
         z=pivot_df.values,
         x=pivot_df.columns,
@@ -36,34 +36,27 @@ def create_heatmap(dataframe):
     fig.update_layout(title="Pole Status Heatmap", height=500)
     return fig
 
-# Dash App
 app = dash.Dash(__name__)
-server = app.server  # Needed for Hugging Face Spaces
-
-# Blinking style (CSS)
-blinking_style = {
-    "animation": "blinker 1s linear infinite",
-    "color": "red",
-    "fontWeight": "bold",
-    "fontSize": "20px",
-    "margin": "10px"
-}
+server = app.server
 
 app.layout = html.Div([
     html.H1("Pole Management Dashboard", style={'textAlign': 'center'}),
-    
     dcc.Graph(figure=create_heatmap(df)),
     
     html.Div([
         html.Div("🚨 Red Alert Poles:", style={"fontWeight": "bold", "marginTop": "20px"}),
         html.Div([
-            html.Span(f"{row['Site']} - {row['Pole']}", style=blinking_style)
+            html.Span(f"{row['Site']} - {row['Pole']}", style={
+                "animation": "blinker 1s linear infinite",
+                "color": "red",
+                "fontWeight": "bold",
+                "fontSize": "20px",
+                "margin": "10px"
+            })
             for _, row in df[df["Status"] > ALERT_THRESHOLD].iterrows()
         ])
     ], style={"textAlign": "center"}),
 
-    # Add CSS for blinking effect
-    html.Div([
     dcc.Markdown('''
         <style>
         @keyframes blinker {
@@ -73,7 +66,6 @@ app.layout = html.Div([
     ''', dangerously_allow_html=True)
 ])
 
-])
-
 if __name__ == "__main__":
     app.run_server(debug=False, host="0.0.0.0", port=7860)
+