From 96be0edb1eaf104201026d18626775cccc6bae9a Mon Sep 17 00:00:00 2001 From: Kostia Date: Wed, 4 Jan 2023 18:58:54 +0200 Subject: [PATCH] Fix for #343 --- website/src/pages/create/assistant_reply.tsx | 8 +++++++- website/src/pages/create/initial_prompt.tsx | 8 +++++++- website/src/pages/create/user_reply.tsx | 8 +++++++- website/src/pages/evaluate/rank_assistant_replies.tsx | 8 +++++++- website/src/pages/evaluate/rank_initial_prompts.tsx | 8 +++++++- website/src/pages/evaluate/rank_user_replies.tsx | 8 +++++++- 6 files changed, 42 insertions(+), 6 deletions(-) diff --git a/website/src/pages/create/assistant_reply.tsx b/website/src/pages/create/assistant_reply.tsx index ceac45be..2f7ca748 100644 --- a/website/src/pages/create/assistant_reply.tsx +++ b/website/src/pages/create/assistant_reply.tsx @@ -1,6 +1,6 @@ import { Container, Textarea } from "@chakra-ui/react"; import { useColorMode } from "@chakra-ui/react"; -import { useRef, useState } from "react"; +import { useEffect, useRef, useState } from "react"; import { LoadingScreen } from "src/components/Loading/LoadingScreen"; import { Messages } from "src/components/Messages"; import { TaskControls } from "src/components/Survey/TaskControls"; @@ -21,6 +21,12 @@ const AssistantReply = () => { }, }); + useEffect(() => { + if (tasks.length == 0) { + mutate(); + } + }, [tasks]); + const { trigger } = useSWRMutation("/api/update_task", poster, { onSuccess: async (data) => { const newTask = await data.json(); diff --git a/website/src/pages/create/initial_prompt.tsx b/website/src/pages/create/initial_prompt.tsx index 5aecf98c..a4aed9c3 100644 --- a/website/src/pages/create/initial_prompt.tsx +++ b/website/src/pages/create/initial_prompt.tsx @@ -1,6 +1,6 @@ import { Container, Textarea } from "@chakra-ui/react"; import { useColorMode } from "@chakra-ui/react"; -import { useRef, useState } from "react"; +import { useEffect, useRef, useState } from "react"; import { LoadingScreen } from "src/components/Loading/LoadingScreen"; import { TaskControls } from "src/components/Survey/TaskControls"; import { TwoColumnsWithCards } from "src/components/Survey/TwoColumnsWithCards"; @@ -27,6 +27,12 @@ const InitialPrompt = () => { }, }); + useEffect(() => { + if (tasks.length == 0) { + mutate(); + } + }, [tasks]); + const submitResponse = (task: { id: string }) => { const text = inputRef.current.value.trim(); trigger({ diff --git a/website/src/pages/create/user_reply.tsx b/website/src/pages/create/user_reply.tsx index b9022e86..40409189 100644 --- a/website/src/pages/create/user_reply.tsx +++ b/website/src/pages/create/user_reply.tsx @@ -1,6 +1,6 @@ import { Textarea } from "@chakra-ui/react"; import { useColorMode } from "@chakra-ui/react"; -import { useRef, useState } from "react"; +import { useEffect, useRef, useState } from "react"; import { LoadingScreen } from "src/components/Loading/LoadingScreen"; import { Messages } from "src/components/Messages"; import { TaskControls } from "src/components/Survey/TaskControls"; @@ -21,6 +21,12 @@ const UserReply = () => { }, }); + useEffect(() => { + if (tasks.length == 0) { + mutate(); + } + }, [tasks]); + const { trigger } = useSWRMutation("/api/update_task", poster, { onSuccess: async (data) => { const newTask = await data.json(); diff --git a/website/src/pages/evaluate/rank_assistant_replies.tsx b/website/src/pages/evaluate/rank_assistant_replies.tsx index e8558c00..0ed69b09 100644 --- a/website/src/pages/evaluate/rank_assistant_replies.tsx +++ b/website/src/pages/evaluate/rank_assistant_replies.tsx @@ -1,6 +1,6 @@ import { useColorMode } from "@chakra-ui/react"; import Head from "next/head"; -import { useState } from "react"; +import { useEffect, useState } from "react"; import { ContextMessages } from "src/components/ContextMessages"; import { LoadingScreen } from "src/components/Loading/LoadingScreen"; import { Message } from "src/components/Messages"; @@ -26,6 +26,12 @@ const RankAssistantReplies = () => { }, }); + useEffect(() => { + if (tasks.length == 0) { + mutate(); + } + }, [tasks]); + const { trigger } = useSWRMutation("/api/update_task", poster, { onSuccess: async (data) => { const newTask = await data.json(); diff --git a/website/src/pages/evaluate/rank_initial_prompts.tsx b/website/src/pages/evaluate/rank_initial_prompts.tsx index 48a67e90..a9d590ac 100644 --- a/website/src/pages/evaluate/rank_initial_prompts.tsx +++ b/website/src/pages/evaluate/rank_initial_prompts.tsx @@ -1,6 +1,6 @@ import { useColorMode } from "@chakra-ui/react"; import Head from "next/head"; -import { useState } from "react"; +import { useEffect, useState } from "react"; import { LoadingScreen } from "src/components/Loading/LoadingScreen"; import { Sortable } from "src/components/Sortable/Sortable"; import { SurveyCard } from "src/components/Survey/SurveyCard"; @@ -32,6 +32,12 @@ const RankInitialPrompts = () => { }, }); + useEffect(() => { + if (tasks.length == 0) { + mutate(); + } + }, [tasks]); + const submitResponse = (task) => { trigger({ id: task.id, diff --git a/website/src/pages/evaluate/rank_user_replies.tsx b/website/src/pages/evaluate/rank_user_replies.tsx index 3f806a8a..9a0577cb 100644 --- a/website/src/pages/evaluate/rank_user_replies.tsx +++ b/website/src/pages/evaluate/rank_user_replies.tsx @@ -1,6 +1,6 @@ import { useColorMode } from "@chakra-ui/react"; import Head from "next/head"; -import { useState } from "react"; +import { useEffect, useState } from "react"; import { ContextMessages } from "src/components/ContextMessages"; import { LoadingScreen } from "src/components/Loading/LoadingScreen"; import { Message } from "src/components/Messages"; @@ -26,6 +26,12 @@ const RankUserReplies = () => { }, }); + useEffect(() => { + if (tasks.length == 0) { + mutate(); + } + }, [tasks]); + const { trigger } = useSWRMutation("/api/update_task", poster, { onSuccess: async (data) => { const newTask = await data.json();