Update app.py

app.py

@@ -120,12 +120,12 @@ def update_table(start_date, end_date, window, user_text):
             time.sleep(30)
         
         print(f"Run ID: {run_id}")
-        data_payments = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_payments.csv")
-        data_metrics = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_metrics.csv")
-        data_predictions = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_predictions.csv")
-        df_payments = pd.read_csv(StringIO(data_payments.decode('utf-8')))
-        new_columns = ['vintage', 'vintage_unique_cases'] + [f'M{col}' for col in df_payments.columns[2:]] 
-        df_metrics = pd.read_csv(StringIO(data_metrics.decode('utf-8')))
+        #data_payments = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_payments.csv")
+        #data_metrics = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_metrics.csv")
+        data_predictions = get_databricks_file("dbfs:/dbfs/FileStore/forecast_alleviatetax_predictions"+str(str(user_text).split('@'))[0]+".csv")
+        #df_payments = pd.read_csv(StringIO(data_payments.decode('utf-8')))
+        #new_columns = ['vintage', 'vintage_unique_cases'] + [f'M{col}' for col in df_payments.columns[2:]] 
+        #df_metrics = pd.read_csv(StringIO(data_metrics.decode('utf-8')))
         df_predictions = pd.read_csv(StringIO(data_predictions.decode('utf-8')))
         df_predictions['revenue'] = (df_predictions['vintage_unique_cases'] / df_predictions['predicted_monthly_payment_rate']).round(2)
         df_predictions['Month'] = [get_month_label(i) for i in range(len(df_predictions))]