Merge pull request #138 from jojopirker/newLoading

New loading screen component

website/src/components/Loading/Loading.stories.jsx

@@ -0,0 +1,16 @@
+import { LoadingScreen } from "./LoadingScreen";
+
+export default {
+  title: "Example/LoadingScreen",
+  component: LoadingScreen,
+  parameters: {
+    layout: "fullscreen",
+  },
+};
+
+const Template = (args) => <LoadingScreen {...args} />;
+
+export const Default = Template.bind({});
+
+export const WithText = Template.bind({});
+WithText.args = { text: "Loading Text ..." };

website/src/components/Loading/LoadingScreen.jsx

@@ -0,0 +1,12 @@
+import { Progress } from "@chakra-ui/react";
+
+export const LoadingScreen = ({ text }) => (
+  <div className="bg-slate-100">
+    <Progress size="xs" isIndeterminate />
+    {text && (
+      <div className="flex h-full">
+        <div className="text-xl font-bold text-gray-800  mx-auto my-auto">{text}</div>
+      </div>
+    )}
+  </div>
+);

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,11 +40,12 @@ const AssistantReply = () => {
     });
   };
 
-  /**
-   * 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 task = tasks[0].task;

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,12 @@ 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,12 @@ const UserReply = () => {
     });
   };
 
-  /**
-   * 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 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,12 +45,14 @@ const RankInitialPrompts = () => {
     });
   };
 
-  /**
-   * 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 (isLoading) {
+    return <LoadingScreen text="Loading..." />;
   }
+
+  if (tasks.length == 0) {
+    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) => ({
     text: prompts[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,12 @@ 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 (