dialoghelper 0.0.27__tar.gz → 0.0.28__tar.gz

{dialoghelper-0.0.27 → dialoghelper-0.0.28}/MANIFEST.in

@@ -2,4 +2,5 @@ include settings.ini
 include LICENSE
 include CONTRIBUTING.md
 include README.md
+include dialoghelper/*.js
 recursive-exclude * __pycache__

{dialoghelper-0.0.27/dialoghelper.egg-info → dialoghelper-0.0.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.27
+Version: 0.0.28
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -24,6 +24,7 @@ Requires-Dist: ghapi
 Requires-Dist: ipykernel-helper
 Requires-Dist: claudette
 Provides-Extra: dev
+Requires-Dist: python-fasthtml; extra == "dev"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

dialoghelper-0.0.28/dialoghelper/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "0.0.28"
+from .core import *

dialoghelper-0.0.28/dialoghelper/screenshot.js

@@ -0,0 +1,129 @@
+/**
+ * Screenshot capture functionality for dialoghelper
+ * Provides persistent screen sharing with HTTP polling for screenshots
+ */
+
+let persistentStream = null;
+let streamStatus = "disconnected"; // 'disconnected', 'connecting', 'connected', 'error'
+
+function sendDataToServer(dataId, data) {
+    return fetch('/push_data_blocking_', {
+        method: 'POST',
+        headers: {'Content-Type': 'application/json'},
+        body: JSON.stringify({data_id: dataId, ...data})
+    });
+}
+
+async function streamToBlob(stream, maxWidth = 1280, maxHeight = 1024) {
+  return new Promise((resolve, reject) => {
+    const video = document.createElement("video");
+    video.srcObject = stream;
+    video.muted = true;
+    video.playsInline = true;
+    video.addEventListener("loadedmetadata", () => {
+      // Downscale to maxWidth x maxHeight using canvas
+      const videoWidth = video.videoWidth;
+      const videoHeight = video.videoHeight;
+      const scaleX = maxWidth / videoWidth;
+      const scaleY = maxHeight / videoHeight;
+      const scale = Math.min(scaleX, scaleY, 1); // don't upscale
+      const newWidth = Math.floor(videoWidth * scale);
+      const newHeight = Math.floor(videoHeight * scale);
+      const canvas = document.createElement("canvas");
+      canvas.width = newWidth;
+      canvas.height = newHeight;
+      const ctx = canvas.getContext("2d");
+      ctx.drawImage(video, 0, 0, newWidth, newHeight);
+      canvas.toBlob(resolve, "image/png");
+    });
+    video.addEventListener("error", reject);
+    video.play().catch(reject);
+  });
+}
+
+async function waitForGetDisplayMedia(timeout = 30000) {
+  const start = Date.now();
+  while (Date.now() - start < timeout) {
+    if (navigator.mediaDevices && navigator.mediaDevices.getDisplayMedia) { return true; }
+    await new Promise((resolve) => setTimeout(resolve, 100));
+  }
+  throw new Error("getDisplayMedia not available after timeout");
+}
+
+async function startPersistentScreenShare(statusId = null) {
+  try {
+    await waitForGetDisplayMedia();
+    persistentStream = await navigator.mediaDevices.getDisplayMedia({
+      video: { mediaSource: "screen", displaySurface: "monitor" }, audio: false,
+    });
+    persistentStream.getVideoTracks()[0].addEventListener("ended", () => {
+      console.log("Screen share ended by user");
+      stopPersistentScreenShare();
+    });
+    console.log("✅ Persistent screen share started");
+    if (statusId) { sendDataToServer(statusId, { js_status: "ready" }); }
+		streamStatus = "connected";
+    return { status: "success", message: "Screen share started" };
+  } catch (error) {
+    console.error("Failed to start persistent screen share:", error);
+    if (statusId) { sendDataToServer(statusId, { js_status: "error", error: error.message }); }
+    return { status: "error", message: error.message };
+  }
+}
+
+function stopPersistentScreenShare() {
+  if (persistentStream) {
+    persistentStream.getTracks().forEach((track) => track.stop());
+    persistentStream = null;
+  }
+  streamStatus = "disconnected";
+  console.log("🛑 Persistent screen share stopped");
+  return { status: "success", message: "Screen share stopped" };
+}
+
+function blobToBase64(blob) {
+  return new Promise((resolve, reject) => {
+    const reader = new FileReader();
+    reader.onload = () => resolve(reader.result);
+    reader.onerror = reject;
+    reader.readAsDataURL(blob);
+  });
+}
+
+async function processScreenshotBlob(blob, dataId) {
+  const base64String = await blobToBase64(blob);
+  const blob_type = blob.type || "image/png";
+  const b64data = base64String.split(",")[1]; // Remove the data URL prefix
+  return { 
+		data_id: dataId,
+		size: blob.size,
+		img_type: blob_type,
+		img_data: b64data,
+  };
+}
+
+async function captureScreenFromStream(dataId) {
+  console.log("Executing screenshot from persistent stream");
+  try {
+    if (!persistentStream || streamStatus !== "connected") {
+			console.log("Stream status:", streamStatus);
+			console.log("Persistent stream:", persistentStream);
+      throw new Error("No active screen share. Call startPersistentScreenShare() first.");
+    }
+    const blob = await streamToBlob(persistentStream);
+    const result = await processScreenshotBlob(blob, dataId);
+    console.log("Screenshot result:", result);
+    const pushResponse = await sendDataToServer(dataId, result)
+    if (pushResponse.ok) { console.log("✅ Screenshot data pushed to server"); }
+		else { console.log("❌ Failed to push screenshot data"); }
+  } catch (error) {
+    console.error("Screenshot error:", error);
+		sendDataToServer(dataId, { error: error.message });
+  }
+  console.log("Finished executing screenshot");
+}
+
+window.startPersistentScreenShare = startPersistentScreenShare;
+window.stopPersistentScreenShare = stopPersistentScreenShare;
+window.captureScreenFromStream = captureScreenFromStream;
+window.getStreamStatus = () => streamStatus;

{dialoghelper-0.0.27 → dialoghelper-0.0.28/dialoghelper.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.27
+Version: 0.0.28
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -24,6 +24,7 @@ Requires-Dist: ghapi
 Requires-Dist: ipykernel-helper
 Requires-Dist: claudette
 Provides-Extra: dev
+Requires-Dist: python-fasthtml; extra == "dev"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{dialoghelper-0.0.27 → dialoghelper-0.0.28}/dialoghelper.egg-info/SOURCES.txt

@@ -9,6 +9,7 @@ dialoghelper/_modidx.py
 dialoghelper/core.py
 dialoghelper/db_dc.py
 dialoghelper/experimental.py
+dialoghelper/screenshot.js
 dialoghelper.egg-info/PKG-INFO
 dialoghelper.egg-info/SOURCES.txt
 dialoghelper.egg-info/dependency_links.txt

{dialoghelper-0.0.27 → dialoghelper-0.0.28}/dialoghelper.egg-info/requires.txt

@@ -5,3 +5,4 @@ ipykernel-helper
 claudette
 
 [dev]
+python-fasthtml

{dialoghelper-0.0.27 → dialoghelper-0.0.28}/settings.ini

@@ -1,11 +1,12 @@
 [DEFAULT]
 repo = dialoghelper
 lib_name = dialoghelper
-version = 0.0.27
+version = 0.0.28
 min_python = 3.9
 license = apache2
 black_formatting = False
 requirements = fastcore>=1.8.5 fastlite ghapi ipykernel-helper claudette
+dev_requirements = python-fasthtml
 doc_path = _docs
 lib_path = dialoghelper
 nbs_path = nbs

dialoghelper-0.0.27/dialoghelper/__init__.py

@@ -1,2 +0,0 @@
-__version__ = "0.0.27"
-from .core import *