area444 commited on
Commit
ed4f224
·
verified ·
1 Parent(s): 311796a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -120,12 +120,12 @@ def update_table(start_date, end_date, window, user_text):
120
  time.sleep(30)
121
 
122
  print(f"Run ID: {run_id}")
123
- data_payments = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_payments.csv")
124
- data_metrics = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_metrics.csv")
125
- data_predictions = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_predictions.csv")
126
- df_payments = pd.read_csv(StringIO(data_payments.decode('utf-8')))
127
- new_columns = ['vintage', 'vintage_unique_cases'] + [f'M{col}' for col in df_payments.columns[2:]]
128
- df_metrics = pd.read_csv(StringIO(data_metrics.decode('utf-8')))
129
  df_predictions = pd.read_csv(StringIO(data_predictions.decode('utf-8')))
130
  df_predictions['revenue'] = (df_predictions['vintage_unique_cases'] / df_predictions['predicted_monthly_payment_rate']).round(2)
131
  df_predictions['Month'] = [get_month_label(i) for i in range(len(df_predictions))]
 
120
  time.sleep(30)
121
 
122
  print(f"Run ID: {run_id}")
123
+ #data_payments = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_payments.csv")
124
+ #data_metrics = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_metrics.csv")
125
+ data_predictions = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_predictions"+str(str(user_text).split('@'))[0]+".csv")
126
+ #df_payments = pd.read_csv(StringIO(data_payments.decode('utf-8')))
127
+ #new_columns = ['vintage', 'vintage_unique_cases'] + [f'M{col}' for col in df_payments.columns[2:]]
128
+ #df_metrics = pd.read_csv(StringIO(data_metrics.decode('utf-8')))
129
  df_predictions = pd.read_csv(StringIO(data_predictions.decode('utf-8')))
130
  df_predictions['revenue'] = (df_predictions['vintage_unique_cases'] / df_predictions['predicted_monthly_payment_rate']).round(2)
131
  df_predictions['Month'] = [get_month_label(i) for i in range(len(df_predictions))]