Add application file

app.py

@@ -62,7 +62,9 @@ def get_current_comment(state):
 def update_display(state):
     example, comment_id = get_current_comment(state)
     if not comment_id:
-        return [gr.update(value="No examples!")] + [gr.update(visible=False)] * len(state["adapter_columns"]) * 3 + [gr.update(value="Complete!"), state]
+        return [gr.update(value="No examples!", visible=False)] + \
+               [gr.update(visible=False)] * len(state["adapter_columns"]) * 4 + \
+               [gr.update(value="Complete!"), state]
     
     existing_annotation = {}
     if example['manual_annotation']:
@@ -73,9 +75,11 @@ def update_display(state):
         text = example.get(adapter, "")
         scores = existing_annotation.get(adapter, {}) if existing_annotation else {}
         updates.extend([
-            gr.update(value=text, visible=True),
-            gr.update(value=scores.get("grammar"), visible=True),
-            gr.update(value=scores.get("coherence"), visible=True)
+            gr.update(value=text, visible=True),  # Text display
+            gr.update(visible=True),              # Grammar title
+            gr.update(value=scores.get("grammar"), visible=True),  # Grammar score
+            gr.update(visible=True),              # Coherence title
+            gr.update(value=scores.get("coherence"), visible=True)  # Coherence score
         ])
     
     return updates + [
@@ -83,6 +87,30 @@ def update_display(state):
         state
     ]
 
+# def update_display(state):
+#     example, comment_id = get_current_comment(state)
+#     if not comment_id:
+#         return [gr.update(value="No examples!")] + [gr.update(visible=False)] * len(state["adapter_columns"]) * 3 + [gr.update(value="Complete!"), state]
+    
+#     existing_annotation = {}
+#     if example['manual_annotation']:
+#         existing_annotation = json.loads(example['manual_annotation'])
+    
+#     updates = []
+#     for adapter in state["adapter_columns"]:
+#         text = example.get(adapter, "")
+#         scores = existing_annotation.get(adapter, {}) if existing_annotation else {}
+#         updates.extend([
+#             gr.update(value=text, visible=True),
+#             gr.update(value=scores.get("grammar"), visible=True),
+#             gr.update(value=scores.get("coherence"), visible=True)
+#         ])
+    
+#     return updates + [
+#         gr.update(value=f"Example {state['current_id_index'] + 1} of {len(state[state['current_view'] + '_ids'])} ({state['current_view']})"),
+#         state
+#     ]
+
 def navigate(direction, state):
     ids = state[state["current_view"] + "_ids"]
     max_index = len(ids) - 1
@@ -135,32 +163,70 @@ def submit(*args):
     
     return navigate("next", state)
 
+# with gr.Blocks() as app:
+#     state = gr.State(load_data())
+    
+#     gr.Markdown("## Text Annotation Tool")
+    
+#     with gr.Row():
+#         # Sidebar Column
+#         with gr.Column(scale=1, min_width=200, elem_classes="sidebar"):
+#             gr.Markdown("### Navigation")
+#             with gr.Column(elem_classes="nav-buttons"):
+#                 annotated_btn = gr.Button("Annotated", variant="primary")
+#                 unannotated_btn = gr.Button("Unannotated", variant="primary")
+#                 gr.Markdown("---")
+#                 counter = gr.Markdown()
+        
+#         # Main Content Column
+#         with gr.Column(scale=4):
+#             adapter_components = []
+#             for adapter in state.value["adapter_columns"]:
+#                 with gr.Row():
+#                     adapter_components.extend([
+#                         gr.Textbox(label=f"{adapter} Output", visible=False, interactive=False),
+#                         gr.Radio(choices=[1, 2, 3, 4, 5], label=f"{adapter} Grammar", visible=False),
+#                         gr.Radio(choices=[1, 2, 3, 4, 5], label=f"{adapter} Coherence", visible=False)
+#                     ])
+#                 gr.Markdown("---")
+            
+#             with gr.Row():
+#                 prev_btn = gr.Button("Previous", variant="secondary")
+#                 next_btn = gr.Button("Next", variant="secondary")
+#                 submit_btn = gr.Button("Submit", variant="primary")
+
 with gr.Blocks() as app:
     state = gr.State(load_data())
     
     gr.Markdown("## Text Annotation Tool")
     
     with gr.Row():
-        # Sidebar Column
-        with gr.Column(scale=1, min_width=200, elem_classes="sidebar"):
-            gr.Markdown("### Navigation")
-            with gr.Column(elem_classes="nav-buttons"):
-                annotated_btn = gr.Button("Annotated", variant="primary")
-                unannotated_btn = gr.Button("Unannotated", variant="primary")
-                gr.Markdown("---")
-                counter = gr.Markdown()
+        # Sidebar Column (keep this the same)
         
         # Main Content Column
         with gr.Column(scale=4):
             adapter_components = []
             for adapter in state.value["adapter_columns"]:
                 with gr.Row():
-                    adapter_components.extend([
-                        gr.Textbox(label=f"{adapter} Output", visible=False, interactive=False),
-                        gr.Radio(choices=[1, 2, 3, 4, 5], label=f"{adapter} Grammar", visible=False),
-                        gr.Radio(choices=[1, 2, 3, 4, 5], label=f"{adapter} Coherence", visible=False)
-                    ])
-                gr.Markdown("---")
+                    adapter_components.append(
+                        gr.Textbox(label=f"{adapter} Output", visible=False, interactive=False)
+                    )
+                with gr.Row():
+                    with gr.Column():
+                        adapter_components.append(
+                            gr.Markdown("### Grammar Score", visible=False)
+                        )
+                        adapter_components.append(
+                            gr.Radio(choices=[1, 2, 3, 4, 5], label="", visible=False)
+                        )
+                    with gr.Column():
+                        adapter_components.append(
+                            gr.Markdown("### Coherence Score", visible=False)
+                        )
+                        adapter_components.append(
+                            gr.Radio(choices=[1, 2, 3, 4, 5], label="", visible=False)
+                        )
+                gr.Markdown("---", visible=False)
             
             with gr.Row():
                 prev_btn = gr.Button("Previous", variant="secondary")
@@ -201,7 +267,7 @@ with gr.Blocks() as app:
     
     submit_btn.click(
         submit,
-        inputs=adapter_components[1::3] + adapter_components[2::3] + [state],
+        inputs=adapter_components[2::4] + adapter_components[4::4] + [state],  # Get every 4th component starting at index 2 and 4
         outputs=navigation_outputs
     )