From 3f56e13ecb1f4d476447d5f948799982d73ebe0f Mon Sep 17 00:00:00 2001
From: skeptrune <nick.k@trieve.ai>
Date: Tue, 19 Nov 2024 21:20:55 -0800
Subject: [PATCH 1/2] feature: include cost estimate on pdf view

---
 pdf2md/server/src/operators/s3.rs        |  2 +-
 pdf2md/server/src/templates/demo-ui.html | 27 +++++++-----
 pdf2md/server/static/output.css          |  8 +++-
 pdf2md/server/static/pdf2md.js           | 53 ++++++++++++++++++++++--
 4 files changed, 72 insertions(+), 18 deletions(-)

diff --git a/pdf2md/server/src/operators/s3.rs b/pdf2md/server/src/operators/s3.rs
index 48deb6cb0b..4e90e3e4c2 100644
--- a/pdf2md/server/src/operators/s3.rs
+++ b/pdf2md/server/src/operators/s3.rs
@@ -45,7 +45,7 @@ pub async fn get_signed_url(bucket: &Bucket, key: &str) -> Result<String, Servic
     custom_queries.insert("response-content-encoding".into(), "utf-8".into());
 
     let url = bucket
-        .presign_get(key, 3600, Some(custom_queries))
+        .presign_get(key, 86400, Some(custom_queries))
         .await
         .map_err(|e| {
             log::error!("Could not get signed url {:?}", e);
diff --git a/pdf2md/server/src/templates/demo-ui.html b/pdf2md/server/src/templates/demo-ui.html
index fd463231d4..ee0e5a16b2 100644
--- a/pdf2md/server/src/templates/demo-ui.html
+++ b/pdf2md/server/src/templates/demo-ui.html
@@ -23,7 +23,7 @@
   id="upload-form-container"
   class="flex justify-center mt-[10vh] sm:mt-[20vh] items-center px-4"
 >
-  <form class="max-w-lg mx-auto">
+  <form class="max-w-2xl mx-auto">
     <div class="text-center">
       <h2
         class="text-balance text-4xl font-semibold tracking-tight text-gray-900 sm:text-5xl"
@@ -31,7 +31,7 @@
         OCR With Intelligence
       </h2>
       <p class="mt-2 text-pretty text-lg/8 text-gray-700">
-        Convert any PDF to LLM-ready Markdown using vision models like GPT-4o.
+        Convert any PDF to LLM-ready Markdown using thrifty vision models like GPT-4o-mini and Gemini-flash-1.5.
       </p>
     </div>
     <button
@@ -75,10 +75,15 @@
         name="model"
         class="mt-2 block w-full rounded-md border-0 py-1.5 pl-3 pr-10 text-gray-900 bg-white ring-1 ring-inset ring-gray-300 focus:ring-2 focus:ring-magenta-600 sm:text-sm/6"
       >
-        <option value="openai/gpt-4o-mini" selected>4o-mini</option>
-        <option value="openai/chatgpt-4o-latest">4o</option>
-        <option value="google/gemini-flash-1.5-8b">gemini-flash-1.5-8b</option>
-        <option value="google/gemini-flash-1.5">gemini-flash-1.5</option>
+        <option value="openai/gpt-4o-mini" selected>
+          4o-mini ($0.15/M input)
+        </option>
+        <option value="google/gemini-flash-1.5-8b">
+          gemini-flash-1.5-8b ($0.0375/M input)
+        </option>
+        <option value="google/gemini-flash-1.5">
+          gemini-flash-1.5 ($0.075/M input)
+        </option>
       </select>
       <label
         for="conversion-prompt"
@@ -134,11 +139,11 @@
     </div>
   </form>
 </div>
-<div class="px-4">
-  <div
-    id="result-container"
-    class="mt-10 sm:mt-14 md:mt-24 grid grid-cols-2 gap-4 max-w-screen-2xl mx-auto"
-  >
+<div class="px-4 mt-10 sm:mt-14 md:mt-24 max-w-screen-2xl mx-auto">
+  <div class="col-span-2 flex justify-between">
+    <p id="document-price" class="text-gray-600 text-sm/6"></p>
+  </div>
+  <div id="result-container" class="grid grid-cols-2 gap-4">
     <div id="my-pdf"></div>
     <div id="markdown-container" class="max-h-[75vh] overflow-y-auto"></div>
   </div>
diff --git a/pdf2md/server/static/output.css b/pdf2md/server/static/output.css
index a9df9a87e1..1edc5cd7b7 100644
--- a/pdf2md/server/static/output.css
+++ b/pdf2md/server/static/output.css
@@ -570,6 +570,10 @@ video {
   position: relative;
 }
 
+.col-span-2 {
+  grid-column: span 2 / span 2;
+}
+
 .-mx-4 {
   margin-left: -1rem;
   margin-right: -1rem;
@@ -677,8 +681,8 @@ video {
   min-width: 100%;
 }
 
-.max-w-lg {
-  max-width: 32rem;
+.max-w-2xl {
+  max-width: 42rem;
 }
 
 .max-w-screen-2xl {
diff --git a/pdf2md/server/static/pdf2md.js b/pdf2md/server/static/pdf2md.js
index 9cb7b19948..d08d8d0fef 100644
--- a/pdf2md/server/static/pdf2md.js
+++ b/pdf2md/server/static/pdf2md.js
@@ -1,3 +1,18 @@
+const costsPerM = {
+  "openai/gpt-4o-mini": {
+    input_price: 0.15,
+    output_price: 0.6,
+  },
+  "google/gemini-flash-1.5-8b": {
+    input_price: 0.0375,
+    output_price: 0.15,
+  },
+  "google/gemini-flash-1.5": {
+    input_price: 0.075,
+    output_price: 0.3,
+  },
+};
+
 const defaultTableRowStr = `
     <td
       class="task-id whitespace-nowrap py-4 pl-4 pr-3 text-sm font-medium text-gray-900 sm:pl-0"
@@ -83,6 +98,9 @@ const displayTask = (task) => {
     task.pages_processed
   );
 
+  let completionTokens = 0;
+  let promptTokens = 0;
+
   sortedPages.forEach((page) => {
     const pageContainer = document.createElement("div");
     pageContainer.classList.add("page-container");
@@ -91,6 +109,9 @@ const displayTask = (task) => {
     const spacerDiv = document.createElement("div");
     spacerDiv.classList.add(...["my-4", "h-1", "bg-gray-700"]);
     markdownContainer.appendChild(spacerDiv);
+
+    promptTokens += page?.usage?.prompt_tokens || 0;
+    completionTokens += page?.usage?.completion_tokens || 0;
   });
   if (!sortedPages.length) {
     const pageContainer = document.createElement("div");
@@ -99,9 +120,33 @@ const displayTask = (task) => {
       "Your file is being converted. We are pinging the server every 5 seconds to check for status updates. See the table below for more detailed status information. Please be patient!";
     markdownContainer.appendChild(pageContainer);
   }
+
+  const modelSelect = document.getElementById("model");
+  const model = modelSelect.options[modelSelect.selectedIndex].value;
+  const costPerM = costsPerM[model];
+  const completionCost = (completionTokens * costPerM.output_price) / 1000000;
+  const promptCost = (promptTokens * costPerM.input_price) / 1000000;
+  document.getElementById("document-price").innerText =
+    `Cost: ${new Intl.NumberFormat("en-US", {
+      style: "currency",
+      currency: "USD",
+      minimumFractionDigits: 5,
+    }).format(promptCost)} (Prompt) + ${new Intl.NumberFormat("en-US", {
+      style: "currency",
+      currency: "USD",
+      minimumFractionDigits: 5,
+    }).format(completionCost)} (Completion) = ${new Intl.NumberFormat("en-US", {
+      style: "currency",
+      currency: "USD",
+      minimumFractionDigits: 5,
+    }).format(completionCost + promptCost)}`;
 };
 
 const getTaskPages = async (taskId, taskIdToDisplay) => {
+  if (!taskId) {
+    return;
+  }
+
   try {
     let paginationToken = "";
     let task = null;
@@ -119,7 +164,7 @@ const getTaskPages = async (taskId, taskIdToDisplay) => {
       );
       const taskWithPages = await resp.json();
       task = taskWithPages;
-      pages.push(...taskWithPages.pages);
+      pages.push(...(taskWithPages.pages || []));
       paginationToken = taskWithPages.pagination_token;
       if (!paginationToken) {
         break;
@@ -243,13 +288,13 @@ const updateTaskStatusTable = () => {
     row.querySelector(".task-id").innerText = task.id;
     row.querySelector(".task-file-name").innerText = task.file_name;
     row.querySelector(".task-status").innerText =
-      task.status.toLowerCase() === "completed"
+      task.status?.toLowerCase() === "completed"
         ? task.status
         : `${task.status} | Please wait. Checking for updates every 5 seconds.`;
     row
       .querySelector(".task-status")
       .classList.add(
-        `status-${task.status.split(" ").join("-").toLowerCase()}`
+        `status-${task.status?.split(" ").join("-").toLowerCase()}`
       );
     row.querySelector(".task-prompt-tokens").innerText = promptTokens;
     row.querySelector(".task-completion-tokens").innerText = completionTokens;
@@ -282,7 +327,7 @@ const refreshTasks = () => {
   const taskIdToDisplay = url.searchParams.get("taskId");
   tasks.forEach((task) => {
     if (
-      task.status.toLowerCase() === "completed" &&
+      task.status?.toLowerCase() === "completed" &&
       task.pages &&
       task.pages.length
     ) {

From 4e45bb024bceaedf2a1428be84da983fd9606482 Mon Sep 17 00:00:00 2001
From: skeptrune <nick.k@trieve.ai>
Date: Tue, 19 Nov 2024 22:01:33 -0800
Subject: [PATCH 2/2] feature: add json toggle + copy button to pdf2md

---
 pdf2md/server/src/templates/demo-ui.html |  64 +++++++++-
 pdf2md/server/static/output.css          | 153 +++++++++++++++++++++++
 pdf2md/server/static/pdf2md.js           |  47 +++++++
 3 files changed, 260 insertions(+), 4 deletions(-)

diff --git a/pdf2md/server/src/templates/demo-ui.html b/pdf2md/server/src/templates/demo-ui.html
index ee0e5a16b2..2d459b760a 100644
--- a/pdf2md/server/src/templates/demo-ui.html
+++ b/pdf2md/server/src/templates/demo-ui.html
@@ -9,6 +9,15 @@
   rel="stylesheet"
   href="https://cdn.jsdelivr.net/npm/notyf@3/notyf.min.css"
 />
+<link
+  rel="stylesheet"
+  href="https://unpkg.com/@highlightjs/cdn-assets@11.9.0/styles/default.min.css"
+/>
+<script src="https://unpkg.com/@highlightjs/cdn-assets@11.9.0/highlight.min.js"></script>
+<script src="https://unpkg.com/@highlightjs/cdn-assets@11.9.0/languages/json.min.js"></script>
+<script>
+  hljs.highlightAll();
+</script>
 <style>
   .pdfobject-container {
     height: 75vh;
@@ -31,7 +40,8 @@
         OCR With Intelligence
       </h2>
       <p class="mt-2 text-pretty text-lg/8 text-gray-700">
-        Convert any PDF to LLM-ready Markdown using thrifty vision models like GPT-4o-mini and Gemini-flash-1.5.
+        Convert any PDF to LLM-ready Markdown using thrifty vision models like
+        GPT-4o-mini and Gemini-flash-1.5.
       </p>
     </div>
     <button
@@ -140,17 +150,63 @@
   </form>
 </div>
 <div class="px-4 mt-10 sm:mt-14 md:mt-24 max-w-screen-2xl mx-auto">
-  <div class="col-span-2 flex justify-between">
+  <div class="col-span-2 flex justify-between items-center py-1">
     <p id="document-price" class="text-gray-600 text-sm/6"></p>
+    <div class="flex items-center gap-2">
+      <button
+        type="button"
+        class="inline-flex items-center gap-x-1.5 rounded-md bg-magenta-600 px-2.5 py-1.5 text-sm font-semibold text-white shadow-sm hover:bg-magenta-500 focus-visible:outline focus-visible:outline-2 focus-visible:outline-offset-2 focus-visible:outline-magenta-600"
+        id="copy-button"
+      >
+        <svg
+          xmlns="http://www.w3.org/2000/svg"
+          width="16"
+          height="16"
+          fill="currentColor"
+          class="bi bi-copy"
+          viewBox="0 0 16 16"
+        >
+          <path
+            fill-rule="evenodd"
+            d="M4 2a2 2 0 0 1 2-2h8a2 2 0 0 1 2 2v8a2 2 0 0 1-2 2H6a2 2 0 0 1-2-2zm2-1a1 1 0 0 0-1 1v8a1 1 0 0 0 1 1h8a1 1 0 0 0 1-1V2a1 1 0 0 0-1-1zM2 5a1 1 0 0 0-1 1v8a1 1 0 0 0 1 1h8a1 1 0 0 0 1-1v-1h1v1a2 2 0 0 1-2 2H2a2 2 0 0 1-2-2V6a2 2 0 0 1 2-2h1v1z"
+          />
+        </svg>
+        Copy
+      </button>
+      <div class="flex items-center">
+        <button
+          type="button"
+          class="relative inline-flex h-6 w-11 shrink-0 cursor-pointer rounded-full border-2 border-transparent bg-gray-200 transition-colors duration-200 ease-in-out focus:outline-none focus:ring-2 focus:ring-magenta-600 focus:ring-offset-2"
+          role="switch"
+          aria-checked="false"
+          aria-labelledby="json"
+          id="json-switch"
+        >
+          <span
+            aria-hidden="true"
+            class="pointer-events-none inline-block size-5 translate-x-0 transform rounded-full bg-white shadow ring-0 transition duration-200 ease-in-out"
+          ></span>
+        </button>
+        <span class="ml-3 text-sm" id="json">
+          <span class="font-medium text-gray-900">Json</span>
+        </span>
+      </div>
+    </div>
   </div>
   <div id="result-container" class="grid grid-cols-2 gap-4">
     <div id="my-pdf"></div>
     <div id="markdown-container" class="max-h-[75vh] overflow-y-auto"></div>
+    <pre id="json-pre" class="hidden max-h-[75vh] overflow-y-auto">
+      <code class="language-json" id="json-container"></code>
+    </pre>
   </div>
 </div>
 <div class="flow-root">
-  <div class="mt-10 sm:mt-14 md:mt-24 pt-4 border border-gray-900 hidden"></div>
-  <div class="my-4 animate-pulse hidden h-1 bg-gray-700"></div>
+  <div
+    class="mt-10 sm:mt-14 md:mt-24 pt-4 border border-gray-900 bg-magenta-600 translate-x-5 hidden"
+  ></div>
+  <div class="my-4 animate-pulse hidden h-1 bg-gray-700 translate-x-0"></div>
+  <div class="hidden bg-gray-200"></div>
 </div>
 <div
   id="task-status-table-container"
diff --git a/pdf2md/server/static/output.css b/pdf2md/server/static/output.css
index 1edc5cd7b7..3ca1b64853 100644
--- a/pdf2md/server/static/output.css
+++ b/pdf2md/server/static/output.css
@@ -566,6 +566,10 @@ video {
   border-width: 0;
 }
 
+.pointer-events-none {
+  pointer-events: none;
+}
+
 .relative {
   position: relative;
 }
@@ -604,6 +608,10 @@ video {
   margin-bottom: 1rem;
 }
 
+.ml-3 {
+  margin-left: 0.75rem;
+}
+
 .mt-10 {
   margin-top: 2.5rem;
 }
@@ -636,6 +644,10 @@ video {
   display: flex;
 }
 
+.inline-flex {
+  display: inline-flex;
+}
+
 .table {
   display: table;
 }
@@ -657,6 +669,11 @@ video {
   height: 3rem;
 }
 
+.size-5 {
+  width: 1.25rem;
+  height: 1.25rem;
+}
+
 .h-1 {
   height: 0.25rem;
 }
@@ -665,10 +682,18 @@ video {
   height: 3rem;
 }
 
+.h-6 {
+  height: 1.5rem;
+}
+
 .max-h-\[75vh\] {
   max-height: 75vh;
 }
 
+.w-11 {
+  width: 2.75rem;
+}
+
 .w-12 {
   width: 3rem;
 }
@@ -689,6 +714,24 @@ video {
   max-width: 1536px;
 }
 
+.shrink-0 {
+  flex-shrink: 0;
+}
+
+.translate-x-0 {
+  --tw-translate-x: 0px;
+  transform: translate(var(--tw-translate-x), var(--tw-translate-y)) rotate(var(--tw-rotate)) skewX(var(--tw-skew-x)) skewY(var(--tw-skew-y)) scaleX(var(--tw-scale-x)) scaleY(var(--tw-scale-y));
+}
+
+.translate-x-5 {
+  --tw-translate-x: 1.25rem;
+  transform: translate(var(--tw-translate-x), var(--tw-translate-y)) rotate(var(--tw-rotate)) skewX(var(--tw-skew-x)) skewY(var(--tw-skew-y)) scaleX(var(--tw-scale-x)) scaleY(var(--tw-scale-y));
+}
+
+.transform {
+  transform: translate(var(--tw-translate-x), var(--tw-translate-y)) rotate(var(--tw-rotate)) skewX(var(--tw-skew-x)) skewY(var(--tw-skew-y)) scaleX(var(--tw-scale-x)) scaleY(var(--tw-scale-y));
+}
+
 @keyframes pulse {
   50% {
     opacity: .5;
@@ -723,10 +766,19 @@ video {
   justify-content: space-between;
 }
 
+.gap-2 {
+  gap: 0.5rem;
+}
+
 .gap-4 {
   gap: 1rem;
 }
 
+.gap-x-1\.5 {
+  -moz-column-gap: 0.375rem;
+       column-gap: 0.375rem;
+}
+
 .gap-x-2 {
   -moz-column-gap: 0.5rem;
        column-gap: 0.5rem;
@@ -781,6 +833,10 @@ video {
   text-wrap: pretty;
 }
 
+.rounded-full {
+  border-radius: 9999px;
+}
+
 .rounded-lg {
   border-radius: 0.5rem;
 }
@@ -797,6 +853,10 @@ video {
   border-width: 0px;
 }
 
+.border-2 {
+  border-width: 2px;
+}
+
 .border-dashed {
   border-style: dashed;
 }
@@ -810,11 +870,25 @@ video {
   border-color: rgb(17 24 39 / 0.25);
 }
 
+.border-transparent {
+  border-color: transparent;
+}
+
+.bg-gray-200 {
+  --tw-bg-opacity: 1;
+  background-color: rgb(229 231 235 / var(--tw-bg-opacity));
+}
+
 .bg-gray-700 {
   --tw-bg-opacity: 1;
   background-color: rgb(55 65 81 / var(--tw-bg-opacity));
 }
 
+.bg-magenta-600 {
+  --tw-bg-opacity: 1;
+  background-color: rgb(125 48 139 / var(--tw-bg-opacity));
+}
+
 .bg-white {
   --tw-bg-opacity: 1;
   background-color: rgb(255 255 255 / var(--tw-bg-opacity));
@@ -832,6 +906,11 @@ video {
   padding: 2rem;
 }
 
+.px-2\.5 {
+  padding-left: 0.625rem;
+  padding-right: 0.625rem;
+}
+
 .px-3 {
   padding-left: 0.75rem;
   padding-right: 0.75rem;
@@ -852,6 +931,11 @@ video {
   padding-right: 2rem;
 }
 
+.py-1 {
+  padding-top: 0.25rem;
+  padding-bottom: 0.25rem;
+}
+
 .py-1\.5 {
   padding-top: 0.375rem;
   padding-bottom: 0.375rem;
@@ -998,12 +1082,29 @@ video {
   color: rgb(125 48 139 / var(--tw-text-opacity));
 }
 
+.text-white {
+  --tw-text-opacity: 1;
+  color: rgb(255 255 255 / var(--tw-text-opacity));
+}
+
+.shadow {
+  --tw-shadow: 0 1px 3px 0 rgb(0 0 0 / 0.1), 0 1px 2px -1px rgb(0 0 0 / 0.1);
+  --tw-shadow-colored: 0 1px 3px 0 var(--tw-shadow-color), 0 1px 2px -1px var(--tw-shadow-color);
+  box-shadow: var(--tw-ring-offset-shadow, 0 0 #0000), var(--tw-ring-shadow, 0 0 #0000), var(--tw-shadow);
+}
+
 .shadow-sm {
   --tw-shadow: 0 1px 2px 0 rgb(0 0 0 / 0.05);
   --tw-shadow-colored: 0 1px 2px 0 var(--tw-shadow-color);
   box-shadow: var(--tw-ring-offset-shadow, 0 0 #0000), var(--tw-ring-shadow, 0 0 #0000), var(--tw-shadow);
 }
 
+.ring-0 {
+  --tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color);
+  --tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(0px + var(--tw-ring-offset-width)) var(--tw-ring-color);
+  box-shadow: var(--tw-ring-offset-shadow), var(--tw-ring-shadow), var(--tw-shadow, 0 0 #0000);
+}
+
 .ring-1 {
   --tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color);
   --tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(1px + var(--tw-ring-offset-width)) var(--tw-ring-color);
@@ -1019,6 +1120,28 @@ video {
   --tw-ring-color: rgb(209 213 219 / var(--tw-ring-opacity));
 }
 
+.transition {
+  transition-property: color, background-color, border-color, text-decoration-color, fill, stroke, opacity, box-shadow, transform, filter, -webkit-backdrop-filter;
+  transition-property: color, background-color, border-color, text-decoration-color, fill, stroke, opacity, box-shadow, transform, filter, backdrop-filter;
+  transition-property: color, background-color, border-color, text-decoration-color, fill, stroke, opacity, box-shadow, transform, filter, backdrop-filter, -webkit-backdrop-filter;
+  transition-timing-function: cubic-bezier(0.4, 0, 0.2, 1);
+  transition-duration: 150ms;
+}
+
+.transition-colors {
+  transition-property: color, background-color, border-color, text-decoration-color, fill, stroke;
+  transition-timing-function: cubic-bezier(0.4, 0, 0.2, 1);
+  transition-duration: 150ms;
+}
+
+.duration-200 {
+  transition-duration: 200ms;
+}
+
+.ease-in-out {
+  transition-timing-function: cubic-bezier(0.4, 0, 0.2, 1);
+}
+
 .placeholder\:text-gray-400::-moz-placeholder {
   --tw-text-opacity: 1;
   color: rgb(156 163 175 / var(--tw-text-opacity));
@@ -1049,6 +1172,11 @@ video {
   --tw-ring-offset-width: 2px;
 }
 
+.hover\:bg-magenta-500:hover {
+  --tw-bg-opacity: 1;
+  background-color: rgb(163 62 181 / var(--tw-bg-opacity));
+}
+
 .hover\:text-magenta-500:hover {
   --tw-text-opacity: 1;
   color: rgb(163 62 181 / var(--tw-text-opacity));
@@ -1059,6 +1187,11 @@ video {
   color: rgb(12 5 14 / var(--tw-text-opacity));
 }
 
+.focus\:outline-none:focus {
+  outline: 2px solid transparent;
+  outline-offset: 2px;
+}
+
 .focus\:ring-2:focus {
   --tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color);
   --tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(2px + var(--tw-ring-offset-width)) var(--tw-ring-color);
@@ -1074,6 +1207,26 @@ video {
   --tw-ring-color: rgb(125 48 139 / var(--tw-ring-opacity));
 }
 
+.focus\:ring-offset-2:focus {
+  --tw-ring-offset-width: 2px;
+}
+
+.focus-visible\:outline:focus-visible {
+  outline-style: solid;
+}
+
+.focus-visible\:outline-2:focus-visible {
+  outline-width: 2px;
+}
+
+.focus-visible\:outline-offset-2:focus-visible {
+  outline-offset: 2px;
+}
+
+.focus-visible\:outline-magenta-600:focus-visible {
+  outline-color: #7D308B;
+}
+
 @media (min-width: 640px) {
   .sm\:-mx-6 {
     margin-left: -1.5rem;
diff --git a/pdf2md/server/static/pdf2md.js b/pdf2md/server/static/pdf2md.js
index d08d8d0fef..c5705417ed 100644
--- a/pdf2md/server/static/pdf2md.js
+++ b/pdf2md/server/static/pdf2md.js
@@ -59,6 +59,7 @@ const upsertTaskToStorage = (task) => {
 
 const displayTask = (task) => {
   const markdownContainer = document.getElementById("markdown-container");
+  const jsonContainer = document.getElementById("json-container");
   const taskId = markdownContainer.getAttribute("data-task-id");
   const taskStatus = markdownContainer.getAttribute("data-task-status");
   const taskNumPagesProcessed = markdownContainer.getAttribute(
@@ -79,6 +80,8 @@ const displayTask = (task) => {
   }
   const sortedPages = pages.sort((a, b) => a.page_num - b.page_num);
 
+  jsonContainer.innerText = JSON.stringify(task, null, 2);
+
   PDFObject.embed(task.file_url, "#my-pdf", {
     pdfOpenParams: {
       view: "FitH",
@@ -206,6 +209,50 @@ advancedOptionsButton.addEventListener("click", (e) => {
     .classList.toggle("hidden");
 });
 
+const jsonSwitch = document.getElementById("json-switch");
+const jsonPre = document.getElementById("json-pre");
+const markdownContainer = document.getElementById("markdown-container");
+jsonSwitch.addEventListener("click", (e) => {
+  e.preventDefault();
+  e.stopPropagation();
+  if (jsonSwitch.classList.contains("bg-magenta-600")) {
+    jsonSwitch.classList.remove("bg-magenta-600");
+    jsonSwitch.classList.add("bg-gray-200");
+    jsonSwitch.querySelector("span").classList.remove("translate-x-5");
+    jsonSwitch.querySelector("span").classList.add("translate-x-0");
+    jsonPre.classList.add("hidden");
+    markdownContainer.classList.remove("hidden");
+  } else {
+    jsonSwitch.classList.add("bg-magenta-600");
+    jsonSwitch.classList.remove("bg-gray-200");
+    jsonSwitch.querySelector("span").classList.add("translate-x-5");
+    jsonSwitch.querySelector("span").classList.remove("translate-x-0");
+    jsonPre.classList.remove("hidden");
+    markdownContainer.classList.add("hidden");
+  }
+});
+
+const copyButton = document.getElementById("copy-button");
+copyButton.addEventListener("click", (e) => {
+  e.preventDefault();
+  e.stopPropagation();
+  let textToCopy = "";
+  if (jsonSwitch.classList.contains("bg-magenta-600")) {
+    textToCopy = jsonPre.innerText;
+  } else {
+    textToCopy = markdownContainer.innerText;
+  }
+  navigator.clipboard.writeText(textToCopy);
+
+  notyf.success({
+    message: `Copied ${
+      jsonSwitch.classList.contains("bg-magenta-600") ? "JSON" : "Markdown"
+    } to clipboard!`,
+    dismissable: true,
+    position: { x: "center", y: "top" },
+  });
+});
+
 const fileUploadInput = document.getElementById("file-upload");
 
 fileUploadInput.addEventListener("change", (event) => {