chatgpt-pro-research 0.2.1 → 0.2.3

package/lib/client.js

@@ -34,8 +34,135 @@ var import_react = require("react");
 var RESEARCH_ICON = '<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="-40 -76 200 200"><title>Anyvia R</title><defs><mask id="R-mono-black-standard-gap" maskUnits="userSpaceOnUse" x="-100" y="-100" width="400" height="400"><rect x="-100" y="-100" width="400" height="400" fill="white"/><path d="M77.18032292691355 52.74975779269961 A3 3 0 0 1 79.72446721538282 48.16 L108.02474889607315 48.16 A3 3 0 0 1 110.56889318454242 49.57024220730038 L139.21290427606598 95.41024220730039 A3 3 0 0 1 136.6687599875967 100 L108.36847830690638 100 A3 3 0 0 1 105.82433401843711 98.58975779269961 Z" transform="translate(-30 -12)" fill="black" stroke="black" stroke-width="6" stroke-linejoin="round"/></mask></defs><g mask="url(#R-mono-black-standard-gap)"><path d="M3.46 -40 H67 A39 39 0 0 1 67 38 H36.81354211962983 V84.54 Q36.81354211962983 88 33.35354211962983 88 H3.46 Q0 88 0 84.54 V-36.54 Q0 -40 3.46 -40 Z M36.81354211962983 -14 V12 H67 A13 13 0 0 0 67 -14 Z" fill="currentColor" fill-rule="evenodd"/></g><path d="M77.18032292691355 52.74975779269961 A3 3 0 0 1 79.72446721538282 48.16 L108.02474889607315 48.16 A3 3 0 0 1 110.56889318454242 49.57024220730038 L139.21290427606598 95.41024220730039 A3 3 0 0 1 136.6687599875967 100 L108.36847830690638 100 A3 3 0 0 1 105.82433401843711 98.58975779269961 Z" transform="translate(-30 -12)" fill="currentColor"/></svg>';
 var DESIGN_ICON = '<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="-22.599456503942804 -56 200 200"><title>Anyvia X</title><defs><mask id="X-mono-black-standard-gap" maskUnits="userSpaceOnUse" x="-100" y="-100" width="400" height="400"><rect x="-100" y="-100" width="400" height="400" fill="white"/><path d="M77.18032292691355 52.74975779269961 A3 3 0 0 1 79.72446721538282 48.16 L108.02474889607315 48.16 A3 3 0 0 1 110.56889318454242 49.57024220730038 L139.21290427606598 95.41024220730039 A3 3 0 0 1 136.6687599875967 100 L108.36847830690638 100 A3 3 0 0 1 105.82433401843711 98.58975779269961 Z" transform="translate(-12 0)" fill="black" stroke="black" stroke-width="6" stroke-linejoin="round"/></mask></defs><g mask="url(#X-mono-black-standard-gap)"><path d="M89.67202319894069 -10.373520654246889 A3.46 3.46 0 0 1 92.60626961164192 -12 L127.85608887910462 -12 A3.46 3.46 0 0 1 130.79033529180586 -6.706479345753113 L65.12906379317373 98.37352065424689 A3.46 3.46 0 0 1 62.19481738047249 100 L26.94499811300981 100 A3.46 3.46 0 0 1 24.010751700308575 94.70647934575311 Z" fill="currentColor"/><path d="M28.027942559241307 -6.706479345753113 A3.46 3.46 0 0 1 30.96218897194254 -12 L58.17762652153973 -12 A3.46 3.46 0 0 1 61.111872934240964 -10.373520654246889 L91.78046072595076 38.70647934575311 A3.46 3.46 0 0 1 88.84621431324952 44 L61.63077676365233 44 A3.46 3.46 0 0 1 58.69653035095109 42.37352065424689 Z" fill="currentColor"/></g><path d="M77.18032292691355 52.74975779269961 A3 3 0 0 1 79.72446721538282 48.16 L108.02474889607315 48.16 A3 3 0 0 1 110.56889318454242 49.57024220730038 L139.21290427606598 95.41024220730039 A3 3 0 0 1 136.6687599875967 100 L108.36847830690638 100 A3 3 0 0 1 105.82433401843711 98.58975779269961 Z" transform="translate(-12 0)" fill="currentColor"/></svg>';
 
+// client/sidebar-status.ts
+var POLL_MS = 5e3;
+function countTopics(topics) {
+  const counts = { active: 0, completed: 0, failed: 0 };
+  for (const topic of topics) {
+    switch (topic.status) {
+      case "queued":
+      case "uploading":
+      case "submitted":
+      case "waiting":
+      case "answered":
+      case "adjudicating":
+        counts.active += 1;
+        break;
+      case "completed":
+        counts.completed += 1;
+        break;
+      case "failed":
+      case "stalled":
+        counts.failed += 1;
+        break;
+      default:
+        break;
+    }
+  }
+  return counts;
+}
+function badgeText(counts) {
+  const parts = [];
+  if (counts.active > 0) parts.push(`${counts.active} \u8FDB\u884C`);
+  if (counts.completed > 0) parts.push(`${counts.completed} \u5B8C\u6210`);
+  if (counts.failed > 0) parts.push(`${counts.failed} \u5931\u8D25`);
+  return parts.join(" \xB7 ");
+}
+function mountSidebarStatus(options) {
+  if (typeof document === "undefined") return () => {
+  };
+  let badge;
+  let text = "";
+  let disposed = false;
+  let polling = false;
+  let attachTimer;
+  let observer;
+  const rowFor = () => {
+    const buttons = document.querySelectorAll("button[aria-label]");
+    for (const button of buttons) {
+      if (button.getAttribute("aria-label") === options.label) return button;
+    }
+    return void 0;
+  };
+  const applyText = (next) => {
+    text = next;
+    if (badge !== void 0 && badge.textContent !== text) badge.textContent = text;
+    ensureMounted();
+  };
+  const ensureMounted = () => {
+    if (disposed) return;
+    if (text === "") {
+      badge?.remove();
+      badge = void 0;
+      return;
+    }
+    const row = rowFor();
+    if (row === void 0) return;
+    const title = row.querySelector('[class*="panelTitle"]');
+    if (title === null) {
+      badge?.remove();
+      badge = void 0;
+      return;
+    }
+    if (badge !== void 0 && badge.parentElement === row) {
+      if (badge.textContent !== text) badge.textContent = text;
+      return;
+    }
+    badge = document.createElement("span");
+    badge.dataset.dshProBadge = options.panelId;
+    badge.textContent = text;
+    badge.setAttribute("aria-hidden", "true");
+    badge.style.cssText = [
+      "margin-left:auto",
+      "display:inline-flex",
+      "align-items:center",
+      "flex:0 0 auto",
+      "font-size:10px",
+      "line-height:1",
+      "opacity:.62",
+      "white-space:nowrap",
+      "font-variant-numeric:tabular-nums",
+      "pointer-events:none"
+    ].join(";");
+    row.appendChild(badge);
+  };
+  const refresh = async () => {
+    if (disposed || polling) return;
+    polling = true;
+    try {
+      const response = await fetch(`${options.routes}/topics`, { headers: { accept: "application/json" } });
+      if (!response.ok) return;
+      const data = await response.json();
+      applyText(badgeText(countTopics(data.topics ?? [])));
+    } catch {
+    } finally {
+      polling = false;
+    }
+  };
+  observer = new MutationObserver(() => {
+    if (attachTimer !== void 0) return;
+    attachTimer = window.setTimeout(() => {
+      attachTimer = void 0;
+      ensureMounted();
+    }, 100);
+  });
+  observer.observe(document.body, { childList: true, subtree: true });
+  const timer = window.setInterval(() => {
+    void refresh();
+  }, POLL_MS);
+  void refresh();
+  return () => {
+    disposed = true;
+    window.clearInterval(timer);
+    if (attachTimer !== void 0) window.clearTimeout(attachTimer);
+    observer?.disconnect();
+    badge?.remove();
+    badge = void 0;
+  };
+}
+
 // client/panel.tsx
-var POLL_MS = 3e3;
+var POLL_MS2 = 3e3;
 var config = {
   routes: "/chatgpt-pro-research",
   panelId: "chatgpt-pro-research",
@@ -126,7 +253,7 @@ function Panel() {
     void loadList();
     const timer = setInterval(() => {
       void loadList();
-    }, POLL_MS);
+    }, POLL_MS2);
     return () => clearInterval(timer);
   }, [loadList]);
   (0, import_react.useEffect)(() => {
@@ -134,7 +261,7 @@ function Panel() {
     void loadDetail(selected);
     const timer = setInterval(() => {
       void loadDetail(selected);
-    }, POLL_MS);
+    }, POLL_MS2);
     return () => clearInterval(timer);
   }, [selected, loadDetail, topics]);
   const act = (0, import_react.useCallback)(async (action, conclusion) => {
@@ -315,6 +442,8 @@ function Icon({ kind }) {
 function apply(ctx) {
   ctx.slots.inject("main", () => ctx.slots.register({ name: "main", key: config.panelId }, () => (0, import_react.createElement)(Panel)));
   ctx.slots.inject("sidebar.panellist", () => ctx.slots.register({ name: "sidebar.panellist", id: config.panelId, order: 21, label: () => config.label }, () => (0, import_react.createElement)(Icon, { kind: config.icon })));
+  const disposeBadge = mountSidebarStatus({ label: config.label, routes: config.routes, panelId: config.panelId });
+  if (typeof ctx.effect === "function") ctx.effect(() => disposeBadge, `${config.panelId}: sidebar status badge`);
 }
 
 // client/index.tsx

package/lib/index.js

@@ -733,7 +733,9 @@ var ResearchEngine = class {
         try {
           return await this.options.transport.fetch({ submissionId });
         } catch (error) {
-          if (polled.answer !== void 0) return { answer: polled.answer };
+          if (polled.answer !== void 0) {
+            return { answer: polled.answer.startsWith("image:") ? "" : polled.answer };
+          }
           throw error;
         }
       }
@@ -932,6 +934,16 @@ function advanceCompletion(previous, text, streaming) {
   return { lastText: text, stable: 1, answered: false };
 }
 
+// ../transport/src/assets.ts
+function parseDataUrlImage(source) {
+  const match = /^data:([^;,]+);base64,(.*)$/s.exec(source);
+  if (match === null) return void 0;
+  const mime = match[1];
+  const base64 = match[2];
+  if (mime === void 0 || base64 === void 0) return void 0;
+  return { mime, base64: base64.replace(/\s/g, "") };
+}
+
 // ../transport/src/chatgpt.ts
 function sha256(value) {
   return createHash3("sha256").update(value, "utf8").digest("hex");
@@ -1140,6 +1152,17 @@ ${attachments.join("\n")}`);
     const images = [];
     for (const [index, source] of sources.slice(0, 8).entries()) {
       try {
+        if (source.startsWith("data:")) {
+          const parsed = parseDataUrlImage(source);
+          if (parsed === void 0) continue;
+          if (Buffer.from(parsed.base64, "base64").byteLength < 2e4) continue;
+          images.push({
+            name: `image-${String(index + 1).padStart(2, "0")}.${extensionFor(parsed.mime)}`,
+            mime: parsed.mime,
+            base64: parsed.base64
+          });
+          continue;
+        }
         if (source.startsWith("blob:")) {
           const dataUrl = await page.evaluate(async (src) => {
             const response = await fetch(src);
@@ -1237,8 +1260,8 @@ ${attachments.join("\n")}`);
   async richAnswerText(page) {
     const text = await this.lastAssistantText(page);
     if (text !== "") return text;
-    const images = await page.evaluate(() => [...document.querySelectorAll("main img")].filter((image) => image.naturalWidth >= 300).map((image) => image.src));
-    return images.length > 0 ? `image:${images.join("|")}` : "";
+    const images = await page.evaluate(() => [...document.querySelectorAll("main img")].filter((image) => image.naturalWidth >= 300).map((image) => `${image.naturalWidth}x${image.naturalHeight}`));
+    return images.length > 0 ? `image:${images.length}:${images.join(",")}` : "";
   }
   async lastAssistantText(page) {
     const messages = page.locator(SELECTORS.assistantMessage);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chatgpt-pro-research",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "ChatGPT Pro Research: unattended multi-round research through a real ChatGPT Pro web session, with local records and a DSH sidebar.",
   "private": false,
   "type": "module",