From e87136c4ecb548a953e86a60251311b1d51be033 Mon Sep 17 00:00:00 2001
From: Mckay Wrigley <mckaywrigley@gmail.com>
Date: Wed, 15 Mar 2023 04:46:57 -0600
Subject: [PATCH] single save

---
 pages/index.tsx | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/pages/index.tsx b/pages/index.tsx
index bf23ae9..0a374a6 100644
--- a/pages/index.tsx
+++ b/pages/index.tsx
@@ -44,12 +44,15 @@ export default function Home() {
     const decoder = new TextDecoder();
     let done = false;
     let isFirst = true;
+    let text = "";
 
     while (!done) {
       const { value, done: doneReading } = await reader.read();
       done = doneReading;
       const chunkValue = decoder.decode(value);
 
+      text += chunkValue;
+
       if (isFirst) {
         isFirst = false;
         setMessages((messages) => [
@@ -70,9 +73,29 @@ export default function Home() {
         });
       }
     }
+
+    localStorage.setItem("messageHistory", JSON.stringify([...updatedMessages, { role: "assistant", content: text }]));
   };
 
-  useEffect(() => {}, []);
+  const handleLightMode = (mode: "dark" | "light") => {
+    setLightMode(mode);
+    localStorage.setItem("theme", mode);
+  };
+
+  useEffect(() => {
+    const theme = localStorage.getItem("theme");
+
+    if (theme) {
+      setLightMode(theme as "dark" | "light");
+    }
+
+    const messageHistory = localStorage.getItem("messageHistory");
+    console.log(messageHistory);
+
+    if (messageHistory) {
+      setMessages(JSON.parse(messageHistory));
+    }
+  }, []);
 
   return (
     <>
@@ -95,7 +118,7 @@ export default function Home() {
       <div className={`flex h-screen text-white ${lightMode}`}>
         <Sidebar
           lightMode={lightMode}
-          onToggleLightMode={setLightMode}
+          onToggleLightMode={handleLightMode}
         />
 
         <div className="flex flex-col w-full h-full dark:bg-[#343541]">