Fix: Attempt to fix error with last index; Add new id

app.py

@@ -133,9 +133,9 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         # Go to the next example
         next_button.click(
             next_example,
-            inputs=[annotations, file_list,emotions, confidence, comments, n_clicks, participant_id, ann_completed, current_index],
+            inputs=[annotations, file_list,emotions, confidence, comments, n_clicks, participant_id, start, end, duration, ann_completed, current_index],
             outputs=[annotations,sentence_text, audio_player, emotions, confidence, comments, n_clicks, start, end, duration, ann_completed, 
-            current_index],).then(state_to_number, start, []).then(state_to_number, [start, end, duration, current_index, ann_completed, total], 
+            current_index],).then(state_to_number, [start, end, duration, current_index, ann_completed, total], 
             [start_temp, end_temp, duration_temp, current_index_temp, ann_completed_temp, total_temp]).then(None, [], 
             [start_temp, end_temp, duration_temp, current_index_temp, 
             ann_completed_temp, total_temp], js = js_progress_bar)

load_and_save.py

@@ -18,7 +18,8 @@ possible_ids = {'Tiger-001': 0, 'Falcon-002': 0,
                 'Leopard-009': 4, 'Panda-010': 4,
                 'Cheetah-011': 5, 'Gorilla-012': 5,
                 'Dolphin-013' : 6, 'Lynx-014': 6,
-                'Moose-015': 7, 'Raccoon-016': 7}
+                'Moose-015': 7, 'Raccoon-016': 7,
+                'Rabbit-017': 0}
 
 persistent_storage = Path('/data')
 password_files = os.getenv("password_files")
@@ -146,7 +147,7 @@ def save_annotation(annotations_df, file_list_df, emotions, confidence, comments
     
     return annotations_df, ann_completed
 
-def next_example(annotations_df, file_list_df, emotions, confidence, comments, n_clicks, participant_id, ann_completed, current_index):
+def next_example(annotations_df, file_list_df, emotions, confidence, comments, n_clicks, participant_id, start, end, duration, ann_completed, current_index):
     """Move to the next example.
 
     Parameters:
@@ -185,12 +186,13 @@ def next_example(annotations_df, file_list_df, emotions, confidence, comments, n
         annotations_df, ann_completed = save_annotation(annotations_df, file_list_df, emotions, confidence, comments, n_clicks, participant_id, ann_completed, current_index)
         if current_index < len(file_list_df):
             current_index += 1
+            sentence, audio_path, emotion, confidence, comments, n_clicks, start, end, duration = load_example(annotations_df, file_list_df, current_index)
         
         else:
            gr.Warning("This is the last example, well done!")
     print(f'current_index {current_index}')
-    sentence, audio_path, emotions, confidence, comments, clicks, start, end, duration = load_example(annotations_df, file_list_df, current_index)    
-    return annotations_df, *load_example(annotations_df, file_list_df, current_index), ann_completed, current_index
+    
+    return annotations_df, sentence, audio_path, emotion, confidence, comments, n_clicks, start, end, duration, ann_completed, current_index
 
 def previous_example(annotations_df, file_list_df, emotion, confidence, comments, n_clicks, participant_id,  ann_completed, current_index):
 
@@ -224,6 +226,7 @@ def previous_example(annotations_df, file_list_df, emotion, confidence, comments
 
     if emotion != "Blank":
         annotations_df, ann_completed = save_annotation(annotations_df, file_list_df, emotion, confidence, comments, n_clicks, participant_id,  ann_completed, current_index)
+        
     if current_index > 0:
         current_index -= 1