dialoghelper 0.0.21__tar.gz → 0.0.22__tar.gz

{dialoghelper-0.0.21/dialoghelper.egg-info → dialoghelper-0.0.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.21
+Version: 0.0.22
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -22,6 +22,7 @@ Requires-Dist: fastcore>=1.8.5
 Requires-Dist: fastlite
 Requires-Dist: ghapi
 Requires-Dist: ipykernel-helper
+Requires-Dist: claudette
 Provides-Extra: dev
 Dynamic: author
 Dynamic: author-email

dialoghelper-0.0.22/dialoghelper/__init__.py

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

{dialoghelper-0.0.21 → dialoghelper-0.0.22}/dialoghelper/_modidx.py

@@ -25,4 +25,10 @@ d = { 'settings': { 'branch': 'main',
                                    'dialoghelper.core.read_msg': ('core.html#read_msg', 'dialoghelper/core.py'),
                                    'dialoghelper.core.tool_info': ('core.html#tool_info', 'dialoghelper/core.py'),
                                    'dialoghelper.core.update_msg': ('core.html#update_msg', 'dialoghelper/core.py')},
-            'dialoghelper.db_dc': {}}}
+            'dialoghelper.db_dc': {},
+            'dialoghelper.experimental': { 'dialoghelper.experimental._load_screenshot_js': ( 'experimental.html#_load_screenshot_js',
+                                                                                              'dialoghelper/experimental.py'),
+                                           'dialoghelper.experimental.capture_screen': ( 'experimental.html#capture_screen',
+                                                                                         'dialoghelper/experimental.py'),
+                                           'dialoghelper.experimental.create_iife': ( 'experimental.html#create_iife',
+                                                                                      'dialoghelper/experimental.py')}}}

dialoghelper-0.0.22/dialoghelper/experimental.py

@@ -0,0 +1,68 @@
+"""Experimental `dialoghelper` capabilities."""
+
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/01_experimental.ipynb.
+
+# %% auto 0
+__all__ = ['create_iife', 'capture_screen']
+
+# %% ../nbs/01_experimental.ipynb
+from importlib import resources
+import uuid
+from fasthtml.common import Div, Script
+from time import sleep
+import base64
+import json
+import time
+from claudette import ToolResult
+from .core import *
+from dataclasses import dataclass
+from httpx import get as xget, post as xpost
+
+# %% ../nbs/01_experimental.ipynb
+_js_loaded = False
+
+def _load_screenshot_js():
+    "Load screenshot capability if not already present."
+    global _js_loaded
+    if _js_loaded:
+        print("screenshot.js already loaded")
+        return
+    js_content = (resources.files('dialoghelper') / 'screenshot.js').read_text()
+    add_html(Div(Script(js_content), hx_swap_oob='beforeend:#js-script'))
+    print("Added screenshot.js")
+    _js_loaded = True
+
+# %% ../nbs/01_experimental.ipynb
+def create_iife(code: str) -> str:
+    "Wrap javascript code string in an IIFE"
+    return f'''
+(async () => {{
+{code}
+}})();
+'''
+
+# %% ../nbs/01_experimental.ipynb
+def capture_screen():
+    "Capture current desktop returning the screenshot as a ToolImageResult."
+    _load_screenshot_js()        
+    data_id = str(uuid.uuid4())  # Generate random ID for data communication
+    screenshot_code = f'captureScreenAndUpload("{data_id}");'
+    trigger_script = create_iife(screenshot_code)
+    add_html(Div(Script(trigger_script), hx_swap_oob=f'beforeend:#js-script'))
+    max_attempts = 50
+    for attempt in range(max_attempts):
+        result = xpost('http://localhost:5001/pop_data_', data={'data_id': data_id})
+        if result.status_code == 200:
+            try:
+                if not result.text.strip():
+                    time.sleep(0.5)
+                    continue
+                data = result.json() # data is a dict
+                if 'error' in data: return f"Screenshot failed: {data['error']}"
+                if 'img_data' in data and 'img_type' in data:
+                    return ToolResult(data=data['img_data'], result_type=data['img_type'])
+                return f"Screenshot data incomplete: {data}"
+            except json.JSONDecodeError as e: print(f"JSON decode error: {e}, Raw text: '{result.text}'")
+            except Exception as e: print(f"Error processing screenshot data: {e}")
+        time.sleep(0.5)
+    return "Screenshot capture timed out"

{dialoghelper-0.0.21 → dialoghelper-0.0.22/dialoghelper.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.21
+Version: 0.0.22
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -22,6 +22,7 @@ Requires-Dist: fastcore>=1.8.5
 Requires-Dist: fastlite
 Requires-Dist: ghapi
 Requires-Dist: ipykernel-helper
+Requires-Dist: claudette
 Provides-Extra: dev
 Dynamic: author
 Dynamic: author-email

{dialoghelper-0.0.21 → dialoghelper-0.0.22}/dialoghelper.egg-info/SOURCES.txt

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

{dialoghelper-0.0.21 → dialoghelper-0.0.22}/dialoghelper.egg-info/requires.txt

@@ -2,5 +2,6 @@ fastcore>=1.8.5
 fastlite
 ghapi
 ipykernel-helper
+claudette
 
 [dev]

{dialoghelper-0.0.21 → dialoghelper-0.0.22}/settings.ini

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

dialoghelper-0.0.21/dialoghelper/__init__.py

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