546 lines
16 KiB
TypeScript
546 lines
16 KiB
TypeScript
import { Chat } from "@/components/Chat/Chat";
|
|
import { Navbar } from "@/components/Mobile/Navbar";
|
|
import { Sidebar } from "@/components/Sidebar/Sidebar";
|
|
import { ChatBody, ChatFolder, Conversation, KeyValuePair, Message, OpenAIModel, OpenAIModelID, OpenAIModels } from "@/types";
|
|
import { cleanConversationHistory, cleanSelectedConversation } from "@/utils/app/clean";
|
|
import { DEFAULT_SYSTEM_PROMPT } from "@/utils/app/const";
|
|
import { saveConversation, saveConversations, updateConversation } from "@/utils/app/conversation";
|
|
import { saveFolders } from "@/utils/app/folders";
|
|
import { exportData, importData } from "@/utils/app/importExport";
|
|
import { IconArrowBarLeft, IconArrowBarRight } from "@tabler/icons-react";
|
|
import { GetServerSideProps } from "next";
|
|
import Head from "next/head";
|
|
import { useEffect, useRef, useState } from "react";
|
|
|
|
interface HomeProps {
|
|
serverSideApiKeyIsSet: boolean;
|
|
}
|
|
|
|
const Home: React.FC<HomeProps> = ({ serverSideApiKeyIsSet }) => {
|
|
const [folders, setFolders] = useState<ChatFolder[]>([]);
|
|
const [conversations, setConversations] = useState<Conversation[]>([]);
|
|
const [selectedConversation, setSelectedConversation] = useState<Conversation>();
|
|
const [loading, setLoading] = useState<boolean>(false);
|
|
const [models, setModels] = useState<OpenAIModel[]>([]);
|
|
const [lightMode, setLightMode] = useState<"dark" | "light">("dark");
|
|
const [messageIsStreaming, setMessageIsStreaming] = useState<boolean>(false);
|
|
const [showSidebar, setShowSidebar] = useState<boolean>(true);
|
|
const [apiKey, setApiKey] = useState<string>("");
|
|
const [messageError, setMessageError] = useState<boolean>(false);
|
|
const [modelError, setModelError] = useState<boolean>(false);
|
|
const [currentMessage, setCurrentMessage] = useState<Message>();
|
|
|
|
const stopConversationRef = useRef<boolean>(false);
|
|
|
|
const handleSend = async (message: Message, isResend: boolean) => {
|
|
if (selectedConversation) {
|
|
let updatedConversation: Conversation;
|
|
|
|
if (isResend) {
|
|
const updatedMessages = [...selectedConversation.messages];
|
|
updatedMessages.pop();
|
|
|
|
updatedConversation = {
|
|
...selectedConversation,
|
|
messages: [...updatedMessages, message]
|
|
};
|
|
} else {
|
|
updatedConversation = {
|
|
...selectedConversation,
|
|
messages: [...selectedConversation.messages, message]
|
|
};
|
|
}
|
|
|
|
setSelectedConversation(updatedConversation);
|
|
setLoading(true);
|
|
setMessageIsStreaming(true);
|
|
setMessageError(false);
|
|
|
|
const chatBody: ChatBody = {
|
|
model: updatedConversation.model,
|
|
messages: updatedConversation.messages,
|
|
key: apiKey,
|
|
prompt: updatedConversation.prompt
|
|
};
|
|
|
|
const controller = new AbortController();
|
|
const response = await fetch("/api/chat", {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json"
|
|
},
|
|
signal: controller.signal,
|
|
body: JSON.stringify(chatBody)
|
|
});
|
|
|
|
if (!response.ok) {
|
|
setLoading(false);
|
|
setMessageIsStreaming(false);
|
|
setMessageError(true);
|
|
return;
|
|
}
|
|
|
|
const data = response.body;
|
|
|
|
if (!data) {
|
|
setLoading(false);
|
|
setMessageIsStreaming(false);
|
|
setMessageError(true);
|
|
|
|
return;
|
|
}
|
|
|
|
if (updatedConversation.messages.length === 1) {
|
|
const { content } = message;
|
|
const customName = content.length > 30 ? content.substring(0, 30) + "..." : content;
|
|
|
|
updatedConversation = {
|
|
...updatedConversation,
|
|
name: customName
|
|
};
|
|
}
|
|
|
|
setLoading(false);
|
|
|
|
const reader = data.getReader();
|
|
const decoder = new TextDecoder();
|
|
let done = false;
|
|
let isFirst = true;
|
|
let text = "";
|
|
|
|
while (!done) {
|
|
if (stopConversationRef.current === true) {
|
|
controller.abort();
|
|
done = true;
|
|
break;
|
|
}
|
|
const { value, done: doneReading } = await reader.read();
|
|
done = doneReading;
|
|
const chunkValue = decoder.decode(value);
|
|
|
|
text += chunkValue;
|
|
|
|
if (isFirst) {
|
|
isFirst = false;
|
|
const updatedMessages: Message[] = [...updatedConversation.messages, { role: "assistant", content: chunkValue }];
|
|
|
|
updatedConversation = {
|
|
...updatedConversation,
|
|
messages: updatedMessages
|
|
};
|
|
|
|
setSelectedConversation(updatedConversation);
|
|
} else {
|
|
const updatedMessages: Message[] = updatedConversation.messages.map((message, index) => {
|
|
if (index === updatedConversation.messages.length - 1) {
|
|
return {
|
|
...message,
|
|
content: text
|
|
};
|
|
}
|
|
|
|
return message;
|
|
});
|
|
|
|
updatedConversation = {
|
|
...updatedConversation,
|
|
messages: updatedMessages
|
|
};
|
|
|
|
setSelectedConversation(updatedConversation);
|
|
}
|
|
}
|
|
|
|
saveConversation(updatedConversation);
|
|
|
|
const updatedConversations: Conversation[] = conversations.map((conversation) => {
|
|
if (conversation.id === selectedConversation.id) {
|
|
return updatedConversation;
|
|
}
|
|
|
|
return conversation;
|
|
});
|
|
|
|
if (updatedConversations.length === 0) {
|
|
updatedConversations.push(updatedConversation);
|
|
}
|
|
|
|
setConversations(updatedConversations);
|
|
|
|
saveConversations(updatedConversations);
|
|
|
|
setMessageIsStreaming(false);
|
|
}
|
|
};
|
|
|
|
const fetchModels = async (key: string) => {
|
|
const response = await fetch("/api/models", {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json"
|
|
},
|
|
body: JSON.stringify({
|
|
key
|
|
})
|
|
});
|
|
|
|
if (!response.ok) {
|
|
setModelError(true);
|
|
return;
|
|
}
|
|
|
|
const data = await response.json();
|
|
|
|
if (!data) {
|
|
setModelError(true);
|
|
return;
|
|
}
|
|
|
|
setModels(data);
|
|
setModelError(false);
|
|
};
|
|
|
|
const handleLightMode = (mode: "dark" | "light") => {
|
|
setLightMode(mode);
|
|
localStorage.setItem("theme", mode);
|
|
};
|
|
|
|
const handleApiKeyChange = (apiKey: string) => {
|
|
setApiKey(apiKey);
|
|
localStorage.setItem("apiKey", apiKey);
|
|
};
|
|
|
|
const handleExportData = () => {
|
|
exportData();
|
|
};
|
|
|
|
const handleImportConversations = (data: { conversations: Conversation[]; folders: ChatFolder[] }) => {
|
|
importData(data.conversations, data.folders);
|
|
setConversations(data.conversations);
|
|
setSelectedConversation(data.conversations[data.conversations.length - 1]);
|
|
setFolders(data.folders);
|
|
};
|
|
|
|
const handleSelectConversation = (conversation: Conversation) => {
|
|
setSelectedConversation(conversation);
|
|
saveConversation(conversation);
|
|
};
|
|
|
|
const handleCreateFolder = (name: string) => {
|
|
const lastFolder = folders[folders.length - 1];
|
|
|
|
const newFolder: ChatFolder = {
|
|
id: lastFolder ? lastFolder.id + 1 : 1,
|
|
name
|
|
};
|
|
|
|
const updatedFolders = [...folders, newFolder];
|
|
|
|
setFolders(updatedFolders);
|
|
saveFolders(updatedFolders);
|
|
};
|
|
|
|
const handleDeleteFolder = (folderId: number) => {
|
|
const updatedFolders = folders.filter((f) => f.id !== folderId);
|
|
setFolders(updatedFolders);
|
|
saveFolders(updatedFolders);
|
|
|
|
const updatedConversations: Conversation[] = conversations.map((c) => {
|
|
if (c.folderId === folderId) {
|
|
return {
|
|
...c,
|
|
folderId: 0
|
|
};
|
|
}
|
|
|
|
return c;
|
|
});
|
|
setConversations(updatedConversations);
|
|
saveConversations(updatedConversations);
|
|
};
|
|
|
|
const handleUpdateFolder = (folderId: number, name: string) => {
|
|
const updatedFolders = folders.map((f) => {
|
|
if (f.id === folderId) {
|
|
return {
|
|
...f,
|
|
name
|
|
};
|
|
}
|
|
|
|
return f;
|
|
});
|
|
|
|
setFolders(updatedFolders);
|
|
saveFolders(updatedFolders);
|
|
};
|
|
|
|
const handleNewConversation = () => {
|
|
const lastConversation = conversations[conversations.length - 1];
|
|
|
|
const newConversation: Conversation = {
|
|
id: lastConversation ? lastConversation.id + 1 : 1,
|
|
name: `Conversation ${lastConversation ? lastConversation.id + 1 : 1}`,
|
|
messages: [],
|
|
model: OpenAIModels[OpenAIModelID.GPT_3_5],
|
|
prompt: DEFAULT_SYSTEM_PROMPT,
|
|
folderId: 0
|
|
};
|
|
|
|
const updatedConversations = [...conversations, newConversation];
|
|
|
|
setSelectedConversation(newConversation);
|
|
setConversations(updatedConversations);
|
|
|
|
saveConversation(newConversation);
|
|
saveConversations(updatedConversations);
|
|
|
|
setLoading(false);
|
|
};
|
|
|
|
const handleDeleteConversation = (conversation: Conversation) => {
|
|
const updatedConversations = conversations.filter((c) => c.id !== conversation.id);
|
|
setConversations(updatedConversations);
|
|
saveConversations(updatedConversations);
|
|
|
|
if (updatedConversations.length > 0) {
|
|
setSelectedConversation(updatedConversations[updatedConversations.length - 1]);
|
|
saveConversation(updatedConversations[updatedConversations.length - 1]);
|
|
} else {
|
|
setSelectedConversation({
|
|
id: 1,
|
|
name: "New conversation",
|
|
messages: [],
|
|
model: OpenAIModels[OpenAIModelID.GPT_3_5],
|
|
prompt: DEFAULT_SYSTEM_PROMPT,
|
|
folderId: 0
|
|
});
|
|
localStorage.removeItem("selectedConversation");
|
|
}
|
|
};
|
|
|
|
const handleUpdateConversation = (conversation: Conversation, data: KeyValuePair) => {
|
|
const updatedConversation = {
|
|
...conversation,
|
|
[data.key]: data.value
|
|
};
|
|
|
|
const { single, all } = updateConversation(updatedConversation, conversations);
|
|
|
|
setSelectedConversation(single);
|
|
setConversations(all);
|
|
};
|
|
|
|
const handleClearConversations = () => {
|
|
setConversations([]);
|
|
localStorage.removeItem("conversationHistory");
|
|
|
|
setSelectedConversation({
|
|
id: 1,
|
|
name: "New conversation",
|
|
messages: [],
|
|
model: OpenAIModels[OpenAIModelID.GPT_3_5],
|
|
prompt: DEFAULT_SYSTEM_PROMPT,
|
|
folderId: 0
|
|
});
|
|
localStorage.removeItem("selectedConversation");
|
|
|
|
setFolders([]);
|
|
localStorage.removeItem("folders");
|
|
};
|
|
|
|
const handleEditMessage = (message: Message, messageIndex: number) => {
|
|
if (selectedConversation) {
|
|
const updatedMessages = selectedConversation.messages
|
|
.map((m, i) => {
|
|
if (i < messageIndex) {
|
|
return m;
|
|
}
|
|
})
|
|
.filter((m) => m) as Message[];
|
|
|
|
const updatedConversation = {
|
|
...selectedConversation,
|
|
messages: updatedMessages
|
|
};
|
|
|
|
const { single, all } = updateConversation(updatedConversation, conversations);
|
|
|
|
setSelectedConversation(single);
|
|
setConversations(all);
|
|
|
|
setCurrentMessage(message);
|
|
}
|
|
};
|
|
|
|
const handleDeleteMessage = (message: Message, messageIndex: number) => {};
|
|
|
|
const handleRegenerate = () => {};
|
|
|
|
useEffect(() => {
|
|
if (currentMessage) {
|
|
handleSend(currentMessage, false);
|
|
setCurrentMessage(undefined);
|
|
}
|
|
}, [currentMessage]);
|
|
|
|
useEffect(() => {
|
|
if (window.innerWidth < 640) {
|
|
setShowSidebar(false);
|
|
}
|
|
}, [selectedConversation]);
|
|
|
|
useEffect(() => {
|
|
if (apiKey) {
|
|
fetchModels(apiKey);
|
|
}
|
|
}, [apiKey]);
|
|
|
|
useEffect(() => {
|
|
const theme = localStorage.getItem("theme");
|
|
if (theme) {
|
|
setLightMode(theme as "dark" | "light");
|
|
}
|
|
|
|
const apiKey = localStorage.getItem("apiKey");
|
|
if (apiKey) {
|
|
setApiKey(apiKey);
|
|
fetchModels(apiKey);
|
|
} else if (serverSideApiKeyIsSet) {
|
|
fetchModels("");
|
|
}
|
|
|
|
if (window.innerWidth < 640) {
|
|
setShowSidebar(false);
|
|
}
|
|
|
|
const folders = localStorage.getItem("folders");
|
|
if (folders) {
|
|
setFolders(JSON.parse(folders));
|
|
}
|
|
|
|
const conversationHistory = localStorage.getItem("conversationHistory");
|
|
if (conversationHistory) {
|
|
const parsedConversationHistory: Conversation[] = JSON.parse(conversationHistory);
|
|
const cleanedConversationHistory = cleanConversationHistory(parsedConversationHistory);
|
|
setConversations(cleanedConversationHistory);
|
|
}
|
|
|
|
const selectedConversation = localStorage.getItem("selectedConversation");
|
|
if (selectedConversation) {
|
|
const parsedSelectedConversation: Conversation = JSON.parse(selectedConversation);
|
|
const cleanedSelectedConversation = cleanSelectedConversation(parsedSelectedConversation);
|
|
setSelectedConversation(cleanedSelectedConversation);
|
|
} else {
|
|
setSelectedConversation({
|
|
id: 1,
|
|
name: "New conversation",
|
|
messages: [],
|
|
model: OpenAIModels[OpenAIModelID.GPT_3_5],
|
|
prompt: DEFAULT_SYSTEM_PROMPT,
|
|
folderId: 0
|
|
});
|
|
}
|
|
}, [serverSideApiKeyIsSet]);
|
|
|
|
return (
|
|
<>
|
|
<Head>
|
|
<title>Chatbot UI</title>
|
|
<meta
|
|
name="description"
|
|
content="ChatGPT but better."
|
|
/>
|
|
<meta
|
|
name="viewport"
|
|
content="width=device-width, initial-scale=1"
|
|
/>
|
|
<link
|
|
rel="icon"
|
|
href="/favicon.ico"
|
|
/>
|
|
</Head>
|
|
{selectedConversation && (
|
|
<main className={`flex flex-col h-screen w-screen text-white dark:text-white text-sm ${lightMode}`}>
|
|
<div className="sm:hidden w-full fixed top-0">
|
|
<Navbar
|
|
selectedConversation={selectedConversation}
|
|
onNewConversation={handleNewConversation}
|
|
/>
|
|
</div>
|
|
|
|
<div className="flex h-full w-full pt-[48px] sm:pt-0">
|
|
{showSidebar ? (
|
|
<div>
|
|
<Sidebar
|
|
loading={messageIsStreaming}
|
|
conversations={conversations}
|
|
lightMode={lightMode}
|
|
selectedConversation={selectedConversation}
|
|
apiKey={apiKey}
|
|
folders={folders}
|
|
onToggleLightMode={handleLightMode}
|
|
onCreateFolder={handleCreateFolder}
|
|
onDeleteFolder={handleDeleteFolder}
|
|
onUpdateFolder={handleUpdateFolder}
|
|
onNewConversation={handleNewConversation}
|
|
onSelectConversation={handleSelectConversation}
|
|
onDeleteConversation={handleDeleteConversation}
|
|
onToggleSidebar={() => setShowSidebar(!showSidebar)}
|
|
onUpdateConversation={handleUpdateConversation}
|
|
onApiKeyChange={handleApiKeyChange}
|
|
onClearConversations={handleClearConversations}
|
|
onExportConversations={handleExportData}
|
|
onImportConversations={handleImportConversations}
|
|
/>
|
|
|
|
<IconArrowBarLeft
|
|
className="z-50 fixed top-5 left-[270px] sm:top-0.5 sm:left-[270px] sm:text-neutral-700 dark:text-white cursor-pointer hover:text-gray-400 dark:hover:text-gray-300 h-7 w-7 sm:h-8 sm:w-8"
|
|
onClick={() => setShowSidebar(!showSidebar)}
|
|
/>
|
|
|
|
<div
|
|
onClick={() => setShowSidebar(!showSidebar)}
|
|
className="sm:hidden bg-black opacity-70 z-10 absolute top-0 left-0 h-full w-full"
|
|
></div>
|
|
</div>
|
|
) : (
|
|
<IconArrowBarRight
|
|
className="fixed text-white z-50 top-2.5 left-4 sm:top-0.5 sm:left-4 sm:text-neutral-700 dark:text-white cursor-pointer hover:text-gray-400 dark:hover:text-gray-300 h-7 w-7 sm:h-8 sm:w-8"
|
|
onClick={() => setShowSidebar(!showSidebar)}
|
|
/>
|
|
)}
|
|
|
|
<Chat
|
|
conversation={selectedConversation}
|
|
messageIsStreaming={messageIsStreaming}
|
|
apiKey={apiKey}
|
|
serverSideApiKeyIsSet={serverSideApiKeyIsSet}
|
|
modelError={modelError}
|
|
messageError={messageError}
|
|
models={models}
|
|
loading={loading}
|
|
lightMode={lightMode}
|
|
onSend={handleSend}
|
|
onUpdateConversation={handleUpdateConversation}
|
|
onEditMessage={handleEditMessage}
|
|
onDeleteMessage={handleDeleteMessage}
|
|
onRegenerate={handleRegenerate}
|
|
stopConversationRef={stopConversationRef}
|
|
/>
|
|
</div>
|
|
</main>
|
|
)}
|
|
</>
|
|
);
|
|
};
|
|
export default Home;
|
|
|
|
export const getServerSideProps: GetServerSideProps = async () => {
|
|
return {
|
|
props: {
|
|
serverSideApiKeyIsSet: !!process.env.OPENAI_API_KEY
|
|
}
|
|
};
|
|
};
|