Update app.py

app.py

@@ -14,13 +14,13 @@ from plotting import (
 # Function to load chunked data based on date range
 def load_timeseries_data(start_date, end_date):
     start_date_str = pd.to_datetime(start_date).strftime('%Y-%m-%d')
-    end_date_str = pd.to_datetime(start_date).strftime('%Y-%m') + '-30'
+    end_date_str = (pd.to_datetime(start_date) + pd.offsets.MonthEnd()).strftime('%Y-%m-%d')
     file_name = f"enel_timeseries_data_{start_date_str}_{end_date_str}.csv"
     return pd.read_csv(file_name, index_col=0, header=[0, 1, 2], parse_dates=True)
 
 def load_other_features_data(start_date, end_date):
     start_date_str = pd.to_datetime(start_date).strftime('%Y-%m-%d')
-    end_date_str = pd.to_datetime(start_date).strftime('%Y-%m') + '-30'
+    end_date_str = (pd.to_datetime(start_date) + pd.offsets.MonthEnd()).strftime('%Y-%m-%d')
     file_name = f"enel_other_features_data_{start_date_str}_{end_date_str}.csv"
     return pd.read_csv(file_name, index_col=0, header=0, parse_dates=True)
 
@@ -40,7 +40,7 @@ date_range_picker = pn.widgets.DateRangePicker(
     name='Date Range', 
     start=start_date, 
     end=end_date,
-    value=(start_date, (pd.Timestamp(start_date) + pd.DateOffset(months=1)).date())
+    value=(None, None)
 )
 
 inverter_ids = pn.widgets.MultiSelect(
@@ -75,32 +75,34 @@ loading_spinner = pn.indicators.LoadingSpinner(width=50, height=50)
     date_range_picker.param.value
 )
 def update_iv_plot(inverter_ids, plot_power_curves, plot_temperature_curves, date_range):
+    if date_range[0] is None or date_range[1] is None:
+        return pn.pane.Markdown("Please select a date range to load data.")
+    
     start_date, end_date = date_range
     timeseries_data = load_timeseries_data(start_date, end_date)
     other_features_data = load_other_features_data(start_date, end_date)
+
+    if timeseries_data is None or other_features_data is None:
+        return pn.pane.Markdown("No data available for the selected date range.")
     
     if not inverter_ids:
         return pn.pane.Markdown("No Inverters selected for Plotting.")
     else:
         # Plot IV + Power + Temperature Curves
         if plot_power_curves and plot_temperature_curves:
-            print("Plotting IV + Power + Temperature Curves")
             return create_iv_plot_with_power_and_temperature_curves(
                 timeseries_data, other_features_data, inverter_ids
             )
         # Plot IV + Temperature Curves
         elif (not plot_power_curves) and plot_temperature_curves:
-            print("Plot IV + Temperature Curves")
             return create_iv_plot_with_temperature_curves(
                 timeseries_data, other_features_data, inverter_ids
             )
         # Plot IV + Power Curves
         elif plot_power_curves and (not plot_temperature_curves):
-            print("Plot IV + Power Curves")
             return create_iv_plot_with_power_curves(timeseries_data, inverter_ids)
         # Plot only IV Curves
         else:
-            print("Plot only IV Curves")
             return create_iv_plot(timeseries_data, inverter_ids)