Update main.py

main.py

@@ -82,11 +82,11 @@ def process_data(input_data: CoachingCodeInput):
         # Label the columns manually
         journal_df.columns = ['user_id', 'productivity_yes_no', 'productivity_rate']
         panic_button_df.columns = ['user_id', 'panic_button']
-
+        print("Journal data processed")
         # Step 2: Merge Journal and Panic Button data
         panic_button_grouped = panic_button_df.groupby('user_id')['panic_button'].apply(lambda x: ','.join(x)).reset_index()
         merged_journal_panic = pd.merge(journal_df, panic_button_grouped, on='user_id', how='outer')
-
+        print("Panic data processed")
         # Step 3: Process Test data
         test_data = []
         for index, row in test_df.iterrows():
@@ -100,6 +100,7 @@ def process_data(input_data: CoachingCodeInput):
                 i += 2
 
         test_df_processed = pd.DataFrame(test_data)
+        print("test data processed")
 
         # Step 4: Merge all data
         merged_data = pd.merge(merged_journal_panic, test_df_processed, on='user_id', how='outer')