add labels

app.py

@@ -23,30 +23,32 @@ LEVELS = [1000, 925, 850, 700, 600, 500, 400, 300, 250, 200, 150, 100, 50]
 SOIL_LEVELS = [1, 2]
 DEFAULT_DATE = OpendataClient().latest()
 
-# First define the variable descriptions
-VARIABLE_DESCRIPTIONS = {
-    # Surface variables (10m)
-    "10u": "10m U Wind Component",
-    "10v": "10m V Wind Component",
-    "2d": "2m Dewpoint Temperature",
-    "2t": "2m Temperature",
-    "msl": "Mean Sea Level Pressure",
-    "skt": "Skin Temperature",
-    "sp": "Surface Pressure",
-    "tcw": "Total Column Water",
-    "lsm": "Land-Sea Mask",
-    "z": "Surface Geopotential",
-    "slor": "Slope of Sub-gridscale Orography",
-    "sdor": "Standard Deviation of Orography",
-    
-    # Soil variables
-    "stl1": "Soil Temperature Level 1",
-    "stl2": "Soil Temperature Level 2",
-    "swvl1": "Soil Water Volume Level 1",
-    "swvl2": "Soil Water Volume Level 2",
+# First organize variables into categories
+VARIABLE_GROUPS = {
+    "Surface Variables": {
+        "10u": "10m U Wind Component",
+        "10v": "10m V Wind Component",
+        "2d": "2m Dewpoint Temperature",
+        "2t": "2m Temperature",
+        "msl": "Mean Sea Level Pressure",
+        "skt": "Skin Temperature",
+        "sp": "Surface Pressure",
+        "tcw": "Total Column Water",
+        "lsm": "Land-Sea Mask",
+        "z": "Surface Geopotential",
+        "slor": "Slope of Sub-gridscale Orography",
+        "sdor": "Standard Deviation of Orography",
+    },
+    "Soil Variables": {
+        "stl1": "Soil Temperature Level 1",
+        "stl2": "Soil Temperature Level 2",
+        "swvl1": "Soil Water Volume Level 1",
+        "swvl2": "Soil Water Volume Level 2",
+    },
+    "Pressure Level Variables": {}  # Will fill this dynamically
 }
 
-# Add pressure level variable descriptions dynamically
+# Add pressure level variables dynamically
 for var in ["t", "u", "v", "w", "q", "z"]:
     var_name = {
         "t": "Temperature",
@@ -58,7 +60,8 @@ for var in ["t", "u", "v", "w", "q", "z"]:
     }[var]
     
     for level in LEVELS:
-        VARIABLE_DESCRIPTIONS[f"{var}_{level}"] = f"{var_name} at {level}hPa"
+        var_id = f"{var}_{level}"
+        VARIABLE_GROUPS["Pressure Level Variables"][var_id] = f"{var_name} at {level}hPa"
 
 def get_open_data(param, levelist=[]):
     fields = {}
@@ -118,27 +121,23 @@ def plot_forecast(state, selected_variable):
     ax.coastlines()
     ax.add_feature(cfeature.BORDERS, linestyle=":")
     triangulation = tri.Triangulation(longitudes, latitudes)
-    contour = ax.tricontourf(triangulation, values, levels=20, transform=ccrs.PlateCarree(), cmap="RdBu")
+    
+    # Use 'RdBu_r' instead of 'RdBu' to reverse the color scheme
+    contour = ax.tricontourf(triangulation, values, levels=20, 
+                            transform=ccrs.PlateCarree(), 
+                            cmap='RdBu_r')
     plt.title(f"{selected_variable} at {state['date']}")
     plt.colorbar(contour)
     return fig
 
-# Then create the available variables list
-AVAILABLE_VARIABLES = (
-    # Surface variables
-    ["10u", "10v", "2d", "2t", "msl", "skt", "sp", "tcw", "lsm", "z", "slor", "sdor"] +
-    # Soil variables
-    ["stl1", "stl2", "swvl1", "swvl2"] +
-    # Pressure level variables (adding level suffix)
-    [f"{var}_{level}" for var in ["t", "u", "v", "w", "q", "z"] 
-     for level in LEVELS]
-)
-
-# Finally create the dropdown choices
-DROPDOWN_CHOICES = [
-    (f"{VARIABLE_DESCRIPTIONS[var_id]} ({var_id})", var_id)
-    for var_id in sorted(AVAILABLE_VARIABLES)
-]
+# Create dropdown choices with groups
+DROPDOWN_CHOICES = []
+for group_name, variables in VARIABLE_GROUPS.items():
+    # Add group separator
+    DROPDOWN_CHOICES.append((f"── {group_name} ──", None))
+    # Add variables in this group
+    for var_id, desc in sorted(variables.items()):
+        DROPDOWN_CHOICES.append((f"{desc} ({var_id})", var_id))
 
 def gradio_interface(date_str, lead_time, device, selected_variable):
     try:
@@ -156,7 +155,7 @@ demo = gr.Interface(
         gr.Radio(choices=["cuda", "cpu"], value="cuda", label="Compute Device"),
         gr.Dropdown(
             choices=DROPDOWN_CHOICES,
-            value="t_850",  # This should be the variable ID
+            value="2t",  # Default to 2m temperature
             label="Select Variable to Plot",
             info="Choose a meteorological variable to visualize"
         )