diff --git a/src/server.ts b/src/server.ts
index 52570f2..5da71c6 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -102,7 +102,8 @@ const displayMessages = (data: any): string => {
 </div>
     <div  class="px-6 pb-6 rounded-md ${
       msg.role === "assistant" ? "bg-gray-200" : msg.role === "system" ? "bg-pink-400 text-white" : "bg-blue-500 text-white"
-    } break-words whitespace-pre-wrap font-mono" >
+      <textarea id="message-content-${index}" class="w-full bg-transparent border-none focus:outline-none focus:ring-0">${msg.content}</textarea>
+      <button onclick="saveMessage(${index})" class="bg-pink-300 rounded-md p-2 hover:bg-pink-500 hover:text-white">Save</button>
       ${msg.content}
     </div>
   `,
@@ -129,6 +130,23 @@ app.get(
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>${id} | Context Viewer</title>
   <script src="https://cdn.tailwindcss.com"></script>
+  <script>
+  const saveMessage = async (index) => {
+    const content = document.getElementById('message-content-' + index).value
+    const res = await fetch('/context/${id}/save/', {
+      method: 'POST',
+      body: JSON.stringify({
+        index,
+        content,
+      })
+    })
+    if (res.ok) {
+      alert('Saved!')
+    } else {
+      alert('Failed to save')
+    }
+  }
+  </script>
 </head>
 <body class="bg-gray-100 flex flex-row">
 <div id="top" class="sticky top-5 mt-2 m-auto w-full text-center">
@@ -171,6 +189,28 @@ ${window.meta.final_state.content}
   },
 )
 
+app.post(
+  "/context/:id/save/",
+  ssgParams(async () => {
+    const windows = await getMetadata()
+    return windows.map((w) => ({ id: w.id }))
+  }),
+  async (c) => {
+    const id = c.req.param("id")
+    const { index, content } = await c.req.json()
+
+    try {
+      const window = await getData(id)
+      window.messages[index].content = content
+
+      const outStream = fs.createWriteStream("./context_windows.jsonl", { flags: "w" })
+      outStream.write(JSON.stringify(window) + "\n")
+      outStream.end()
+
+      return c.text("Saved!")
+    } catch (e: any) {
+      if (e instanceof Error) {
+        return c.text(e.message, 500)
 app.get(
   "/context/:id/json/",
   ssgParams(async () => {
