dialoghelper 0.0.21__py3-none-any.whl → 0.0.22__py3-none-any.whl

dialoghelper/__init__.py

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

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/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.dist-info → dialoghelper-0.0.22.dist-info}/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,11 @@
+dialoghelper/__init__.py,sha256=C9YgbQJOYL1v4JJrxFdU3NX8aFG4GorkZ6mJFay3Sww,43
+dialoghelper/_modidx.py,sha256=dqZjsj-FkFgiTcIeiCB7rSCpUzStGILPMqC0NRt4jO4,3402
+dialoghelper/core.py,sha256=3zWfyFZPDkzHVIh2sjI4PESpezXhOKMnWNqOJ1zXXBA,10859
+dialoghelper/db_dc.py,sha256=mi2Q2am_SoAPGpNQg7KPFS5pR9WEapRXT8ypkNmMfw0,2330
+dialoghelper/experimental.py,sha256=6tDnbY3JBA8MLdzMZ18uCHnT5BapgzH61sGxfUt8HQ8,2459
+dialoghelper-0.0.22.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dialoghelper-0.0.22.dist-info/METADATA,sha256=bbZHjmrMW--7ceB6nDC5Or7DTnRF6CNWlAqSpgoQpDw,2590
+dialoghelper-0.0.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dialoghelper-0.0.22.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
+dialoghelper-0.0.22.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
+dialoghelper-0.0.22.dist-info/RECORD,,

dialoghelper-0.0.21.dist-info/RECORD

@@ -1,10 +0,0 @@
-dialoghelper/__init__.py,sha256=731zsPSn_ikal3tSw5-jTdqXWpxeTh4lO9otJwO7wOE,43
-dialoghelper/_modidx.py,sha256=wWjLy97PBuZVMAydnBUSJlkZuN9x2bkYyY0SHwZQkrk,2654
-dialoghelper/core.py,sha256=3zWfyFZPDkzHVIh2sjI4PESpezXhOKMnWNqOJ1zXXBA,10859
-dialoghelper/db_dc.py,sha256=mi2Q2am_SoAPGpNQg7KPFS5pR9WEapRXT8ypkNmMfw0,2330
-dialoghelper-0.0.21.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dialoghelper-0.0.21.dist-info/METADATA,sha256=RcShJ1NQgVYRjy8ZVC45cKsyiidtcWPBn8Ymp1KilMA,2565
-dialoghelper-0.0.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dialoghelper-0.0.21.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
-dialoghelper-0.0.21.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
-dialoghelper-0.0.21.dist-info/RECORD,,