Updated to include a tone suggestion

app.py

@@ -12,16 +12,19 @@ def load_csv(file):
     global current_row
     # import the csv and set the data types to be int, string, string, string, string, string, string
     df = pd.read_csv(file.name, dtype={'id':int, 'hs': str, 'cs': str, 'topic': str, 'tone': str, 'isCSContextuallyRelevant': str, 'isToneMatch': str})
+    if 'suggestedTone' not in df.columns:
+        df['suggestedTone'] = None
     current_row = 0
     row_dict = df.iloc[current_row].to_dict()
-    return row_dict['id'], row_dict['hs'], row_dict['cs'], row_dict['topic'], row_dict['tone'], row_dict['isCSContextuallyRelevant'], row_dict['isToneMatch']
+    return row_dict['id'], row_dict['hs'], row_dict['cs'], row_dict['topic'], row_dict['tone'], row_dict['isCSContextuallyRelevant'], row_dict['isToneMatch'], row_dict['suggestedTone']
 
-def annotate_row(isCSContextuallyRelevant, isToneMatch):
+def annotate_row(isCSContextuallyRelevant, isToneMatch, suggestedTone):
     global df
     global current_row
 
     df.at[current_row, 'isCSContextuallyRelevant'] = isCSContextuallyRelevant
     df.at[current_row, 'isToneMatch'] = isToneMatch
+    df.at[current_row, 'suggestedTone'] = suggestedTone
 
     if current_row < len(df) - 1:
         current_row += 1
@@ -30,7 +33,7 @@ def annotate_row(isCSContextuallyRelevant, isToneMatch):
     df.to_csv('annotated_data.csv', index=False)
     
     row_dict = df.iloc[current_row].to_dict()
-    return row_dict['id'], row_dict['hs'], row_dict['cs'], row_dict['topic'], row_dict['tone'], row_dict['isCSContextuallyRelevant'], row_dict['isToneMatch'], 'annotated_data.csv'
+    return row_dict['id'], row_dict['hs'], row_dict['cs'], row_dict['topic'], row_dict['tone'], row_dict['isCSContextuallyRelevant'], row_dict['isToneMatch'], row_dict['suggestedTone'], 'annotated_data.csv'
 
 def navigate(direction):
     global current_row
@@ -44,7 +47,7 @@ def navigate(direction):
             current_row = int(unlabeled_row)
 
     row_dict = df.iloc[current_row].to_dict()
-    return row_dict['id'], row_dict['hs'], row_dict['cs'], row_dict['topic'], row_dict['tone'], row_dict['isCSContextuallyRelevant'], row_dict['isToneMatch']
+    return row_dict['id'], row_dict['hs'], row_dict['cs'], row_dict['topic'], row_dict['tone'], row_dict['isCSContextuallyRelevant'], row_dict['isToneMatch'], row_dict['suggestedTone']
 
 with gr.Blocks(theme=gr.themes.Soft()) as annotator:
     gr.Markdown("## Data Annotation")
@@ -68,6 +71,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as annotator:
         with gr.Row():
             isCSContextuallyRelevant = gr.Radio(["1", "0"], label="Contextually Relevant?")
             isToneMatch = gr.Radio(["1", "0"], label="Tone Match?")
+            suggestedTone = gr.Textbox(label='Suggested Tone', interactive=True)
             btn_annotate = gr.Button("Annotate")
 
         with gr.Row():
@@ -79,10 +83,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as annotator:
         gr.Markdown("### Annotated Data File Download")
         file_download = gr.File()
 
-    btn_load.click(load_csv, inputs=[file_upload], outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch])
-    btn_annotate.click(annotate_row, inputs=[isCSContextuallyRelevant, isToneMatch], outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch, file_download])
-    btn_previous.click(navigate, inputs=gr.Textbox("Previous", visible=False), outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch])
-    btn_next.click(navigate, inputs=gr.Textbox("Next", visible=False), outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch])
-    btn_first_unlabeled.click(navigate, inputs=gr.Textbox("First Unlabeled", visible=False), outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch])
+    btn_load.click(load_csv, inputs=[file_upload], outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch, suggestedTone])
+    btn_annotate.click(annotate_row, inputs=[isCSContextuallyRelevant, isToneMatch, suggestedTone], outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch, suggestedTone, file_download])
+    btn_previous.click(navigate, inputs=gr.Textbox("Previous", visible=False), outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch, suggestedTone])
+    btn_next.click(navigate, inputs=gr.Textbox("Next", visible=False), outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch, suggestedTone])
+    btn_first_unlabeled.click(navigate, inputs=gr.Textbox("First Unlabeled", visible=False), outputs=[idx, hs, cs, topic, tone, isCSContextuallyRelevant, isToneMatch, suggestedTone])
 
 annotator.launch()