Updated wrapper functions

main.py

@@ -336,13 +336,13 @@ Returns: DataFrame, filepath
 def random_schedules_wrapper(schedule_name: str, password: str):
     gr.Info('', ALERT_TIME, title='Working on generating schedules! Please do NOT click anything on this page.')
     res = generator(schedule_name, password, api_name='generate_random_schedules')
-    df = res[0]['value']
-    file = res[1]['value']
+    df = res[1]['value']
+    file = res[2]['value']
     if len(df['data'][0]) == 0: 
         gr.Warning('', ALERT_TIME, title='No one has filled out the form yet.')
         return gr.DataFrame(df, visible=False), gr.File(file, visible=False)
     else: 
-        gr.Info('', ALERT_TIME, title='Good news! I was able to make at least one schedule.')
+        gr.Info('', ALERT_TIME, title=res[0]['value'])
         return gr.DataFrame(df, visible=True), gr.File(file, visible=True)
 
 
@@ -356,13 +356,13 @@ Returns: DataFrame, filepath
 def all_schedules_wrapper(schedule_name: str, password: str): 
     gr.Info('', ALERT_TIME, title='Working on generating schedules! Please do NOT click anything on this page.')
     res = generator(schedule_name, password, api_name='generate_all_schedules')
-    df = res[0]['value']
-    file = res[1]['value']
+    df = res[1]['value']
+    file = res[2]['value']
     if len(df['data'][0]) == 0: 
         gr.Warning('', ALERT_TIME, title='No one has filled out the form yet.')
         return gr.DataFrame(df, visible=False), gr.File(file, visible=False)
     else: 
-        gr.Info('', ALERT_TIME, title='Good news! I was able to make at least one schedule.')
+        gr.Info('', ALERT_TIME, title=res[0]['value'])
         return gr.DataFrame(df, visible=True), gr.File(file, visible=True)