Encapsulate df creation

wsgi.py

@@ -36,15 +36,12 @@ def get_tables(url: str) -> tuple[pd.DataFrame, pd.DataFrame]:
     return df, df_state
 
 
-def plot(
-    current_df: pd.DataFrame | None = None,
-    data_folder: str = "data",
-    sheet_name: str = "digital",
-    total_target: int = 1500,
-    alpha: float | None = None,
-) -> str:
-    curr_datetime = datetime.datetime.now()
-
+def create_plot_df(
+    current_df: pd.DataFrame | None,
+    data_folder: str,
+    sheet_name: str,
+    curr_datetime,
+) -> pd.DataFrame:
     data_dict = {}
 
     ## Important: If multiple results are stored for the same date
@@ -79,6 +76,23 @@ def plot(
         if pd.isna(df.loc[df.index.max()][0]):
             df = df.drop([df.index.max()])
 
+    return df
+
+def plot(
+    current_df: pd.DataFrame | None = None,
+    data_folder: str = "data",
+    sheet_name: str = "digital",
+    total_target: int = 1500,
+    alpha: float | None = None,
+) -> str:
+    curr_datetime = datetime.datetime.now()
+    df = create_plot_df(
+        current_df=current_df,
+        data_folder=data_folder,
+        sheet_name=sheet_name,
+        curr_datetime=curr_datetime,
+    )
+
     fig = plt.figure(dpi=300)
 
     # fill weekends