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

dialoghelper/__init__.py

@@ -1,2 +1,2 @@
-__version__ = "0.0.20"
+__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/core.py

@@ -69,9 +69,9 @@ def find_msg_id():
 
 # %% ../nbs/00_core.ipynb
 def msg_idx(
-    msgid=None # Message id to find (defaults to current message)
+    msgid=None,  # Message id to find (defaults to current message)
 ):
-    "Get relative index of message in dialog."
+    "Get absolute index of message in dialog."
     if not msgid: msgid = find_msg_id()
     return call_endp('msg_idx_', json=True, msgid=msgid)['msgid']
 
@@ -141,6 +141,7 @@ def _add_msg_unsafe(
 
 # %% ../nbs/00_core.ipynb
 def _umsg(
+    content:str|None=None, # Content of the message (i.e the message prompt, code, or note text)
     msg_type: str|None = None, # Message type, can be 'code', 'note', or 'prompt'
     output:str|None = None, # For prompts/code, the output
     time_run: str | None = None, # When was message executed
@@ -156,14 +157,13 @@ def _umsg(
 @delegates(_umsg)
 def update_msg(
     msgid:str=None, # id of message to update (if None, uses current message)
-    content:str|None=None, # Content of the message (i.e the message prompt, code, or note text)
     msg:Optional[Dict]=None, # Dictionary of field keys/values to update
     **kwargs):
     """Update an existing message. Provide either `msg` OR field key/values to update.
     Use `content` param to update contents.
     Only include parameters to update--missing ones will be left unchanged."""
     if not msgid and not msg: raise TypeError("update_msg needs either a dict message or `msgid=`")
-    return call_endp('add_relative_', content=content, placement='update', msgid=msgid, **kwargs)
+    return call_endp('add_relative_', placement='update', msgid=msgid, **kwargs)
 
 # %% ../nbs/00_core.ipynb
 def load_gist(gist_id:str):
@@ -236,6 +236,7 @@ def tool_info():
     cts='''Tools available from `dialoghelper`:
 
 - &`curr_dialog`: Get the current dialog info.
+- &`msg_idx`: Get absolute index of message in dialog.
 - &`add_html`: Send HTML to the browser to be swapped into the DOM using hx-swap-oob.
 - &`find_msg_id`: Get the current message id.
 - &`find_msgs`: Find messages in current specific dialog that contain the given information.

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.20
+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.20.dist-info/RECORD

@@ -1,10 +0,0 @@
-dialoghelper/__init__.py,sha256=mkmmPV_a7e3EgojncsapZLZfAn0jyTfnZIQiO1GWA0M,43
-dialoghelper/_modidx.py,sha256=wWjLy97PBuZVMAydnBUSJlkZuN9x2bkYyY0SHwZQkrk,2654
-dialoghelper/core.py,sha256=AlhLbCO4flkphUItM6bHT7jcdsgVZ0a7Yx3zmwhnkBA,10819
-dialoghelper/db_dc.py,sha256=mi2Q2am_SoAPGpNQg7KPFS5pR9WEapRXT8ypkNmMfw0,2330
-dialoghelper-0.0.20.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dialoghelper-0.0.20.dist-info/METADATA,sha256=LRTH4iHmFc3OvCFpl1SpxvEgq9dRS2ZcQ_C_Ny18LsQ,2565
-dialoghelper-0.0.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dialoghelper-0.0.20.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
-dialoghelper-0.0.20.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
-dialoghelper-0.0.20.dist-info/RECORD,,