dialoghelper 0.0.23__py3-none-any.whl → 0.0.24__py3-none-any.whl

dialoghelper/__init__.py

@@ -1,2 +1,2 @@
-__version__ = "0.0.23"
+__version__ = "0.0.24"
 from .core import *

dialoghelper/_modidx.py

@@ -23,12 +23,18 @@ d = { 'settings': { 'branch': 'main',
                                    'dialoghelper.core.mk_toollist': ('core.html#mk_toollist', 'dialoghelper/core.py'),
                                    'dialoghelper.core.msg_idx': ('core.html#msg_idx', 'dialoghelper/core.py'),
                                    'dialoghelper.core.read_msg': ('core.html#read_msg', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.run_msg': ('core.html#run_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.experimental': { 'dialoghelper.experimental._load_screenshot_js': ( 'experimental.html#_load_screenshot_js',
                                                                                               'dialoghelper/experimental.py'),
+                                           'dialoghelper.experimental._wait_for_pop_data': ( 'experimental.html#_wait_for_pop_data',
+                                                                                             '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.iife': ('experimental.html#iife', 'dialoghelper/experimental.py'),
+                                           'dialoghelper.experimental.start_screen_share': ( 'experimental.html#start_screen_share',
+                                                                                             'dialoghelper/experimental.py'),
+                                           'dialoghelper.experimental.stop_screen_share': ( 'experimental.html#stop_screen_share',
+                                                                                            'dialoghelper/experimental.py')}}}

dialoghelper/core.py

@@ -2,8 +2,8 @@
 
 # %% auto 0
 __all__ = ['Placements', 'empty', 'find_var', 'call_endp', 'curr_dialog', 'find_msgs', 'find_msg_id', 'msg_idx', 'read_msg',
-           'add_html', 'del_msg', 'add_msg', 'update_msg', 'load_gist', 'gist_file', 'import_string', 'is_usable_tool',
-           'mk_toollist', 'import_gist', 'tool_info', 'asdict']
+           'add_html', 'del_msg', 'run_msg', 'add_msg', 'update_msg', 'load_gist', 'gist_file', 'import_string',
+           'is_usable_tool', 'mk_toollist', 'import_gist', 'tool_info', 'asdict']
 
 # %% ../nbs/00_core.ipynb
 import json, importlib, linecache
@@ -94,11 +94,18 @@ def add_html(
 
 # %% ../nbs/00_core.ipynb
 def del_msg(
-    msgid:str=None, # id of message that placement is relative to (if None, uses current message)
+    msgid:str=None, # id of message to delete
 ):
-    "Delete a message from the dialog. Be sure to pass a `sid`, not a `mid`."
+    "Delete a message from the dialog."
     call_endp('rm_msg_', raiseex=True, msid=msgid)
 
+# %% ../nbs/00_core.ipynb
+def run_msg(
+    msgid:str=None, # id of message to execute
+):
+    "Adds a message to the run queue. Use read_msg to see the output once it runs."
+    return call_endp('add_runq_', msgid=msgid, api=True)
+
 # %% ../nbs/00_core.ipynb
 Placements = str_enum('Placements', 'add_after', 'add_before', 'at_start', 'at_end')
 
@@ -240,6 +247,7 @@ def tool_info():
 - &`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.
+  - (solveit can often get this id directly from its context, and will not need to use this if the required information is already available to it.)
 - &`read_msg`: Get the message indexed in the current dialog.
 - &`del_msg`: Delete a message from the dialog.
 - &`add_msg`: Add/update a message to the queue to show after code execution completes.

dialoghelper/experimental.py

@@ -3,7 +3,7 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/01_experimental.ipynb.
 
 # %% auto 0
-__all__ = ['create_iife', 'capture_screen']
+__all__ = ['iife', 'start_screen_share', 'stop_screen_share', 'capture_screen']
 
 # %% ../nbs/01_experimental.ipynb
 from importlib import resources
@@ -18,22 +18,46 @@ from .core import *
 from dataclasses import dataclass
 from httpx import get as xget, post as xpost
 
+# %% ../nbs/01_experimental.ipynb
+def _wait_for_pop_data(data_id, timeout=20, retry_interval=1, condition=None):
+    "Wait for data from pop endpoint with optional condition validation."
+    max_attempts = int(timeout / retry_interval)
+    for attempt in range(max_attempts):
+        result = xpost('http://localhost:5001/pop_data_', data={'data_id': data_id})
+        if result.status_code == 200 and result.text.strip():
+            try:
+                data = result.json()
+                if condition and not condition(data):
+                    time.sleep(retry_interval)
+                    continue
+                return data
+            except json.JSONDecodeError: pass
+        time.sleep(retry_interval)
+    return None
+
 # %% ../nbs/01_experimental.ipynb
 _js_loaded = False
 
-def _load_screenshot_js():
-    "Load screenshot capability if not already present."
+def _load_screenshot_js(timeout=10, retry_interval=1):
+    "Load screenshot capability and wait for confirmation it's ready."
     global _js_loaded
-    if _js_loaded:
-        print("screenshot.js already loaded")
-        return
+    if _js_loaded: return True    
     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
+    ready_id = str(uuid.uuid4())
+    js_with_ready = js_content + f'\n\n// Signal ready\nsendDataToServer("{ready_id}", {{"js_status": "ready"}});'
+    add_html(Div(Script(js_with_ready), hx_swap_oob='beforeend:#js-script'))
+    print("Loading screenshot.js ...")    
+    data = _wait_for_pop_data(ready_id, timeout, retry_interval, condition=lambda d: d.get('js_status') == 'ready')
+    if data and data.get('js_status') == 'ready':
+        _js_loaded = True
+        print("Screenshot.js loaded and ready")
+        return True
+    else:
+        print("Failed to load screenshot.js")
+        return False
 
 # %% ../nbs/01_experimental.ipynb
-def create_iife(code: str) -> str:
+def iife(code: str) -> str:
     "Wrap javascript code string in an IIFE"
     return f'''
 (async () => {{
@@ -41,28 +65,51 @@ def create_iife(code: str) -> str:
 }})();
 '''
 
+# %% ../nbs/01_experimental.ipynb
+_screen_share_active = False
+
+def start_screen_share(timeout=30, retry_interval=1):
+    "Start persistent screen sharing session, waiting for confirmation."
+    global _screen_share_active
+    _load_screenshot_js()
+    status_id = str(uuid.uuid4())
+    trigger_script = iife(f'startPersistentScreenShare("{status_id}");')
+    add_html(Div(Script(trigger_script), hx_swap_oob=f'beforeend:#js-script'))
+    print("Requesting screen share permission ...")
+    data = _wait_for_pop_data(status_id, timeout, retry_interval, condition=lambda d: d.get('js_status') == 'ready')
+    if not data: print(f"Screen share timed out after {timeout} seconds.")
+    js_status = data.get('js_status')
+    if js_status == 'ready':
+        _screen_share_active = True
+        print("Screen share started successfully.")
+    elif js_status == 'error': print(f"Screen share failed: {data.get('error', 'Unknown error')}")
+    elif js_status == 'connecting': print("Screen share timed out after {timeout} seconds.")
+
+# %% ../nbs/01_experimental.ipynb
+def stop_screen_share():
+  "Stop persistent screen sharing session."
+  global _screen_share_active
+  _load_screenshot_js()
+  trigger_script = iife('stopPersistentScreenShare();')
+  add_html(Div(Script(trigger_script), hx_swap_oob=f'beforeend:#js-script'))
+  _screen_share_active = False
+  print("Screen share stopped.")
+
 # %% ../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)
+    "Capture screenshot, automatically starting screen share if needed."
+    global _screen_share_active
+    _load_screenshot_js()
+    if not _screen_share_active:
+        print("🔄 No active screen share, starting one...")
+        result = start_screen_share()
+        if not _screen_share_active: return f"Failed to start screen share: {result}"
+    data_id = str(uuid.uuid4())
+    screenshot_code = f'captureScreenFromStream("{data_id}");'
+    print("📸 Capturing from persistent stream...")
+    trigger_script = 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"
+    data = _wait_for_pop_data(data_id, timeout=10, retry_interval=1, condition=lambda d: 'img_data' in d and 'img_type' in d)
+    if not data: return "Screenshot capture timed out"
+    if 'error' in data: return f"Screenshot failed: {data['error']}"
+    return ToolResult(data=data['img_data'], result_type=data['img_type'])

{dialoghelper-0.0.23.dist-info → dialoghelper-0.0.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.23
+Version: 0.0.24
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard

dialoghelper-0.0.24.dist-info/RECORD

@@ -0,0 +1,11 @@
+dialoghelper/__init__.py,sha256=XceGW5BBYEHXaLYMK7w89BsJFYgkMux1YgdeR7Odp4A,43
+dialoghelper/_modidx.py,sha256=agYQ9ims3OEGqX9xZZMHvnFP35Ks-tyqJ4yHAwbe7-M,4186
+dialoghelper/core.py,sha256=U9FvC8ZR6-v7X_PoP-D2etYrZX_Agzed5xpyNFL6a-8,11160
+dialoghelper/db_dc.py,sha256=mi2Q2am_SoAPGpNQg7KPFS5pR9WEapRXT8ypkNmMfw0,2330
+dialoghelper/experimental.py,sha256=cCH1ndJgj7cM6m5-074d7cJ0kFnjkPJPLO3E19Nquvs,4728
+dialoghelper-0.0.24.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dialoghelper-0.0.24.dist-info/METADATA,sha256=KAaLJ0bJXWsMvZ_GnYgJ-9VvCmczMeKGfQShaAKL2Mk,2590
+dialoghelper-0.0.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dialoghelper-0.0.24.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
+dialoghelper-0.0.24.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
+dialoghelper-0.0.24.dist-info/RECORD,,

dialoghelper-0.0.23.dist-info/RECORD

@@ -1,11 +0,0 @@
-dialoghelper/__init__.py,sha256=AqD8emlypM0LZM9Km2Wj1qvaXCNpm0U46JDDXPEy5d8,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.23.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dialoghelper-0.0.23.dist-info/METADATA,sha256=wQW5EBgPIq4cQbqNn7EqZ2Qm8nv4VxgAwUUEXvwRnbk,2590
-dialoghelper-0.0.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dialoghelper-0.0.23.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
-dialoghelper-0.0.23.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
-dialoghelper-0.0.23.dist-info/RECORD,,