Upload app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-from interface_utils import *
+from gradio_interfaces.conversation_labeling.interface_utils import *
 
 maxim = 'quantity'
 submaxims = ["The response provides a sufficient amount of information.",
@@ -10,7 +10,7 @@ checkbox_choices = [
     ["Yes", "No", "NA"]
 ]
 
-conversation_data_sliced = load_from_jsonl('./data/conversations_unlabeled_sliced.jsonl')
+conversation_data_sliced = load_from_jsonl('../../../data/unlabeled/conversations_unlabeled_sliced.jsonl')
 max_conversation_length = max([len(conversation['transcript']) for conversation in conversation_data_sliced])
 
 conversation = get_conversation(conversation_data_sliced)
@@ -25,9 +25,9 @@ def save_labels(conv_id, slice_idx, skipped, submaxim_0=None, submaxim_1=None):
         'submaxim_0': submaxim_0,
         'submaxim_1': submaxim_1,
     }
-    os.makedirs("./labels", exist_ok=True)
+    os.makedirs("../labels", exist_ok=True)
 
-    with open(f"./labels/{maxim}_human_labels_{conv_id}_{slice_idx}.json", 'w') as f:
+    with open(f"../labels/{maxim}_human_labels_{conv_id}_{slice_idx}.json", 'w') as f:
         json.dump(data, f, indent=4)
 
 
@@ -39,19 +39,20 @@ def update_interface(new_conversation):
     markdown_blocks = [None] * max_conversation_length
     for i in range(max_conversation_length):
         if i < len(new_transcript) and new_transcript[i]['speaker'] != '':
-            markdown_blocks[i] = gr.Markdown(
-                f"""&nbsp;&nbsp;**{new_transcript[i]['speaker']}**: &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;{new_transcript[i]['response']}""",
-                visible=True)
+            if i < len(transcript) - 1:
+                markdown_blocks[i] = gr.Markdown(f"""&nbsp;&nbsp;**{new_transcript[i]['speaker']}**: &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;{new_transcript[i]['response']}""", visible=True)
+            if i == len(transcript) - 1:
+                markdown_blocks[i] = gr.Markdown(f"""&nbsp;&nbsp;**{transcript[i]['speaker']}**: &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;<mark style="background-color: lightyellow">{transcript[i]['response']}</mark>""", visible=True)
         else:
             markdown_blocks[i] = gr.Markdown("", visible=False)
 
-    new_last_response = gr.Text(value=get_last_response(new_transcript),
-                                label="",
-                                lines=1,
-                                container=False,
-                                interactive=False,
-                                autoscroll=True,
-                                visible=True)
+    # new_last_response = gr.Text(value=get_last_response(new_transcript),
+    #                             label="",
+    #                             lines=1,
+    #                             container=False,
+    #                             interactive=False,
+    #                             autoscroll=True,
+    #                             visible=True)
     new_radio_0_base = gr.Radio(label=submaxims[0],
                                 choices=checkbox_choices[0],
                                 value=None,
@@ -62,7 +63,7 @@ def update_interface(new_conversation):
                                 visible=True)
     conv_len = gr.Number(value=len(new_transcript), visible=False)
 
-    return [new_conv_id] + [new_slice_idx] + list(markdown_blocks) + [new_last_response] + [new_radio_0_base] + [new_radio_1_base] + [conv_len]
+    return [new_conv_id] + [new_slice_idx] + list(markdown_blocks) + [new_radio_0_base] + [new_radio_1_base] + [conv_len]
 
 
 def submit(*args):
@@ -95,12 +96,14 @@ with gr.Blocks(theme=gr.themes.Default()) as interface:
     markdown_blocks = [None] * max_conversation_length
     with gr.Column(scale=1, min_width=600):
         with gr.Group():
-            gr.Markdown("""<span style='font-size: 16px;'>&nbsp;&nbsp;&nbsp;&nbsp;**Conversational context** </span>""",
+            gr.Markdown("""<span style='font-size: 16px;'>&nbsp;&nbsp;&nbsp;&nbsp;**Conversation** </span>""",
                         visible=True)
         for i in range(max_conversation_length):
             if i < len(transcript):
-                markdown_blocks[i] = gr.Markdown(
-                    f"""&nbsp;&nbsp;**{transcript[i]['speaker']}**: &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;{transcript[i]['response']}""")
+                if i < len(transcript) - 1:
+                    markdown_blocks[i] = gr.Markdown(f"""&nbsp;&nbsp;**{transcript[i]['speaker']}**: &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;{transcript[i]['response']}""")
+                if i == len(transcript) - 1:
+                    markdown_blocks[i] = gr.Markdown(f"""&nbsp;&nbsp;**{transcript[i]['speaker']}**: &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;<mark style="background-color: lightyellow">{transcript[i]['response']}</mark>""")
             else:
                 markdown_blocks[i] = gr.Markdown("")
             if i >= conv_len.value:
@@ -109,15 +112,15 @@ with gr.Blocks(theme=gr.themes.Default()) as interface:
         with gr.Row():
             with gr.Group(elem_classes="bottom-aligned-group"):
                 speaker_adapted = gr.Markdown(
-                    f"""<span style='font-size: 16px;'>&nbsp;&nbsp;&nbsp;&nbsp;**Response to label** </span>""",
+                    f"""<span style='font-size: 16px;'>&nbsp;&nbsp;&nbsp;&nbsp;**Labels** </span>""",
                     visible=True)
-                last_response = gr.Textbox(value=get_last_response(transcript),
-                                           label="",
-                                           lines=1,
-                                           container=False,
-                                           interactive=False,
-                                           autoscroll=True,
-                                           visible=True)
+                # last_response = gr.Textbox(value=get_last_response(transcript),
+                #                            label="",
+                #                            lines=1,
+                #                            container=False,
+                #                            interactive=False,
+                #                            autoscroll=True,
+                #                            visible=True)
                 radio_submaxim_0_base = gr.Radio(label=submaxims[0],
                                                  choices=checkbox_choices[0],
                                                  value=None,
@@ -135,7 +138,6 @@ with gr.Blocks(theme=gr.themes.Default()) as interface:
     input_list = [conv_id_element] + \
                  [slice_idx_element] + \
                  markdown_blocks + \
-                 [last_response] + \
                  [radio_submaxim_0_base] + \
                  [radio_submaxim_1_base] + \
                  [conv_len]
@@ -145,7 +147,6 @@ with gr.Blocks(theme=gr.themes.Default()) as interface:
         outputs=[conv_id_element,
                  slice_idx_element,
                  *markdown_blocks,
-                 last_response,
                  radio_submaxim_0_base,
                  radio_submaxim_1_base,
                  conv_len]
@@ -156,7 +157,6 @@ with gr.Blocks(theme=gr.themes.Default()) as interface:
         outputs=[conv_id_element,
                  slice_idx_element,
                  *markdown_blocks,
-                 last_response,
                  radio_submaxim_0_base,
                  radio_submaxim_1_base,
                  conv_len]