simplex 1.2.1__tar.gz → 1.2.2__tar.gz

{simplex-1.2.1 → simplex-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.1
 Name: simplex
-Version: 1.2.1
+Version: 1.2.2
 Summary: Official Python SDK for Simplex API
 Home-page: https://github.com/shreyka/simplex-python
 Author: Simplex Labs, Inc.
@@ -21,15 +21,6 @@ Requires-Dist: rich>=13.0.0
 Requires-Dist: prompt_toolkit>=3.0.0
 Requires-Dist: playwright>=1.0.0
 Requires-Dist: Pillow>=9.0.0
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
 
 # Simplex AI Python SDK
 

{simplex-1.2.1 → simplex-1.2.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="simplex",
-    version="1.2.1",
+    version="1.2.2",
     packages=find_packages(),
     install_requires=[
         "openai>=1.0.0",

{simplex-1.2.1 → simplex-1.2.2}/simplex/simplex.py

@@ -26,10 +26,9 @@ class Simplex:
             self.driver = self.browser.new_page()
         else:
             self.playwright = None
-            if len(self.browser.contexts) > 0 and len(self.browser.contexts[0].pages) > 0:
-                self.driver = self.browser.contexts[0].pages[0]
-            else:
-                self.driver = self.browser.new_page()
+            self.driver = self.browser.contexts[0].pages[0]
+            self.driver.set_default_navigation_timeout(0)
+            self.driver.set_default_timeout(0)
 
     def find_element(self, element_description: str, state: Image.Image | None = None, annotate: bool = True) -> List[int]:
         """
@@ -42,6 +41,7 @@ class Simplex:
         Returns:
             bounding_box (tuple): [x1, y1, x2, y2] bounding box of the found element
         """
+        print(f"Finding element \"{element_description}\"...")
         if state is None:
             state = self.take_stable_screenshot()
 
@@ -58,7 +58,7 @@ class Simplex:
             'element_description': (None, element_description),
             'api_key': (None, self.api_key)
         }        
-
+        print("Sending screenshot to server...")
         # Make the request
         response = requests.post(
             endpoint,
@@ -122,6 +122,7 @@ class Simplex:
                     }
                 """, bbox)
                 self.driver.wait_for_selector('#simplex-bbox-overlay')
+            print(f"Found element \"{element_description}\" at pixel coordinates {bbox}.")
             return bbox 
         else:
             print("Error:", response.text)
@@ -182,6 +183,8 @@ class Simplex:
         """
         if new_tab:
             self.driver = self.browser.new_page()
+        self.driver.wait_for_load_state()
+        print(f"Navigating to URL {url}...")
         self.driver.goto(url)
 
     def execute_action(self, action: List[List[str]], state: Image.Image | None = None, annotate: bool = True) -> None:
@@ -191,10 +194,8 @@ class Simplex:
         Args:
             action (List[List[str]]): List of actions to perform
         """
+        print(f"Executing action {action}...")
         action_type, description = action
-        if state is None:
-            state = self.take_stable_screenshot()
-
         try:
             if action_type == "CLICK":
                 bbox = self.find_element(description, state, annotate=annotate)
@@ -207,6 +208,7 @@ class Simplex:
                 self.driver.mouse.move(center_x, center_y)
 
             elif action_type == "TYPE":
+                print(f"Typing \"{description}\"...")
                 self.driver.keyboard.type(description)
 
             elif action_type == "ENTER":
@@ -217,6 +219,7 @@ class Simplex:
 
             elif action_type == "WAIT":
                 self.driver.wait_for_timeout(int(description))
+                
 
         except Exception as e:
             print(f"Error executing action: {e}")
@@ -230,6 +233,14 @@ class Simplex:
         actions = self.step_to_action(step_description, state)
         for action in actions:
             self.execute_action(action, annotate=annotate)
+    
+    def extract(self, step_description: str, annotate: bool = True) -> None:
+        """
+        Extract an element from the page
+        """
+        state = self.take_stable_screenshot()
+        
+            # self.execute_action(action, annotate=annotate)
             
     def take_stable_screenshot(self) -> Image.Image:
         """
@@ -238,6 +249,7 @@ class Simplex:
         Returns:
             PIL.Image.Image: Screenshot of the current page
         """
+        print("Taking screenshot of the page...")
         self.driver.wait_for_load_state('networkidle')
         return screenshot_to_image(self.driver.screenshot())
             

{simplex-1.2.1 → simplex-1.2.2}/simplex.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.1
 Name: simplex
-Version: 1.2.1
+Version: 1.2.2
 Summary: Official Python SDK for Simplex API
 Home-page: https://github.com/shreyka/simplex-python
 Author: Simplex Labs, Inc.
@@ -21,15 +21,6 @@ Requires-Dist: rich>=13.0.0
 Requires-Dist: prompt_toolkit>=3.0.0
 Requires-Dist: playwright>=1.0.0
 Requires-Dist: Pillow>=9.0.0
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
 
 # Simplex AI Python SDK
 

{simplex-1.2.1 → simplex-1.2.2}/simplex.egg-info/SOURCES.txt

@@ -3,7 +3,6 @@ README.md
 pyproject.toml
 setup.py
 simplex/__init__.py
-simplex/constants.py
 simplex/simplex.py
 simplex/utils.py
 simplex.egg-info/PKG-INFO

{simplex-1.2.1 → simplex-1.2.2}/tests/test_local.py

@@ -9,6 +9,7 @@ from playwright.sync_api import sync_playwright
 from PIL import Image
 import time
 import os
+from browserbase import Browserbase
 from hyperbrowser import Hyperbrowser
 from hyperbrowser.models.session import CreateSessionParams
 
@@ -155,12 +156,20 @@ def test_find_element_2():
     simplex = Simplex(api_key=os.getenv("SIMPLEX_API_KEY"))
     simplex.goto("https://www.cgtrader.com/")
     simplex.find_element("search bar")
-
+import requests
 def test_hyperbrowser_integration():
     """Test Simplex integration with Hyperbrowser"""
+    import time
+    start_time = time.time()
+
+    bb = Browserbase(api_key=os.getenv("BROWSERBASE_API_KEY"))
+    session_start = time.time()
+    session = bb.sessions.create(project_id=os.getenv("BROWSERBASE_PROJECT_ID"))
+    print(session.id)
+    print(f"Session creation took {time.time() - session_start:.2f}s")
 
     # Initialize Hyperbrowser client
-    client = Hyperbrowser(api_key=os.getenv("HYPERBROWSER_API_KEY"))
+    # client = Hyperbrowser(api_key=os.getenv("HYPERBROWSER_API_KEY"))
     
     # Create session params with CAPTCHA solving enabled
     session_params = CreateSessionParams(
@@ -169,28 +178,45 @@ def test_hyperbrowser_integration():
     )
     
     # Create a new session with params
-    session = client.sessions.create(session_params)
-    ws_endpoint = session.ws_endpoint
+    # session = client.sessions.create(session_params)
+    ws_endpoint = session.connect_url
 
     try:
         with sync_playwright() as p:
             # Connect browser to Hyperbrowser session
+            browser_start = time.time()
             browser = p.chromium.connect_over_cdp(ws_endpoint)
-            
+            print(f"Browser connection took {time.time() - browser_start:.2f}s")
+            url = f"https://api.browserbase.com/v1/sessions/{session.id}/debug"
+            headers = {"X-BB-API-Key": os.getenv("BROWSERBASE_API_KEY")}
+            response = requests.get(url, headers=headers)
+            print(response.text)
+            print(response.json()["pages"][0]["debuggerUrl"])
+
             # Initialize Simplex with the Hyperbrowser-connected page
             simplex = Simplex(api_key=os.getenv("SIMPLEX_API_KEY"), browser=browser)
             
             # Test basic functionality
+            nav_start = time.time()
             simplex.goto("https://www.turbosquid.com/")
-            simplex.do("search for iphone")
+            print(f"Navigation took {time.time() - nav_start:.2f}s")
+            
+            # search_start = time.time()
+            # simplex.do("search for iphone")
+            # print(f"Search action took {time.time() - search_start:.2f}s")
             
             # Verify the search worked by finding the search results
-            bbox = simplex.find_element("search results")
+            find_start = time.time()
+            bbox = simplex.find_element("search results", annotate=True)
+            print(f"Finding results took {time.time() - find_start:.2f}s")
             assert bbox is not None, "Search results not found"
             
     finally:
         # Always stop the Hyperbrowser session
-        client.sessions.stop(session.id)
+        # client.sessions.stop(session.id)
+        browser.close()
+        print(f"Total test time: {time.time() - start_time:.2f}s")
+        
 
 
 def execute_action_test():
@@ -217,6 +243,6 @@ def execute_action_test():
         playwright.stop()
 
 if __name__ == "__main__":
-    execute_action_test()
+    test_hyperbrowser_integration()
     
     

simplex-1.2.1/simplex/constants.py

@@ -1 +0,0 @@
-BASE_URL = "https://u3mvtbirxf.us-east-1.awsapprunner.com"