import os
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import sys

data = pd.read_csv(
    "../assets/datasets/time_series_solar.csv",
    parse_dates=["Datetime"],
    index_col="Datetime",
)
series = data["Incoming Solar"]

if len(sys.argv) == 2 and sys.argv[1] == "pandas":
    print("Plotting with pandas")
    series.plot(figsize=(12, 6), title="Solar radiation time series")
    plt.show()
else:
    print("Plotting with seaborn (sns)")

    series_df = series.reset_index()

    plt.rcParams["figure.figsize"] = [12, 6]

    sns.set_theme(style="darkgrid")
    sns.lineplot(data=series_df, x="Datetime", y="Incoming Solar")

    plt.ylabel("Solar Radiation")
    plt.xlabel("")
    plt.title("Solar radiation time series")

    os.makedirs('assets', exist_ok=True)
    plt.savefig('assets/timeseries_plot.png')