Update app.py via AI Editor

app.py

@@ -50,11 +50,11 @@ API_KEY = os.getenv('ACCUWEATHER_API_KEY')
 BASE_URL = "http://dataservice.accuweather.com"
 
 INDEX_IDS = {
-    "Health": 31,
-    "Environmental": 34,
-    "Pollen": 11,
-    "Mosquito": 50,
-    "Pests": 53
+    "Health": 10,
+    "Environmental": 5,
+    "Pollen": 30,
+    "Mosquito": 59,
+    "Pest": 61
 }
 
 def get_location_key(lat, lon):
@@ -268,6 +268,42 @@ def create_environmental_api_card(environmental_data):
         ])
     ], className="mb-4")
 
+def create_environmental_indices_card(indices_dict):
+    items = []
+    index_display_names = {
+        "Health": "Health API",
+        "Environmental": "Environmental API",
+        "Pollen": "Pollen",
+        "Pest": "Pest",
+        "Mosquito": "Mosquito"
+    }
+    for key in ["Health", "Environmental", "Pollen", "Pest", "Mosquito"]:
+        data = indices_dict.get(key)
+        display_name = index_display_names[key]
+        if data and isinstance(data, list) and len(data) > 0:
+            d = data[0]
+            items.append(
+                html.Div([
+                    html.H6(display_name, style={"marginBottom": "0.25rem", "fontWeight": "bold"}),
+                    html.P(f"Category: {d.get('Category', 'N/A')}", style={"marginBottom": "0.25rem"}),
+                    html.P(f"Value: {d.get('Value', 'N/A')}", style={"marginBottom": "0.25rem"}),
+                    html.P(f"Text: {d.get('Text', 'N/A')}", style={"marginBottom": "0.75rem"})
+                ])
+            )
+        else:
+            items.append(
+                html.Div([
+                    html.H6(display_name, style={"marginBottom": "0.25rem", "fontWeight": "bold"}),
+                    html.P("No data available.", style={"marginBottom": "0.75rem"})
+                ])
+            )
+    return dbc.Card([
+        dbc.CardBody([
+            html.H4("Environmental Indices", className="card-title"),
+            *items
+        ])
+    ], className="mb-4")
+
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 server = app.server
 
@@ -287,6 +323,7 @@ app.layout = dbc.Container([
                 type="default",
                 children=[
                     html.Div(id="current-weather-output"),
+                    html.Div(id="environmental-indices-output"),
                     html.Div(id="environmental-api-output")
                 ],
                 style={"width": "100%"}
@@ -342,6 +379,7 @@ def set_session_cookie(response):
 @app.callback(
     [
         Output("current-weather-output", "children"),
+        Output("environmental-indices-output", "children"),
         Output("environmental-api-output", "children"),
         Output("forecast-output", "children"),
     ],
@@ -355,10 +393,10 @@ def update_weather(location, session_data):
     if not location or 'error' in location:
         error_message = location.get('error', 'Waiting for location data...') if location else 'Waiting for location data...'
         logger.warning(f"Session {session_id} waiting for location: {error_message}")
-        return [dbc.Spinner(color="primary"), "", ""]
+        return [dbc.Spinner(color="primary"), "", "", ""]
 
     lat, lon = location["latitude"], location["longitude"]
-    results = {"current": "", "environmental": "", "forecast": ""}
+    results = {"current": "", "indices": "", "environmental": "", "forecast": ""}
     def fetch_weather_data():
         try:
             location_key = get_data_from_session(session_id, "location_key")
@@ -372,16 +410,23 @@ def update_weather(location, session_data):
             forecast_5day = get_forecast_5day(location_key)
             environmental_data = get_indices_1day(location_key, INDEX_IDS["Environmental"])
 
+            # Environmental Indices
+            indices_dict = {}
+            for name, idx in INDEX_IDS.items():
+                indices_dict[name] = get_indices_1day(location_key, idx)
+
             if current is None or forecast_5day is None:
                 raise ValueError("Failed to fetch weather data")
 
             results["current"] = create_current_weather_card(current)
+            results["indices"] = create_environmental_indices_card(indices_dict)
             results["environmental"] = create_environmental_api_card(environmental_data)
             results["forecast"] = create_forecast_5day_card(forecast_5day)
             save_session_data(session_id, "weather_results", results)
         except Exception as e:
             logger.error(f"Session {session_id} error: {str(e)}")
             results["current"] = ""
+            results["indices"] = ""
             results["environmental"] = ""
             results["forecast"] = dbc.Card([
                 dbc.CardBody([
@@ -396,11 +441,12 @@ def update_weather(location, session_data):
     if weather_results:
         return [
             weather_results.get("current", ""),
+            weather_results.get("indices", ""),
             weather_results.get("environmental", ""),
             weather_results.get("forecast", ""),
         ]
     else:
-        return [dbc.Spinner(color="primary"), "", ""]
+        return [dbc.Spinner(color="primary"), "", "", ""]
 
 if __name__ == '__main__':
     print("Starting the Dash application...")