Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

[turn annotations static] more customization (annotation question, buckets, speaker label) #3926

Merged
merged 4 commits into from
Aug 19, 2021
Merged

[turn annotations static] more customization (annotation question, buckets, speaker label) #3926

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
086701a
annotation buckets customization from task data
meganung Aug 5, 2021
88f3b20
customize annotation question per utterance
meganung Aug 5, 2021
fba9e9f
customize speaker label
meganung Aug 5, 2021
03941ee
use turnIdx for determining speaker label
meganung Aug 16, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 35 additions & 6 deletions parlai/crowdsourcing/tasks/turn_annotations_static/webapp/src/components/task_components.jsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,24 @@ var handleUserInputUpdate = function (subtaskData) {
}
}

var getAnnotationBuckets = function (taskData, annotationBuckets) {
// return list of all the bucket ids
var buckets = []
if (annotationBuckets !== null && 'config' in annotationBuckets){
buckets = Object.keys(annotationBuckets.config);
}
taskData.forEach(elem => {
if ('annotation_buckets' in elem) {
(Object.keys(elem.annotation_buckets.config)).forEach(bucketKey => {
if (!buckets.includes(bucketKey)) {
buckets.push(bucketKey)
}
})
}
})
return buckets
}

var handleSubtaskSubmit = function (subtaskIndex, setIndex, numSubtasks, initialTaskData, annotationBuckets, mephistoSubmit) {
// initialTaskData is the initial task data for this index
console.log('In handleSubtaskSubmit for subtask: ' + subtaskIndex);
Expand All @@ -103,8 +121,8 @@ var handleSubtaskSubmit = function (subtaskIndex, setIndex, numSubtasks, initial
'agent_idx': initialTaskData[i].agent_idx,
'other_metadata': initialTaskData[i].other_metadata
};
if (annotationBuckets !== null) {
var buckets = Object.keys(annotationBuckets.config);
var buckets = getAnnotationBuckets(initialTaskData, annotationBuckets)
if (buckets !== null && buckets.length > 0) {
for (var j = 0; j < buckets.length; j++) {
answersForTurn[buckets[j]] = null;
var checkbox = document.getElementById(buckets[j] + '_' + i);
Expand Down Expand Up @@ -198,9 +216,18 @@ function SubtaskSubmitButton({ subtaskIndex, numSubtasks, onSubtaskSubmit }) {
)
}

function ChatMessage({ text, agentIdx, annotationQuestion, annotationBuckets, turnIdx, doAnnotateMessage, askReason, responseField, onUserInputUpdate }) {
function ChatMessage({ text, agentIdx, annotationQuestion, annotationBuckets, turnIdx, doAnnotateMessage, askReason, responseField, speakerLabel, onUserInputUpdate }) {
var extraElements = '';
var responseInputElement = '';
if (speakerLabel == null) {
speakerLabel = turnIdx % 2 == 0 ? 'YOU' : 'THEM'
}
var speakerElements = (
<div>
<b>{speakerLabel}:</b> {text}
</div>
)

if (doAnnotateMessage) {
if (annotationBuckets !== null) {
extraElements = (<span key={'extra_' + turnIdx}><br /><br />
Expand Down Expand Up @@ -233,7 +260,8 @@ function ChatMessage({ text, agentIdx, annotationQuestion, annotationBuckets, tu
return (
<div>
<div className={`alert ${agentIdx == 0 ? "alert-info" : "alert-warning"}`} style={{ float: `${agentIdx == 0 ? "right" : "left"}`, display: 'table', minWidth: `${agentIdx == 0 ? "30%" : "80%"}`, marginTop: "auto" }}>
<span><b>{turnIdx % 2 == 0 ? 'YOU' : 'THEM'}:</b> {text}
<span>
{speakerElements}
<ErrorBoundary>
{extraElements}
</ErrorBoundary>
Expand Down Expand Up @@ -266,11 +294,12 @@ function ContentPane({ subtaskData, taskConfig, subtaskIndex, numSubtasks }) {
text={m.text}
agentIdx={m.agent_idx}
turnIdx={idx}
annotationQuestion={annotationQuestion}
annotationBuckets={annotationBuckets}
annotationQuestion={ ('annotation_question' in m) ? m.annotation_question : annotationQuestion}
annotationBuckets={ ('annotation_buckets' in m) ? m.annotation_buckets : annotationBuckets}
doAnnotateMessage={m.do_annotate}
askReason={askReason}
responseField={responseField}
speakerLabel={('speaker_label' in m) ? m.speaker_label : null}
onUserInputUpdate={() => handleUserInputUpdate(subtaskData)}
/>
</div>
Expand Down