simplex 1.2.11__py3-none-any.whl → 1.2.12__py3-none-any.whl

simplex/cli.py

@@ -1,6 +1,24 @@
 import click
 from .deploy import push_directory, run_directory
+from .simplex import Simplex
 import os
+from dotenv import load_dotenv
+import webbrowser
+from colorama import init, Fore, Style
+import time
+import sys
+
+init()  # Initialize colorama
+
+load_dotenv()
+
+def animated_print(message, color=Fore.CYAN):
+    """Print with animated ellipsis"""
+    for i in range(3):
+        sys.stdout.write(f'\r{color}{message}{"." * (i+1)}{" " * (2-i)}{Style.RESET_ALL}')
+        sys.stdout.flush()
+        time.sleep(0.5)
+    sys.stdout.write('\n')
 
 @click.group()
 def cli():
@@ -13,7 +31,7 @@ def push(directory):
     try:
         push_directory(directory)
     except Exception as e:
-        print(f"Error running job: {e}")
+        print(f"{Fore.RED}[SIMPLEX] Error running job: {e}{Style.RESET_ALL}")
         raise
     
     
@@ -24,8 +42,42 @@ def run(directory):
     try:
         run_directory(directory)
     except Exception as e:
-        print(f"Error running job: {e}")
+        print(f"{Fore.RED}[SIMPLEX] Error running job: {e}{Style.RESET_ALL}")
         raise
 
+@cli.command()
+@click.argument('website')
+def login(website):
+    """Capture login session for a website"""
+    try:
+        # Initialize Simplex with API key from environment
+        api_key = os.getenv("SIMPLEX_API_KEY")
+        if not api_key:
+            raise click.ClickException("SIMPLEX_API_KEY environment variable not set")
+        
+        simplex = Simplex(api_key)
+        
+        # Ensure website has proper URL format
+        if not website.startswith(('http://', 'https://')):
+            website = 'https://' + website
+            
+        animated_print(f"[SIMPLEX] Creating login session for {website}")
+        result = simplex.capture_login_session(website)
+        
+        if result.get('succeeded'):
+            # Open the login URL in a new browser tab
+            login_url = result.get('login_session_url')
+            if login_url:
+                animated_print("[SIMPLEX] Opening login page in your browser")
+                webbrowser.open_new_tab(login_url)
+            else:
+                print(f"{Fore.YELLOW}[SIMPLEX] Warning: No login URL returned from the server{Style.RESET_ALL}")
+        else:
+            print(f"{Fore.RED}[SIMPLEX] Failed to capture login session: {result.get('error', 'Unknown error')}{Style.RESET_ALL}")
+    except Exception as e:
+        raise click.ClickException(str(e))
+    finally:
+        print(f"{Fore.GREEN}[SIMPLEX] Login session closed.{Style.RESET_ALL}")
+
 def main():
     cli() 

simplex/deploy/push.py

@@ -7,8 +7,7 @@ from dotenv import load_dotenv
 import requests
 from typing import Optional
 from datetime import datetime, timedelta
-
-
+load_dotenv()  # This loads the .env file
 
 S3_URL = "https://u3mvtbirxf.us-east-1.awsapprunner.com"
 JOB_URL = "https://simplex--job-scheduler-app-handler.modal.run"
@@ -42,7 +41,6 @@ def create_archive(directory: str, config: dict) -> str:
 
 def push_directory(directory: str):
     """Main function to handle parsing yaml and creating zip archive"""
-    
     config = read_config(directory)
     archive_path = create_archive(directory, config)
     
@@ -55,7 +53,7 @@ def push_directory(directory: str):
         raise ValueError("Cron schedule not specified in simplex.yaml")
     
     try:
-        s3_uri = upload_files_to_s3(directory, archive_path, project_name)
+        s3_uri = upload_files_to_s3(archive_path, project_name)
     except Exception as e:
         print(f"Error uploading project: {e}")
         raise
@@ -63,7 +61,7 @@ def push_directory(directory: str):
         os.unlink(archive_path)
 
     try:
-        create_job_scheduled(directory, project_name, cron_schedule, s3_uri)
+        create_job_scheduled(project_name, cron_schedule, s3_uri)
     except Exception as e:
         print(f"Error creating job: {e}")
         raise
@@ -72,13 +70,7 @@ def push_directory(directory: str):
     
     
 
-def upload_files_to_s3(directory: str, archive_path: str, project_name: str):
-    """Upload files to S3"""
-    # Load .env file from the target directory
-    env_path = Path(directory) / '.env'
-    if env_path.exists():
-        load_dotenv(env_path)
-
+def upload_files_to_s3(archive_path: str, project_name: str):
     # Read API key from environment
     api_key = os.getenv('SIMPLEX_API_KEY')
     if not api_key:
@@ -115,14 +107,8 @@ def upload_files_to_s3(directory: str, archive_path: str, project_name: str):
         print(f"Error uploading project: {e}")
         raise
 
-def create_job_scheduled(directory: str, project_name: str, cron_schedule: str, s3_uri: str):
+def create_job_scheduled(project_name: str, cron_schedule: str, s3_uri: str):
     """Create a new job via the API"""
-
-    # Load .env file from the target directory
-    env_path = Path(directory) / '.env'
-    if env_path.exists():
-        load_dotenv(env_path)
-
     # Read API key from environment
     api_key = os.getenv('SIMPLEX_API_KEY')
     if not api_key:
@@ -159,14 +145,8 @@ def create_job_scheduled(directory: str, project_name: str, cron_schedule: str,
         raise
 
     
-def create_job_immediately(directory: str, project_name: str, s3_uri: str):
+def create_job_immediately(project_name: str, s3_uri: str):
     """Create a new job via the API"""
-
-    # Load .env file from the target directory
-    env_path = Path(directory) / '.env'
-    if env_path.exists():
-        load_dotenv(env_path)
-
     # Read API key from environment
     api_key = os.getenv('SIMPLEX_API_KEY')
     if not api_key:
@@ -205,11 +185,6 @@ def create_job_immediately(directory: str, project_name: str, s3_uri: str):
 
 def run_directory(directory: str):
     """Run a directory immediately"""
-    # Load .env file from the target directory
-    env_path = Path(directory) / '.env'
-    if env_path.exists():
-        load_dotenv(env_path)
-        
     config = read_config(directory)
     archive_path = create_archive(directory, config)
     
@@ -218,7 +193,7 @@ def run_directory(directory: str):
         raise ValueError("Project name not specified in simplex.yaml")
     
     try:
-        s3_uri = upload_files_to_s3(directory, archive_path, project_name)
+        s3_uri = upload_files_to_s3(archive_path, project_name)
     except Exception as e:
         print(f"Error uploading project: {e}")
         raise
@@ -227,7 +202,7 @@ def run_directory(directory: str):
         os.unlink(archive_path)
 
     try:
-        create_job_immediately(directory, project_name, s3_uri)
+        create_job_immediately(project_name, s3_uri)
     except Exception as e:
         print(f"Error creating job: {e}")
         raise

simplex/simplex.py

@@ -3,24 +3,18 @@ import requests
 import atexit
 
 BASE_URL = "https://api.simplex.sh"
+
 import json
 
 class Simplex:
     def __init__(self, api_key: str):
         self.api_key = api_key
         self.session_id = None
-        atexit.register(self.cleanup_session)
-
-    def cleanup_session(self):
-        if self.session_id:
-            try:
-                self.close_session(self.session_id)
-            except:
-                pass
+        atexit.register(self.close_session)
 
     def close_session(self):
         response = requests.post(
-            f"{BASE_URL}/close",
+            f"{BASE_URL}/close-session",
             headers={
                 'x-api-key': self.api_key
             },
@@ -41,10 +35,6 @@ class Simplex:
         self.session_id = response_json['session_id']
         livestream_url = response_json['livestream_url']
 
-        print(f"\nSession created successfully!")
-        print(f"Session ID: {self.session_id}")
-        print(f"Livestream URL: {livestream_url}\n")
-
         return self.session_id, livestream_url
 
     def goto(self, url: str, cdp_url: str = None):
@@ -70,7 +60,7 @@ class Simplex:
         )
         return response.json()
 
-    def click(self, element_description: str, cdp_url: str = None):
+    def click(self, element_description: str, cdp_url: str = None, use_vision: bool = False):
         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}'")
 
@@ -81,6 +71,9 @@ class Simplex:
         else:
             data['session_id'] = self.session_id
 
+        if use_vision:
+            data['use_vision'] = use_vision
+
         response = requests.post(
             f"{BASE_URL}/click",
             headers={
@@ -268,11 +261,55 @@ class Simplex:
         )
         return response.json()
 
-    def run_agent(self, prompt: str, cdp_url: str = None):
+    def click_and_upload(self, element_description: str, file_path: str):
+        if not self.session_id:
+            raise ValueError(f"Must call create_session before calling action click_and_upload with element_description='{element_description}'")
+
+        files = {
+            'zip_file': open(file_path, 'rb')
+        }
+        
+        data = {
+            'element_description': element_description
+        }
+
+        data['session_id'] = self.session_id
+
+        response = requests.post(
+            f"{BASE_URL}/click_and_upload",
+            headers={
+                'x-api-key': self.api_key
+            },
+            files=files,
+            data=data
+        )
+        return response.json()
+    
+    def click_and_download(self, element_description: str):
+        if not self.session_id:
+            raise ValueError(f"Must call create_session before calling action click_and_download with element_description='{element_description}'")
+        
+        data = {
+            'element_description': element_description
+        }
+
+        data['session_id'] = self.session_id
+
+        response = requests.post(
+            f"{BASE_URL}/click_and_download",
+            headers={
+                'x-api-key': self.api_key
+            },
+            data=data
+        )
+        
+        return response.json()
+    
+    def exists(self, element_description: str, cdp_url: str = None):
         if not cdp_url and not self.session_id:
-            raise ValueError(f"Must call create_session before calling action run_agent with and prompt='{prompt}'")
+            raise ValueError(f"Must call create_session before calling action exists with element_description='{element_description}'")
 
-        data = {'prompt': prompt}
+        data = {'element_description': element_description}
 
         if cdp_url:
             data['cdp_url'] = cdp_url
@@ -280,7 +317,7 @@ class Simplex:
             data['session_id'] = self.session_id
 
         response = requests.post(
-            f"{BASE_URL}/run-agent",
+            f"{BASE_URL}/exists",
             headers={
                 'x-api-key': self.api_key
             },

{simplex-1.2.11.dist-info → simplex-1.2.12.dist-info}/METADATA

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

simplex-1.2.12.dist-info/RECORD

@@ -0,0 +1,11 @@
+simplex/__init__.py,sha256=1mbM4XUk0FNW161WOkM4ayC1s_QSsaBEls6PZ0iBScY,74
+simplex/cli.py,sha256=yirCLDIF7u0d6LIDY2WWZ5IJJrlxdNsGv9B3H8nh34U,2643
+simplex/simplex.py,sha256=B6IJ1Ua0wLNXZDmEOkDN-WvXtPZmXS5VOgKb48KLB9I,9784
+simplex/deploy/__init__.py,sha256=_JQ81F_Nu7hSAfMA691gzs6a4-8oZ-buJ9h3Au12BKw,96
+simplex/deploy/push.py,sha256=hRAbtFZaECKnBljaOLQ5nzJ6hk7tZgc1c7QdgxKQFoY,6123
+simplex-1.2.12.dist-info/LICENSE,sha256=Xh0SJjYZfNI71pCNMB40aKlBLLuOB0blx5xkTtufFNQ,1075
+simplex-1.2.12.dist-info/METADATA,sha256=-CnNY6j8ZYWSiPUdvFXnDAkXXxxeDJbdRX1qSl7leH0,1350
+simplex-1.2.12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+simplex-1.2.12.dist-info/entry_points.txt,sha256=3veL2w3c5vxb3dm8I_M8Fs-370n1ZnvD8uu1nSsL7z8,45
+simplex-1.2.12.dist-info/top_level.txt,sha256=cbMH1bYpN0A3gP-ecibPRHasHoqB-01T_2BUFS8p0CE,8
+simplex-1.2.12.dist-info/RECORD,,

simplex-1.2.11.dist-info/RECORD

@@ -1,11 +0,0 @@
-simplex/__init__.py,sha256=1mbM4XUk0FNW161WOkM4ayC1s_QSsaBEls6PZ0iBScY,74
-simplex/cli.py,sha256=PkHt3sBKYfu0Smil4aCMoZUD-JbPw8xsBewpFcBYDKM,675
-simplex/simplex.py,sha256=wXys677gzVGlsPPdILNJi03mjNKo6fOfyGntD4Pa3Gg,8624
-simplex/deploy/__init__.py,sha256=_JQ81F_Nu7hSAfMA691gzs6a4-8oZ-buJ9h3Au12BKw,96
-simplex/deploy/push.py,sha256=RB2ZL-jTiLNvdLDD-TqibHM5I9WCmH7DXsIgtWBLjcs,6795
-simplex-1.2.11.dist-info/LICENSE,sha256=Xh0SJjYZfNI71pCNMB40aKlBLLuOB0blx5xkTtufFNQ,1075
-simplex-1.2.11.dist-info/METADATA,sha256=eMWpq4E-4rGapzadf-SJQI5yTFDQ2jz5Ra7zC0dGets,1350
-simplex-1.2.11.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-simplex-1.2.11.dist-info/entry_points.txt,sha256=3veL2w3c5vxb3dm8I_M8Fs-370n1ZnvD8uu1nSsL7z8,45
-simplex-1.2.11.dist-info/top_level.txt,sha256=cbMH1bYpN0A3gP-ecibPRHasHoqB-01T_2BUFS8p0CE,8
-simplex-1.2.11.dist-info/RECORD,,