BananaSauce commited on
Commit
e7b03fa
·
1 Parent(s): b21c4dc

fix level_1 error

Browse files
Files changed (2) hide show
  1. app.py +5 -4
  2. multiple.py +5 -1
app.py CHANGED
@@ -75,13 +75,14 @@ def single_main(uploaded_file):
75
  else:
76
  grouped_filtered_scenarios = None
77
 
78
-
79
  grouped_filtered_scenarios.reset_index(inplace=True)
80
- grouped_filtered_scenarios.drop(columns=['level_1'], inplace=True)
 
 
 
 
81
  grouped_filtered_scenarios.index = grouped_filtered_scenarios.index + 1
82
-
83
  st.dataframe(grouped_filtered_scenarios)
84
-
85
  # Sort the average time spent table by start datetime
86
  average_time_spent_seconds = average_time_spent_seconds.sort_values(by='Start datetime')
87
 
 
75
  else:
76
  grouped_filtered_scenarios = None
77
 
 
78
  grouped_filtered_scenarios.reset_index(inplace=True)
79
+
80
+ # Only drop 'level_1' if it exists in the DataFrame
81
+ if 'level_1' in grouped_filtered_scenarios.columns:
82
+ grouped_filtered_scenarios.drop(columns=['level_1'], inplace=True)
83
+
84
  grouped_filtered_scenarios.index = grouped_filtered_scenarios.index + 1
 
85
  st.dataframe(grouped_filtered_scenarios)
 
86
  # Sort the average time spent table by start datetime
87
  average_time_spent_seconds = average_time_spent_seconds.sort_values(by='Start datetime')
88
 
multiple.py CHANGED
@@ -75,7 +75,11 @@ def perform_analysis(uploaded_dataframes):
75
  else:
76
  grouped_filtered_scenarios = None
77
  grouped_filtered_scenarios.reset_index(inplace=True)
78
- grouped_filtered_scenarios.drop(columns=['level_1'], inplace=True)
 
 
 
 
79
  grouped_filtered_scenarios.index = grouped_filtered_scenarios.index + 1
80
  st.dataframe(grouped_filtered_scenarios)
81
 
 
75
  else:
76
  grouped_filtered_scenarios = None
77
  grouped_filtered_scenarios.reset_index(inplace=True)
78
+
79
+ # Only drop 'level_1' if it exists in the DataFrame
80
+ if 'level_1' in grouped_filtered_scenarios.columns:
81
+ grouped_filtered_scenarios.drop(columns=['level_1'], inplace=True)
82
+
83
  grouped_filtered_scenarios.index = grouped_filtered_scenarios.index + 1
84
  st.dataframe(grouped_filtered_scenarios)
85