new LoadingScreen to replace old loading divs

website/src/pages/create/assistant_reply.tsx

@@ -8,6 +8,7 @@ import poster from "src/lib/poster";
 import { Messages } from "src/components/Messages";
 import { TwoColumns } from "src/components/TwoColumns";
 import { Button } from "src/components/Button";
+import { LoadingScreen } from "src/components/Loading/LoadingScreen";
 
 const AssistantReply = () => {
   const [tasks, setTasks] = useState([]);
@@ -39,12 +40,13 @@ const AssistantReply = () => {
     });
   };
 
-  /**
-   * TODO: Make this a nicer loading screen.
-   */
-  if (tasks.length == 0) {
-    return <div className="p-6 bg-slate-100 text-gray-800">Loading...</div>;
+  if (true) {
+    return (<LoadingScreen text="Loading..."/>);
   }
+  if (tasks.length == 0) {
+    return <div className="p-6 bg-slate-100 text-gray-800">No tasks found...</div>;
+  }
+
 
   const task = tasks[0].task;
   return (

website/src/pages/create/summarize_story.tsx

@@ -9,6 +9,7 @@ import poster from "src/lib/poster";
 
 import { TwoColumns } from "src/components/TwoColumns";
 import { Button } from "src/components/Button";
+import { LoadingScreen } from "@/components/Loading/LoadingScreen";
 
 const SummarizeStory = () => {
   // Use an array of tasks that record the sequence of steps until a task is
@@ -49,11 +50,11 @@ const SummarizeStory = () => {
     });
   };
 
-  /**
-   * TODO: Make this a nicer loading screen.
-   */
+  if (isLoading) {
+    return (<LoadingScreen text="Loading..."/>);
+  }
   if (tasks.length == 0) {
-    return <div className=" p-6 bg-slate-100 text-gray-800">Loading...</div>;
+    return <div className="p-6 bg-slate-100 text-gray-800">No tasks found...</div>;
   }
 
   return (

website/src/pages/create/user_reply.tsx

@@ -8,6 +8,7 @@ import poster from "src/lib/poster";
 import { Messages } from "src/components/Messages";
 import { TwoColumns } from "src/components/TwoColumns";
 import { Button } from "src/components/Button";
+import { LoadingScreen } from "@/components/Loading/LoadingScreen";
 
 const UserReply = () => {
   const [tasks, setTasks] = useState([]);
@@ -39,11 +40,16 @@ const UserReply = () => {
     });
   };
 
+
+  if (isLoading) {
+    return (<LoadingScreen text="Loading..."/>);
+  }
+
   /**
    * TODO: Make this a nicer loading screen.
    */
   if (tasks.length == 0) {
-    return <div className="p-6 bg-slate-100 text-gray-800">Loading...</div>;
+    return <div className="p-6 bg-slate-100 text-gray-800">No tasks found...</div>;
   }
 
   const task = tasks[0].task;

website/src/pages/evaluate/rank_initial_prompts.tsx

@@ -9,6 +9,7 @@ import fetcher from "src/lib/fetcher";
 import poster from "src/lib/poster";
 
 import { Button } from "src/components/Button";
+import { LoadingScreen } from "@/components/Loading/LoadingScreen";
 
 const RankInitialPrompts = () => {
   const [tasks, setTasks] = useState([]);
@@ -44,11 +45,11 @@ const RankInitialPrompts = () => {
     });
   };
 
-  /**
-   * TODO: Make this a nicer loading screen.
-   */
+  if (isLoading) {
+    return (<LoadingScreen text="Loading..."/>);
+  }
   if (tasks.length == 0) {
-    return <div className="p-6 bg-slate-100 text-gray-800">Loading...</div>;
+    return <div className="p-6 bg-slate-100 text-gray-800">No tasks found...</div>;
   }
   const prompts = tasks[0].task.prompts as string[];
   const items = ranking.map((i) => ({

website/src/pages/evaluate/rate_summary.tsx

@@ -11,6 +11,7 @@ import poster from "src/lib/poster";
 
 import { TwoColumns } from "src/components/TwoColumns";
 import { Button } from "src/components/Button";
+import { LoadingScreen } from "@/components/Loading/LoadingScreen";
 
 const RateSummary = () => {
   // Use an array of tasks that record the sequence of steps until a task is
@@ -49,11 +50,11 @@ const RateSummary = () => {
     });
   };
 
-  /**
-   * TODO: Make this a nicer loading screen.
-   */
+  if (isLoading) {
+    return (<LoadingScreen text="Loading..."/>);
+  }
   if (tasks.length == 0) {
-    return <div className="p-6 bg-slate-100 text-gray-800">Loading...</div>;
+    return <div className="p-6 bg-slate-100 text-gray-800">No tasks found...</div>;
   }
 
   return (