From ce331a1bbdbbeacef05c941aaf94b36da15a24ea Mon Sep 17 00:00:00 2001
From: Mckay Wrigley <mckaywrigley@gmail.com>
Date: Wed, 15 Mar 2023 04:24:09 -0600
Subject: [PATCH] boom

---
 components/Chat/Chat.tsx               | 61 ++++++++++++++++++--------
 components/Chat/ChatInput.tsx          | 34 +++++++-------
 components/Chat/ChatLoader.tsx         | 11 ++---
 components/Chat/ChatMessage.tsx        | 14 +++---
 components/ModelSelect.tsx             | 30 +++++++++++++
 components/Sidebar/Sidebar.tsx         | 31 +++++++++++++
 components/Sidebar/SidebarButton.tsx   | 19 ++++++++
 components/Sidebar/SidebarSettings.tsx | 20 +++++++++
 pages/api/chat.ts                      |  7 +--
 pages/index.tsx                        | 54 +++++++++--------------
 tailwind.config.js                     |  1 +
 types/index.ts                         | 10 ++++-
 utils/index.ts                         |  4 +-
 13 files changed, 212 insertions(+), 84 deletions(-)
 create mode 100644 components/ModelSelect.tsx
 create mode 100644 components/Sidebar/Sidebar.tsx
 create mode 100644 components/Sidebar/SidebarButton.tsx
 create mode 100644 components/Sidebar/SidebarSettings.tsx

diff --git a/components/Chat/Chat.tsx b/components/Chat/Chat.tsx
index d7a3c94..b716340 100644
--- a/components/Chat/Chat.tsx
+++ b/components/Chat/Chat.tsx
@@ -1,34 +1,59 @@
-import { Message } from "@/types";
-import { FC } from "react";
+import { Message, OpenAIModel, OpenAIModelNames } from "@/types";
+import { FC, useEffect, useRef } from "react";
+import { ModelSelect } from "../ModelSelect";
 import { ChatInput } from "./ChatInput";
 import { ChatLoader } from "./ChatLoader";
 import { ChatMessage } from "./ChatMessage";
 
 interface Props {
+  model: OpenAIModel;
   messages: Message[];
   loading: boolean;
   onSend: (message: Message) => void;
+  onSelect: (model: OpenAIModel) => void;
 }
 
-export const Chat: FC<Props> = ({ messages, loading, onSend }) => {
-  return (
-    <div className="flex flex-col rounded-lg px-2 sm:p-4 sm:border border-neutral-300">
-      {messages.map((message, index) => (
-        <div
-          key={index}
-          className="my-1 sm:my-1.5"
-        >
-          <ChatMessage message={message} />
-        </div>
-      ))}
+export const Chat: FC<Props> = ({ model, messages, loading, onSend, onSelect }) => {
+  const messagesEndRef = useRef<HTMLDivElement>(null);
 
-      {loading && (
-        <div className="my-1 sm:my-1.5">
-          <ChatLoader />
-        </div>
+  const scrollToBottom = () => {
+    messagesEndRef.current?.scrollIntoView({ behavior: "smooth" });
+  };
+
+  useEffect(() => {
+    scrollToBottom();
+  }, [messages]);
+
+  return (
+    <div className="h-full flex flex-col">
+      {messages.length === 0 ? (
+        <>
+          <div className="flex justify-center pt-8">
+            <ModelSelect
+              model={model}
+              onSelect={onSelect}
+            />
+          </div>
+
+          <div className="flex-1 text-4xl text-center text-neutral-300 pt-[280px]">Chatbot UI Pro</div>
+        </>
+      ) : (
+        <>
+          <div className="flex-1 overflow-auto">
+            <div className="text-center py-3 dark:bg-[#434654] dark:text-neutral-300 text-neutral-500 text-sm border border-b-neutral-300 dark:border-none">Model: {OpenAIModelNames[model]}</div>
+
+            {messages.map((message, index) => (
+              <div key={index}>
+                <ChatMessage message={message} />
+              </div>
+            ))}
+            {loading && <ChatLoader />}
+            <div ref={messagesEndRef} />
+          </div>
+        </>
       )}
 
-      <div className="mt-4 sm:mt-8 bottom-[56px] left-0 w-full">
+      <div className="h-[140px] w-[800px] mx-auto">
         <ChatInput onSend={onSend} />
       </div>
     </div>
diff --git a/components/Chat/ChatInput.tsx b/components/Chat/ChatInput.tsx
index 75bc7a3..cf9f460 100644
--- a/components/Chat/ChatInput.tsx
+++ b/components/Chat/ChatInput.tsx
@@ -1,5 +1,5 @@
 import { Message } from "@/types";
-import { IconArrowUp } from "@tabler/icons-react";
+import { IconSend } from "@tabler/icons-react";
 import { FC, KeyboardEvent, useEffect, useRef, useState } from "react";
 
 interface Props {
@@ -46,20 +46,24 @@ export const ChatInput: FC<Props> = ({ onSend }) => {
 
   return (
     <div className="relative">
-      <textarea
-        ref={textareaRef}
-        className="min-h-[44px] rounded-lg pl-4 pr-12 py-2 w-full focus:outline-none focus:ring-1 focus:ring-neutral-300 border-2 border-neutral-200"
-        style={{ resize: "none" }}
-        placeholder="Type a message..."
-        value={content}
-        rows={1}
-        onChange={handleChange}
-        onKeyDown={handleKeyDown}
-      />
-
-      <button onClick={() => handleSend()}>
-        <IconArrowUp className="absolute right-2 bottom-3 h-8 w-8 hover:cursor-pointer rounded-full p-1 bg-blue-500 text-white hover:opacity-80" />
-      </button>
+      <div className="absolute bottom-[-120px] w-full">
+        <textarea
+          ref={textareaRef}
+          className="rounded-lg pl-4 pr-8 py-3 w-full focus:outline-none max-h-[280px] dark:bg-[#40414F] dark:border-opacity-50 dark:border-neutral-800 dark:text-neutral-100 border border-neutral-300 shadow text-neutral-900"
+          style={{ resize: "none", bottom: `${textareaRef?.current?.scrollHeight}px` }}
+          placeholder="Type a message..."
+          value={content}
+          rows={1}
+          onChange={handleChange}
+          onKeyDown={handleKeyDown}
+        />
+        <button
+          className="absolute right-2 bottom-[14px] text-neutral-400 p-2 hover:dark:bg-neutral-800 hover:bg-neutral-400 hover:text-white rounded-md"
+          onClick={handleSend}
+        >
+          <IconSend size={18} />
+        </button>
+      </div>
     </div>
   );
 };
diff --git a/components/Chat/ChatLoader.tsx b/components/Chat/ChatLoader.tsx
index f4e8fdf..9960b00 100644
--- a/components/Chat/ChatLoader.tsx
+++ b/components/Chat/ChatLoader.tsx
@@ -5,11 +5,12 @@ interface Props {}
 
 export const ChatLoader: FC<Props> = () => {
   return (
-    <div className="flex flex-col flex-start">
-      <div
-        className={`flex items-center bg-neutral-200 text-neutral-900 rounded-2xl px-4 py-2 w-fit`}
-        style={{ overflowWrap: "anywhere" }}
-      >
+    <div
+      className={`flex justify-center px-[120px] py-[30px] whitespace-pre-wrap dark:bg-[#434654] dark:text-neutral-100 bg-neutral-100 text-neutral-900 dark:border-none"`}
+      style={{ overflowWrap: "anywhere" }}
+    >
+      <div className="w-[650px] flex">
+        <div className="mr-4 font-bold min-w-[30px]">AI:</div>
         <IconDots className="animate-pulse" />
       </div>
     </div>
diff --git a/components/Chat/ChatMessage.tsx b/components/Chat/ChatMessage.tsx
index 8ee936e..24cde30 100644
--- a/components/Chat/ChatMessage.tsx
+++ b/components/Chat/ChatMessage.tsx
@@ -7,12 +7,14 @@ interface Props {
 
 export const ChatMessage: FC<Props> = ({ message }) => {
   return (
-    <div className={`flex flex-col ${message.role === "assistant" ? "items-start" : "items-end"}`}>
-      <div
-        className={`flex items-center ${message.role === "assistant" ? "bg-neutral-200 text-neutral-900" : "bg-blue-500 text-white"} rounded-2xl px-3 py-2 max-w-[67%] whitespace-pre-wrap`}
-        style={{ overflowWrap: "anywhere" }}
-      >
-        {message.content}
+    <div
+      className={`flex justify-center px-[120px] py-[30px] whitespace-pre-wrap] ${message.role === "assistant" ? "dark:bg-[#434654] dark:text-neutral-100 bg-neutral-100 text-neutral-900 border border-neutral-300 dark:border-none" : "dark:bg-[#343541] dark:text-white text-neutral-900"}`}
+      style={{ overflowWrap: "anywhere" }}
+    >
+      <div className="w-[650px] flex">
+        <div className="mr-4 font-bold min-w-[40px]">{message.role === "assistant" ? "AI:" : "You:"}</div>
+
+        <div className="whitespace-pre-wrap">{message.content}</div>
       </div>
     </div>
   );
diff --git a/components/ModelSelect.tsx b/components/ModelSelect.tsx
new file mode 100644
index 0000000..d092c45
--- /dev/null
+++ b/components/ModelSelect.tsx
@@ -0,0 +1,30 @@
+import { OpenAIModel, OpenAIModelNames } from "@/types";
+import { FC } from "react";
+
+interface Props {
+  model: OpenAIModel;
+  onSelect: (model: OpenAIModel) => void;
+}
+
+export const ModelSelect: FC<Props> = ({ model, onSelect }) => {
+  return (
+    <div className="flex flex-col">
+      <label className="text-left mb-2 dark:text-neutral-400 text-neutral-700">Model</label>
+      <select
+        className="w-[300px] p-3 dark:text-white dark:bg-[#343541] border border-neutral-500 rounded-lg appearance-none focus:shadow-outline text-neutral-900"
+        placeholder="Select a model"
+        value={model}
+        onChange={(e) => onSelect(e.target.value as OpenAIModel)}
+      >
+        {Object.entries(OpenAIModelNames).map(([value, name]) => (
+          <option
+            key={value}
+            value={value}
+          >
+            {name}
+          </option>
+        ))}
+      </select>
+    </div>
+  );
+};
diff --git a/components/Sidebar/Sidebar.tsx b/components/Sidebar/Sidebar.tsx
new file mode 100644
index 0000000..8e7279a
--- /dev/null
+++ b/components/Sidebar/Sidebar.tsx
@@ -0,0 +1,31 @@
+import { IconPlus } from "@tabler/icons-react";
+import { FC } from "react";
+import { SidebarSettings } from "./SidebarSettings";
+
+interface Props {
+  lightMode: "light" | "dark";
+  onToggleLightMode: (mode: "light" | "dark") => void;
+}
+
+export const Sidebar: FC<Props> = ({ lightMode, onToggleLightMode }) => {
+  return (
+    <div className="flex flex-col bg-[#202123] min-w-[260px]">
+      <div className="flex items-center justify-center h-[60px]">
+        <button className="flex items-center w-[240px] h-[40px] rounded-lg bg-[#202123] border border-neutral-600 text-sm hover:bg-neutral-700">
+          <IconPlus
+            className="ml-4 mr-3"
+            size={16}
+          />
+          New chat
+        </button>
+      </div>
+
+      <div className="flex-1"></div>
+
+      <SidebarSettings
+        lightMode={lightMode}
+        onToggleLightMode={onToggleLightMode}
+      />
+    </div>
+  );
+};
diff --git a/components/Sidebar/SidebarButton.tsx b/components/Sidebar/SidebarButton.tsx
new file mode 100644
index 0000000..7606055
--- /dev/null
+++ b/components/Sidebar/SidebarButton.tsx
@@ -0,0 +1,19 @@
+import { FC } from "react";
+
+interface Props {
+  text: string;
+  icon: JSX.Element;
+  onClick: () => void;
+}
+
+export const SidebarButton: FC<Props> = ({ text, icon, onClick }) => {
+  return (
+    <div
+      className="flex hover:bg-[#343541] py-2 px-4 rounded-md cursor-pointer w-[200px] items-center"
+      onClick={onClick}
+    >
+      <div className="mr-3">{icon}</div>
+      <div>{text}</div>
+    </div>
+  );
+};
diff --git a/components/Sidebar/SidebarSettings.tsx b/components/Sidebar/SidebarSettings.tsx
new file mode 100644
index 0000000..e288745
--- /dev/null
+++ b/components/Sidebar/SidebarSettings.tsx
@@ -0,0 +1,20 @@
+import { IconMoon, IconSun } from "@tabler/icons-react";
+import { FC } from "react";
+import { SidebarButton } from "./SidebarButton";
+
+interface Props {
+  lightMode: "light" | "dark";
+  onToggleLightMode: (mode: "light" | "dark") => void;
+}
+
+export const SidebarSettings: FC<Props> = ({ lightMode, onToggleLightMode }) => {
+  return (
+    <div className="flex flex-col items-center border-t border-neutral-500 py-4">
+      <SidebarButton
+        text={lightMode === "light" ? "Dark mode" : "Light mode"}
+        icon={lightMode === "light" ? <IconMoon /> : <IconSun />}
+        onClick={() => onToggleLightMode(lightMode === "light" ? "dark" : "light")}
+      />
+    </div>
+  );
+};
diff --git a/pages/api/chat.ts b/pages/api/chat.ts
index 42cc3a3..987062f 100644
--- a/pages/api/chat.ts
+++ b/pages/api/chat.ts
@@ -1,4 +1,4 @@
-import { Message } from "@/types";
+import { Message, OpenAIModel } from "@/types";
 import { OpenAIStream } from "@/utils";
 
 export const config = {
@@ -7,7 +7,8 @@ export const config = {
 
 const handler = async (req: Request): Promise<Response> => {
   try {
-    const { messages } = (await req.json()) as {
+    const { model, messages } = (await req.json()) as {
+      model: OpenAIModel;
       messages: Message[];
     };
 
@@ -24,7 +25,7 @@ const handler = async (req: Request): Promise<Response> => {
       messagesToSend.push(message);
     }
 
-    const stream = await OpenAIStream(messagesToSend);
+    const stream = await OpenAIStream(model, messagesToSend);
 
     return new Response(stream);
   } catch (error) {
diff --git a/pages/index.tsx b/pages/index.tsx
index 4da9ad7..bf23ae9 100644
--- a/pages/index.tsx
+++ b/pages/index.tsx
@@ -1,19 +1,14 @@
 import { Chat } from "@/components/Chat/Chat";
-import { Footer } from "@/components/Layout/Footer";
-import { Navbar } from "@/components/Layout/Navbar";
-import { Message } from "@/types";
+import { Sidebar } from "@/components/Sidebar/Sidebar";
+import { Message, OpenAIModel } from "@/types";
 import Head from "next/head";
-import { useEffect, useRef, useState } from "react";
+import { useEffect, useState } from "react";
 
 export default function Home() {
   const [messages, setMessages] = useState<Message[]>([]);
   const [loading, setLoading] = useState<boolean>(false);
-
-  const messagesEndRef = useRef<HTMLDivElement>(null);
-
-  const scrollToBottom = () => {
-    messagesEndRef.current?.scrollIntoView({ behavior: "smooth" });
-  };
+  const [model, setModel] = useState<OpenAIModel>(OpenAIModel.GPT_3_5);
+  const [lightMode, setLightMode] = useState<"dark" | "light">("dark");
 
   const handleSend = async (message: Message) => {
     const updatedMessages = [...messages, message];
@@ -27,6 +22,7 @@ export default function Home() {
         "Content-Type": "application/json"
       },
       body: JSON.stringify({
+        model,
         messages: updatedMessages
       })
     });
@@ -76,18 +72,7 @@ export default function Home() {
     }
   };
 
-  useEffect(() => {
-    scrollToBottom();
-  }, [messages]);
-
-  useEffect(() => {
-    setMessages([
-      {
-        role: "assistant",
-        content: `Hi there! I'm Chatbot UI, an AI assistant. I can help you with things like answering questions, providing information, and helping with tasks. How can I help you?`
-      }
-    ]);
-  }, []);
+  useEffect(() => {}, []);
 
   return (
     <>
@@ -107,20 +92,21 @@ export default function Home() {
         />
       </Head>
 
-      <div className="flex flex-col h-screen">
-        <Navbar />
+      <div className={`flex h-screen text-white ${lightMode}`}>
+        <Sidebar
+          lightMode={lightMode}
+          onToggleLightMode={setLightMode}
+        />
 
-        <div className="flex-1 overflow-auto sm:px-10 pb-4 sm:pb-10">
-          <div className="max-w-[800px] mx-auto mt-4 sm:mt-12">
-            <Chat
-              messages={messages}
-              loading={loading}
-              onSend={handleSend}
-            />
-            <div ref={messagesEndRef} />
-          </div>
+        <div className="flex flex-col w-full h-full dark:bg-[#343541]">
+          <Chat
+            model={model}
+            messages={messages}
+            loading={loading}
+            onSend={handleSend}
+            onSelect={setModel}
+          />
         </div>
-        <Footer />
       </div>
     </>
   );
diff --git a/tailwind.config.js b/tailwind.config.js
index 149d0e6..21c7518 100644
--- a/tailwind.config.js
+++ b/tailwind.config.js
@@ -1,6 +1,7 @@
 /** @type {import('tailwindcss').Config} */
 module.exports = {
   content: ["./app/**/*.{js,ts,jsx,tsx}", "./pages/**/*.{js,ts,jsx,tsx}", "./components/**/*.{js,ts,jsx,tsx}"],
+  darkMode: "class",
   theme: {
     extend: {}
   },
diff --git a/types/index.ts b/types/index.ts
index e0df43b..aa30a31 100644
--- a/types/index.ts
+++ b/types/index.ts
@@ -1,7 +1,15 @@
 export enum OpenAIModel {
-  DAVINCI_TURBO = "gpt-3.5-turbo"
+  GPT_3_5 = "gpt-3.5-turbo",
+  GPT_3_5_LEGACY = "gpt-3.5-turbo-0301"
+  // GPT_4 = "gpt-4"
 }
 
+export const OpenAIModelNames: Record<OpenAIModel, string> = {
+  [OpenAIModel.GPT_3_5]: "Default (GPT-3.5)",
+  [OpenAIModel.GPT_3_5_LEGACY]: "Legacy (GPT-3.5)"
+  // [OpenAIModel.GPT_4]: "GPT-4"
+};
+
 export interface Message {
   role: Role;
   content: string;
diff --git a/utils/index.ts b/utils/index.ts
index 2597151..b8c9836 100644
--- a/utils/index.ts
+++ b/utils/index.ts
@@ -1,7 +1,7 @@
 import { Message, OpenAIModel } from "@/types";
 import { createParser, ParsedEvent, ReconnectInterval } from "eventsource-parser";
 
-export const OpenAIStream = async (messages: Message[]) => {
+export const OpenAIStream = async (model: OpenAIModel, messages: Message[]) => {
   const encoder = new TextEncoder();
   const decoder = new TextDecoder();
 
@@ -12,7 +12,7 @@ export const OpenAIStream = async (messages: Message[]) => {
     },
     method: "POST",
     body: JSON.stringify({
-      model: OpenAIModel.DAVINCI_TURBO,
+      model,
       messages: [
         {
           role: "system",