Update app.py

app.py

@@ -208,7 +208,7 @@ def fill_missing_quarters(df, lob, acc, transaction):
         l_df = df[df[lob] == l]
         l_quarters = set(quarters) - set(l_df[acc])
         l_missing_df = pd.DataFrame({acc: list(l_quarters),
-                                      transaction: ['202212'] * len(l_quarters)})
+                                      transaction: [str(end_year)+'12'] * len(l_quarters)})
         for col in df.columns: # Fill the missing
             if col != lob: # These two checks are nesscary in case we are filling for the premium then we only fill it with the missing quarters without the 202212 for transactions
                 if col == acc:
@@ -232,7 +232,7 @@ def fill_missing_quarters(df, lob, acc, transaction):
     filled_df = filled_df.reset_index(drop=True)
     # Print the filled quarters or a message if there are no missing quarters
     filled_quarters = filled_df[acc].unique()
-    filtered_quarters = [q for q in filled_quarters if q[:4] in ['2017', '2018', '2019', '2020', '2021', '2022']]
+    filtered_quarters = [q for q in filled_quarters if q[:4] in [str(year1) for year1 in range(start_year, end_year + 1)]]
     if len(filtered_quarters) == 0:
         print("No missing quarters between 2017-2022")
     else: