Add drias indicators

climateqa/engine/talk_to_data/config.py

@@ -51,6 +51,23 @@ DRIAS_MODELS = [
     'REMO2009_MPI-ESM-LR',
     'CCLM4-8-17_HadGEM2-ES'
 ]
+# Mapping between indicator columns and their units
+INDICATOR_TO_UNIT = {
+    "total_winter_precipitation": "mm",
+    "total_summer_precipitation": "mm",
+    "total_annual_precipitation": "mm",
+    "total_remarkable_daily_precipitation": "mm",
+    "frequency_of_remarkable_daily_precipitation": "days",
+    "extreme_precipitation_intensity": "mm",
+    "mean_winter_temperature": "°C",
+    "mean_summer_temperature": "°C",
+    "mean_annual_temperature": "°C",
+    "number_tropical_nights": "days",
+    "maximum_summer_temperature": "°C",
+    "number_of_days_with_tx_above_30": "days",
+    "number_of_days_with_tx_above_35": "days",
+    "number_of_days_with_dry_ground": "days"
+}
 
 DRIAS_UI_TEXT = """
 Hi, I'm **Talk to Drias**, designed to answer your questions using [**DRIAS - TRACC 2023**](https://www.drias-climat.fr/accompagnement/sections/401) data.  

climateqa/engine/talk_to_data/plot.py

@@ -9,6 +9,9 @@ from climateqa.engine.talk_to_data.sql_query import (
     indicator_for_given_year_query,
     indicator_per_year_at_location_query,
 )
+from climateqa.engine.talk_to_data.config import INDICATOR_TO_UNIT
+
+
 
 
 class Plot(TypedDict):
@@ -58,6 +61,7 @@ def plot_indicator_evolution_at_location(params: dict) -> Callable[..., Figure]:
     indicator = params["indicator_column"]
     location = params["location"]
     indicator_label = " ".join([word.capitalize() for word in indicator.split("_")])
+    unit = INDICATOR_TO_UNIT.get(indicator, "")
 
     def plot_data(df: pd.DataFrame) -> Figure:
         """Generates the actual plot from the data.
@@ -111,6 +115,7 @@ def plot_indicator_evolution_at_location(params: dict) -> Callable[..., Figure]:
             name=f"Yearly {indicator_label}",
             mode="lines",
             marker=dict(color="#1f77b4"),
+            hovertemplate=f"{indicator_label}: %{{y:.2f}} {unit}<br>Year: %{{x}}<extra></extra>"
         )
 
         # Sliding average dashed line
@@ -121,11 +126,12 @@ def plot_indicator_evolution_at_location(params: dict) -> Callable[..., Figure]:
             name="10 years rolling average",
             line=dict(dash="dash"),
             marker=dict(color="#d62728"),
+            hovertemplate=f"10-year average: %{{y:.2f}} {unit}<br>Year: %{{x}}<extra></extra>"
         )
         fig.update_layout(
             title=f"Plot of {indicator_label} in {location} ({model_label})",
             xaxis_title="Year",
-            yaxis_title=indicator_label,
+            yaxis_title=f"{indicator_label} ({unit})",
             template="plotly_white",
         )
         return fig
@@ -161,6 +167,8 @@ def plot_indicator_number_of_days_per_year_at_location(
     """
     indicator = params["indicator_column"]
     location = params["location"]
+    indicator_label = " ".join([word.capitalize() for word in indicator.split("_")])
+    unit = INDICATOR_TO_UNIT.get(indicator, "")
 
     def plot_data(df: pd.DataFrame) -> Figure:
         """Generate the figure thanks to the dataframe
@@ -195,15 +203,14 @@ def plot_indicator_number_of_days_per_year_at_location(
                 y=indicators,
                 width=0.5,
                 marker=dict(color="#1f77b4"),
+                hovertemplate=f"{indicator_label}: %{{y:.2f}} {unit}<br>Year: %{{x}}<extra></extra>"
             )
         )
 
-        indicator_label = " ".join([word.capitalize() for word in indicator.split("_")])
-
         fig.update_layout(
             title=f"{indicator_label} in {location} ({model_label})",
             xaxis_title="Year",
-            yaxis_title=indicator,
+            yaxis_title=f"{indicator_label} ({unit})",
             yaxis=dict(range=[0, max(indicators)]),
             bargap=0.5,
             template="plotly_white",
@@ -243,6 +250,7 @@ def plot_distribution_of_indicator_for_given_year(
     indicator = params["indicator_column"]
     year = params["year"]
     indicator_label = " ".join([word.capitalize() for word in indicator.split("_")])
+    unit = INDICATOR_TO_UNIT.get(indicator, "")
 
     def plot_data(df: pd.DataFrame) -> Figure:
         """Generate the figure thanks to the dataframe
@@ -277,13 +285,14 @@ def plot_distribution_of_indicator_for_given_year(
                 opacity=0.8,
                 histnorm="percent",
                 marker=dict(color="#1f77b4"),
+                hovertemplate=f"{indicator_label}: %{{x:.2f}} {unit}<br>Frequency: %{{y:.2f}}%<extra></extra>"
             )
         )
 
         fig.update_layout(
             title=f"Distribution of {indicator_label} in {year} ({model_label})",
-            xaxis_title=indicator_label,
-            yaxis_title="Frequency",
+            xaxis_title=f"{indicator_label} ({unit})",
+            yaxis_title="Frequency (%)",
             plot_bgcolor="rgba(0, 0, 0, 0)",
             showlegend=False,
         )
@@ -322,6 +331,7 @@ def plot_map_of_france_of_indicator_for_given_year(
     indicator = params["indicator_column"]
     year = params["year"]
     indicator_label = " ".join([word.capitalize() for word in indicator.split("_")])
+    unit = INDICATOR_TO_UNIT.get(indicator, "")
 
     def plot_data(df: pd.DataFrame) -> Figure:
         fig = go.Figure()
@@ -358,6 +368,8 @@ def plot_map_of_france_of_indicator_for_given_year(
                     cmax=max(indicators),  # Maximum color range
                     showscale=True,  # Show colorbar
                 ),
+                text=[f"{indicator_label}: {value:.2f} {unit}" for value in indicators],  # Add hover text showing the indicator value
+                hoverinfo="text"  # Only show the custom text on hover
             )
         )
 
@@ -365,7 +377,7 @@ def plot_map_of_france_of_indicator_for_given_year(
             mapbox_style="open-street-map",  # Use OpenStreetMap
             mapbox_zoom=3,
             mapbox_center={"lat": 46.6, "lon": 2.0},
-            coloraxis_colorbar=dict(title=f"{indicator_label}"),  # Add legend
+            coloraxis_colorbar=dict(title=f"{indicator_label} ({unit})"),  # Add legend
             title=f"{indicator_label} in {year} in France ({model_label}) " # Title
         )
         return fig

style.css

@@ -671,3 +671,26 @@ div#tab-vanna{
 #table-names thead{
     display: none;
 }
+
+/* DRIAS Data Table Styles */
+#vanna-table {
+    height: 400px !important;
+    overflow-y: auto !important;
+}
+
+#vanna-table > div[class*="table"] {
+    height: 400px !important;
+    overflow-y: None !important;
+}
+
+#vanna-table .table-wrap {
+    height: 400px !important;
+    overflow-y: None !important;
+}
+
+#vanna-table thead {
+    position: sticky;
+    top: 0;
+    background: white;
+    z-index: 1;
+}