simplex 1.2.57__py3-none-any.whl → 1.2.58__py3-none-any.whl

simplex/simplex.py

@@ -4,7 +4,7 @@ from typing import Optional, Union
 import os
 import json
 import warnings
-
+import time
 # Try to import playwright, but don't fail if it's not available (lite version)
 try:
     from playwright.sync_api import sync_playwright
@@ -17,8 +17,8 @@ except ImportError:
         ImportWarning
     )
 
+# BASE_URL = "https://simplex-dev--api-server-fastapi-app.modal.run"
 BASE_URL = "https://api.simplex.sh"
-
 class Playwright:
     def __init__(self, simplex_instance):
         self.simplex = simplex_instance
@@ -61,7 +61,7 @@ class Playwright:
             print(response.json())
             raise ValueError(f"It looks like the click action with playwright failed to return a response. Did you set your api_key when creating the Simplex class?")
         
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to click element: {response.json()['error']}")
@@ -81,6 +81,7 @@ class Simplex:
             self.pw = None
         
     def close_session(self):
+        time.sleep(30)
         if PLAYWRIGHT_AVAILABLE and self.pw_browser:
             try:
                 self.pw_browser.close()
@@ -105,7 +106,7 @@ class Simplex:
         self.session_id = None
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the close_session action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to close session: {response.json()['error']}")
@@ -158,7 +159,9 @@ class Simplex:
             self.pw = sync_playwright().start()
         self.connect_url = response_json['connect_url']
         # self.pw_browser = self.pw.chromium.connect_over_cdp(response_json['connect_url'])
-
+        if not "api.simplex.sh" in BASE_URL:
+            livestream_url = f"http://localhost:3000/session/{self.session_id}"
+        
         if show_in_console:
             print(f"Livestream URL: {livestream_url}")
 
@@ -188,12 +191,14 @@ class Simplex:
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the goto action failed to return a response. Did you set your api_key when creating the Simplex class?")
     
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to goto url: {response.json()['error']}")
 
     def click(self, element_description: str, cdp_url: str = None):
+        if not element_description or not element_description.strip():
+            raise ValueError("element_description cannot be empty")
         if not cdp_url and not self.session_id:
             raise ValueError(f"Must call create_session before calling action click with element_description='{element_description}'")
 
@@ -213,12 +218,14 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the click action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return response.json()["element_clicked"]
         else:
             raise ValueError(f"Failed to click element: {response.json()['error']}")
 
     def type(self, text: str, cdp_url: str = None):
+        if not text or not text.strip():
+            raise ValueError("text cannot be empty")
         if not cdp_url and not self.session_id:
             raise ValueError(f"Must call create_session before calling action type with text='{text}'")
 
@@ -236,11 +243,38 @@ class Simplex:
             },
             data=data
         )
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
+            print(response.json())
             raise ValueError(f"Failed to type text: {response.json()['error']}")
+    
 
+    def reload(self, cdp_url: str = None):
+        if not cdp_url and not self.session_id:
+            raise ValueError("Must call create_session before calling action reload")
+
+        data = {}
+
+        if cdp_url:
+            data['cdp_url'] = cdp_url
+        else:
+            data['session_id'] = self.session_id
+
+        response = requests.post(
+            f"{BASE_URL}/reload",
+            headers={
+                'x-api-key': self.api_key
+            },
+            data=data
+        )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the reload action failed to return a response. Did you set your api_key when creating the Simplex class?")
+        if "succeeded" in response.json():
+            return
+        else:
+            raise ValueError(f"Failed to reload: {response.json()['error']}")
+        
     def press_enter(self, cdp_url: str = None):
         if not cdp_url and not self.session_id:
             raise ValueError("Must call create_session before calling action press_enter")
@@ -261,7 +295,7 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the press_enter action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to press enter: {response.json()['error']}")
@@ -286,7 +320,7 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the press_tab action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to press tab: {response.json()['error']}")
@@ -311,12 +345,14 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the delete_text action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to delete text: {response.json()['error']}")
 
     def extract_bbox(self, element_description: str, cdp_url: str = None):
+        if not element_description or not element_description.strip():
+            raise ValueError("element_description cannot be empty")
         if not cdp_url and not self.session_id:
             raise ValueError(f"Must call create_session before calling action extract_bbox with element_description='{element_description}'")
 
@@ -336,12 +372,14 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the extract_bbox action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return response.json()["bbox"]
         else:
             raise ValueError(f"Failed to extract bbox: {response.json()['error']}")
 
     def extract_text(self, element_description: str, cdp_url: str = None):
+        if not element_description or not element_description.strip():
+            raise ValueError("element_description cannot be empty")
         if not cdp_url and not self.session_id:
             raise ValueError(f"Must call create_session before calling action extract_text with element_description='{element_description}'")
 
@@ -362,12 +400,14 @@ class Simplex:
         response_json = response.json()
         if 'succeeded' not in response_json:
             raise ValueError(f"It looks like the extract_text action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response_json["succeeded"]:
+        if "succeeded" in response_json:
             return response_json["text"]
         else:
             raise ValueError(f"Failed to extract text: {response_json['error']}")
 
     def extract_image(self, element_description: str, cdp_url: str = None):
+        if not element_description or not element_description.strip():
+            raise ValueError("element_description cannot be empty")
         if not cdp_url and not self.session_id:
             raise ValueError(f"Must call create_session before calling action extract_image with element_description='{element_description}'")
 
@@ -387,7 +427,7 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the extract_image action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return response.json()["image"]
         else:
             raise ValueError(f"Failed to extract image: {response.json()['error']}")
@@ -412,7 +452,7 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the scroll action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to scroll: {response.json()['error']}")
@@ -437,7 +477,7 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the wait action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to wait: {response.json()['error']}")
@@ -554,7 +594,7 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the restore_login_session action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to restore login session: {response.json()['error']}")
@@ -565,6 +605,8 @@ class Simplex:
             element_description: Description of the element to click and upload to
             file_path_or_callable: Either a path to the file to be uploaded or a callable that returns a file-like object in 'rb' mode
         """
+        if not element_description or not element_description.strip():
+            raise ValueError("element_description cannot be empty")
         if not self.session_id:
             raise ValueError(f"Must call create_session before calling action click_and_upload with element_description='{element_description}'")
         
@@ -596,12 +638,14 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the click_and_upload action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return
         else:
             raise ValueError(f"Failed to click and upload: {response.json()['error']}")
 
     def click_and_download(self, element_description: str):
+        if not element_description or not element_description.strip():
+            raise ValueError("element_description cannot be empty")
         if not self.session_id:
             raise ValueError(f"Must call create_session before calling action click_and_download with element_description='{element_description}'")
         
@@ -620,12 +664,14 @@ class Simplex:
         )
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the click_and_download action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
+        if "succeeded" in response.json():
             return response.json()["b64"], response.json()["filename"]
         else:
             raise ValueError(f"Failed to click and download: {response.json()['error']}")
     
     def exists(self, element_description: str, cdp_url: str = None):
+        if not element_description or not element_description.strip():
+            raise ValueError("element_description cannot be empty")
         if not cdp_url and not self.session_id:
             raise ValueError(f"Must call create_session before calling action exists with element_description='{element_description}'")
 
@@ -646,7 +692,7 @@ class Simplex:
         if 'succeeded' not in response.json():
             raise ValueError(f"It looks like the exists action failed to return a response. Did you set your api_key when creating the Simplex class?")
         response_json = response.json()
-        if response_json['succeeded']:
+        if "succeeded" in response_json:
             return response_json['exists'], response_json['reasoning']
         else:
             raise ValueError(f"Failed to check if element exists: {response_json['error']}")

{simplex-1.2.57.dist-info → simplex-1.2.58.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: simplex
-Version: 1.2.57
+Version: 1.2.58
 Summary: Official Python SDK for Simplex API
 Home-page: https://simplex.sh
 Author: Simplex Labs, Inc.

simplex-1.2.58.dist-info/RECORD

@@ -0,0 +1,11 @@
+simplex/__init__.py,sha256=L_5i__xt_ZDkr6e-Wx9cr84t9sXpioOT7j01NJYJCTE,75
+simplex/cli.py,sha256=0K_pzoVdF7vfTJPUONhFzzTvjZk2M7FZPpEONjZWJC8,2268
+simplex/simplex.py,sha256=1rHcnBxP9KGXAARMg0R-sHpvWHha3oqvVUQIv4qvzIY,27649
+simplex/deploy/__init__.py,sha256=_JQ81F_Nu7hSAfMA691gzs6a4-8oZ-buJ9h3Au12BKw,96
+simplex/deploy/push.py,sha256=hRAbtFZaECKnBljaOLQ5nzJ6hk7tZgc1c7QdgxKQFoY,6123
+simplex-1.2.58.dist-info/LICENSE,sha256=Xh0SJjYZfNI71pCNMB40aKlBLLuOB0blx5xkTtufFNQ,1075
+simplex-1.2.58.dist-info/METADATA,sha256=XLJEsvkYBQZpZPYobkRrGbhECHqsC7VW2bXlU64dUuI,1045
+simplex-1.2.58.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+simplex-1.2.58.dist-info/entry_points.txt,sha256=3veL2w3c5vxb3dm8I_M8Fs-370n1ZnvD8uu1nSsL7z8,45
+simplex-1.2.58.dist-info/top_level.txt,sha256=cbMH1bYpN0A3gP-ecibPRHasHoqB-01T_2BUFS8p0CE,8
+simplex-1.2.58.dist-info/RECORD,,

simplex-1.2.57.dist-info/RECORD

@@ -1,11 +0,0 @@
-simplex/__init__.py,sha256=L_5i__xt_ZDkr6e-Wx9cr84t9sXpioOT7j01NJYJCTE,75
-simplex/cli.py,sha256=0K_pzoVdF7vfTJPUONhFzzTvjZk2M7FZPpEONjZWJC8,2268
-simplex/simplex.py,sha256=qz9vUcp0HVIN_weBxv_sBCY1pw8LTObR3fcHcU_0gZQ,25413
-simplex/deploy/__init__.py,sha256=_JQ81F_Nu7hSAfMA691gzs6a4-8oZ-buJ9h3Au12BKw,96
-simplex/deploy/push.py,sha256=hRAbtFZaECKnBljaOLQ5nzJ6hk7tZgc1c7QdgxKQFoY,6123
-simplex-1.2.57.dist-info/LICENSE,sha256=Xh0SJjYZfNI71pCNMB40aKlBLLuOB0blx5xkTtufFNQ,1075
-simplex-1.2.57.dist-info/METADATA,sha256=actbMwYSlAmBFJC9spU_A845aGqUEd3VhUhYJWV42os,1045
-simplex-1.2.57.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-simplex-1.2.57.dist-info/entry_points.txt,sha256=3veL2w3c5vxb3dm8I_M8Fs-370n1ZnvD8uu1nSsL7z8,45
-simplex-1.2.57.dist-info/top_level.txt,sha256=cbMH1bYpN0A3gP-ecibPRHasHoqB-01T_2BUFS8p0CE,8
-simplex-1.2.57.dist-info/RECORD,,