small fix

visualize/visualize.py

@@ -62,8 +62,16 @@ def water_deficit(df, latitude, longitude, shading_coef=0, historic=True):
 
 
 def concatenate_historic_forecast(
-    historic, forecast, cols_to_keep, value_period_col="forecast scénario modéré"
-):
+    historic, forecast, cols_to_keep, value_period_col="forecast scénario modéré", don_t = False
+):  
+    if don_t:
+        historic["period"] = value_period_col
+        forecast["period"] = value_period_col
+        historic = historic[cols_to_keep]
+        forecast = forecast[cols_to_keep]
+        full_data = pd.concat([historic, forecast])
+        full_data =   full_data[full_data['year']>=2025]
+        return full_data
     historic["period"] = "historique"
     forecast["period"] = value_period_col
     historic = historic[cols_to_keep]
@@ -87,9 +95,14 @@ def visualize_climate(
         "period",
     ],
 ):
-    concatenated_moderate = concatenate_historic_forecast(
-        historic, moderate, cols_to_keep
-    )
+    if column == "Water Deficit (mm/day)":
+        concatenated_moderate = concatenate_historic_forecast(
+            historic, moderate, cols_to_keep,don_t=True
+        )
+    else:
+        concatenated_moderate = concatenate_historic_forecast(
+            historic, moderate, cols_to_keep
+        )
     concatenated_moderate = concatenated_moderate.sort_values(by=x_axis)  # Ensure order
 
     fig = go.Figure()
@@ -111,10 +124,14 @@ def visualize_climate(
                     ),
                 )
             )
-
-        concatenated_pessimist = concatenate_historic_forecast(
-            historic, pessimist, cols_to_keep, "forecast scénario pessimiste"
-        )
+        if column == "Water Deficit (mm/day)":
+            concatenated_pessimist = concatenate_historic_forecast(
+                historic, pessimist, cols_to_keep, "forecast scénario pessimiste",don_t=True
+            )
+        else:
+            concatenated_pessimist = concatenate_historic_forecast(
+                historic, pessimist, cols_to_keep, "forecast scénario pessimiste"
+            )
         concatenated_pessimist = concatenated_pessimist.sort_values(by=x_axis)
         concatenated_pessimist = concatenated_pessimist[
             concatenated_pessimist["period"] != "historique"
@@ -184,9 +201,15 @@ def visualize_climate(
                 )
 
         # Continue with pessimistic data as in the original function...
-        concatenated_pessimist = concatenate_historic_forecast(
-            historic, pessimist, cols_to_keep, "forecast scénario pessimiste"
-        )
+        if column == "Water Deficit (mm/day)":
+            concatenated_pessimist = concatenate_historic_forecast(
+                historic, pessimist, cols_to_keep, "forecast scénario pessimiste",don_t=True
+            )
+        else:
+            concatenated_pessimist = concatenate_historic_forecast(
+                historic, pessimist, cols_to_keep, "forecast scénario pessimiste"
+            )
+
         concatenated_pessimist = concatenated_pessimist.sort_values(by=x_axis)
         for condition_value in concatenated_pessimist["period"].unique():
             segment = concatenated_pessimist[