Merge pull request #393 from kostiak/fix-343

Fix for #343

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();

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({

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();

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();

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,

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();