From 0987a64a85d5399c8d906833cb2bdb98edaf5f4e Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Tue, 27 Dec 2022 11:14:21 +0100
Subject: [PATCH] Implement `*_reply` for the web

Also, start extracting shared components
---
 website/src/components/Button.tsx            | 19 +++++
 website/src/components/Messages.tsx          | 18 +++++
 website/src/components/TwoColumns.tsx        | 14 ++++
 website/src/pages/create/assistant_reply.tsx | 83 +++++++++++++++++++
 website/src/pages/create/user_reply.tsx      | 84 ++++++++++++++++++++
 website/src/pages/index.tsx                  |  6 ++
 6 files changed, 224 insertions(+)
 create mode 100644 website/src/components/Button.tsx
 create mode 100644 website/src/components/Messages.tsx
 create mode 100644 website/src/components/TwoColumns.tsx
 create mode 100644 website/src/pages/create/assistant_reply.tsx
 create mode 100644 website/src/pages/create/user_reply.tsx

diff --git a/website/src/components/Button.tsx b/website/src/components/Button.tsx
new file mode 100644
index 00000000..5ae1e7b4
--- /dev/null
+++ b/website/src/components/Button.tsx
@@ -0,0 +1,19 @@
+import clsx from "clsx";
+
+export const Button = (
+  props: React.DetailedHTMLProps<React.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>
+) => {
+  const { className, children, ...rest } = props;
+  return (
+    <button
+      type="button"
+      className={clsx(
+        "inline-flex items-center rounded-md border border-transparent px-4 py-2 text-sm font-medium focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2",
+        className
+      )}
+      {...rest}
+    >
+      {children}
+    </button>
+  );
+};
diff --git a/website/src/components/Messages.tsx b/website/src/components/Messages.tsx
new file mode 100644
index 00000000..4bc747d5
--- /dev/null
+++ b/website/src/components/Messages.tsx
@@ -0,0 +1,18 @@
+export interface Message {
+  text: string;
+  is_assistant: boolean;
+}
+
+const getColor = (isAssistant: boolean) => (isAssistant ? "bg-slate-800" : "bg-sky-900");
+
+export const Messages = ({ messages }: { messages: Message[] }) => {
+  const items = messages.map(({ text, is_assistant }: Message, i: number) => {
+    return (
+      <div key={i + text} className={`${getColor(is_assistant)} p-4 my-1 rounded-xl text-white whitespace-pre-wrap`}>
+        {text}
+      </div>
+    );
+  });
+  // Maybe also show a legend of the colors?
+  return <>{items}</>;
+};
diff --git a/website/src/components/TwoColumns.tsx b/website/src/components/TwoColumns.tsx
new file mode 100644
index 00000000..5792f7d0
--- /dev/null
+++ b/website/src/components/TwoColumns.tsx
@@ -0,0 +1,14 @@
+export const TwoColumns = ({ children }: { children: React.ReactNode[] }) => {
+  if (!Array.isArray(children) || children.length !== 2) {
+    throw new Error("TwoColumns expects 2 children");
+  }
+
+  const [first, second] = children;
+
+  return (
+    <section className="mb-8 lt-lg:mb-12 grid lg:gap-x-12 lg:grid-cols-2">
+      <div className="rounded-lg shadow-lg h-full block bg-white p-6">{first}</div>
+      <div className="rounded-lg shadow-lg h-full block bg-white p-6 mt-6 lg:mt-0">{second}</div>
+    </section>
+  );
+};
diff --git a/website/src/pages/create/assistant_reply.tsx b/website/src/pages/create/assistant_reply.tsx
new file mode 100644
index 00000000..14e78e54
--- /dev/null
+++ b/website/src/pages/create/assistant_reply.tsx
@@ -0,0 +1,83 @@
+import { Textarea } from "@chakra-ui/react";
+import { useRef, useState } from "react";
+import useSWRMutation from "swr/mutation";
+import useSWRImmutable from "swr/immutable";
+
+import fetcher from "src/lib/fetcher";
+import poster from "src/lib/poster";
+import { Messages } from "src/components/Messages";
+import { TwoColumns } from "src/components/TwoColumns";
+import { Button } from "src/components/Button";
+
+const AssistantReply = () => {
+  const [tasks, setTasks] = useState([]);
+
+  const inputRef = useRef<HTMLTextAreaElement>(null);
+
+  const { isLoading } = useSWRImmutable("/api/new_task/assistant_reply ", fetcher, {
+    onSuccess: (data) => {
+      console.log(data);
+      setTasks([data]);
+    },
+  });
+
+  const { trigger, isMutating } = useSWRMutation("/api/update_task", poster, {
+    onSuccess: async (data) => {
+      const newTask = await data.json();
+      setTasks((oldTasks) => [...oldTasks, newTask]);
+    },
+  });
+
+  const submitResponse = (task: { id: string }) => {
+    const text = inputRef.current.value.trim();
+    trigger({
+      id: task.id,
+      update_type: "text_reply_to_post",
+      content: {
+        text,
+      },
+    });
+  };
+
+  /**
+   * TODO: Make this a nicer loading screen.
+   */
+  if (tasks.length == 0) {
+    return <div className="p-6 h-full mx-auto bg-slate-100 text-gray-800">Loading...</div>;
+  }
+
+  const task = tasks[0].task;
+  return (
+    <div className="p-6 h-full mx-auto bg-slate-100 text-gray-800">
+      <TwoColumns>
+        <>
+          <h5 className="text-lg font-semibold">Reply as the assistant</h5>
+          <p className="text-lg py-1">Given the following conversation, provide an adequate reply</p>
+          <Messages messages={task.conversation.messages} />
+        </>
+        <Textarea name="reply" placeholder="Reply..." ref={inputRef} />
+      </TwoColumns>
+
+      <section className="mb-8 p-4 rounded-lg shadow-lg bg-white flex flex-row justify-items-stretch ">
+        <div className="grid grid-cols-[min-content_auto] gap-x-2 text-gray-700">
+          <b>Prompt</b>
+          <span>{tasks[0].id}</span>
+          <b>Output</b>
+          <span>Submit your answer</span>
+        </div>
+
+        <div className="flex justify-center ml-auto">
+          <Button className="mr-2 bg-indigo-100 text-indigo-700 hover:bg-indigo-200">Skip</Button>
+          <Button
+            onClick={() => submitResponse(tasks[0])}
+            className="bg-indigo-600 text-white shadow-sm hover:bg-indigo-700"
+          >
+            Submit
+          </Button>
+        </div>
+      </section>
+    </div>
+  );
+};
+
+export default AssistantReply;
diff --git a/website/src/pages/create/user_reply.tsx b/website/src/pages/create/user_reply.tsx
new file mode 100644
index 00000000..b9024650
--- /dev/null
+++ b/website/src/pages/create/user_reply.tsx
@@ -0,0 +1,84 @@
+import { Textarea } from "@chakra-ui/react";
+import { useRef, useState } from "react";
+import useSWRMutation from "swr/mutation";
+import useSWRImmutable from "swr/immutable";
+
+import fetcher from "src/lib/fetcher";
+import poster from "src/lib/poster";
+import { Messages } from "src/components/Messages";
+import { TwoColumns } from "src/components/TwoColumns";
+import { Button } from "src/components/Button";
+
+const UserReply = () => {
+  const [tasks, setTasks] = useState([]);
+
+  const inputRef = useRef<HTMLTextAreaElement>(null);
+
+  const { isLoading } = useSWRImmutable("/api/new_task/user_reply", fetcher, {
+    onSuccess: (data) => {
+      console.log(data);
+      setTasks([data]);
+    },
+  });
+
+  const { trigger, isMutating } = useSWRMutation("/api/update_task", poster, {
+    onSuccess: async (data) => {
+      const newTask = await data.json();
+      setTasks((oldTasks) => [...oldTasks, newTask]);
+    },
+  });
+
+  const submitResponse = (task: { id: string }) => {
+    const text = inputRef.current.value.trim();
+    trigger({
+      id: task.id,
+      update_type: "text_reply_to_post",
+      content: {
+        text,
+      },
+    });
+  };
+
+  /**
+   * TODO: Make this a nicer loading screen.
+   */
+  if (tasks.length == 0) {
+    return <div className="p-6 h-full mx-auto bg-slate-100 text-gray-800">Loading...</div>;
+  }
+
+  const task = tasks[0].task;
+  return (
+    <div className="p-6 h-full mx-auto bg-slate-100 text-gray-800">
+      <TwoColumns>
+        <>
+          <h5 className="text-lg font-semibold">Reply as a user</h5>
+          <p className="text-lg py-1">Given the following conversation, provide an adequate reply</p>
+          <Messages messages={task.conversation.messages} />
+          {task.hint && <p className="text-lg py-1">Hint: {task.hint}</p>}
+        </>
+        <Textarea name="reply" placeholder="Reply..." ref={inputRef} />
+      </TwoColumns>
+
+      <section className="mb-8 p-4 rounded-lg shadow-lg bg-white flex flex-row justify-items-stretch ">
+        <div className="grid grid-cols-[min-content_auto] gap-x-2 text-gray-700">
+          <b>Prompt</b>
+          <span>{tasks[0].id}</span>
+          <b>Output</b>
+          <span>Submit your answer</span>
+        </div>
+
+        <div className="flex justify-center ml-auto">
+          <Button className="mr-2 bg-indigo-100 text-indigo-700 hover:bg-indigo-200">Skip</Button>
+          <Button
+            onClick={() => submitResponse(tasks[0])}
+            className="bg-indigo-600 text-white shadow-sm hover:bg-indigo-700"
+          >
+            Submit
+          </Button>
+        </div>
+      </section>
+    </div>
+  );
+};
+
+export default UserReply;
diff --git a/website/src/pages/index.tsx b/website/src/pages/index.tsx
index 4aa93585..71b9360f 100644
--- a/website/src/pages/index.tsx
+++ b/website/src/pages/index.tsx
@@ -53,6 +53,12 @@ export default function Home() {
         <Button size="lg" colorScheme="blue" className="drop-shadow">
           <Link href="/summarize/story">Summarize a story</Link>
         </Button>
+        <Button size="lg" colorScheme="blue" className="drop-shadow">
+          <Link href="/create/user_reply">Reply as a user</Link>
+        </Button>
+        <Button size="lg" colorScheme="blue" className="drop-shadow">
+          <Link href="/create/assistant_reply">Reply as the assistant</Link>
+        </Button>
       </main>
       <Footer />
     </>