create-browser-app 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

create_browser_app/main.py

@@ -10,13 +10,13 @@ from rich.panel import Panel
 from rich.text import Text
 import shutil
 from .templates import TEMPLATE_BASIC, TEMPLATE_REQUIREMENTS, TEMPLATE_ENV, TEMPLATE_README, TEMPLATE_GITIGNORE
-from .template_fetcher import get_template_by_name, fetch_template_content
+from .template_fetcher import fetch_all_template_files
 
 console = Console()
 
 @click.command()
 @click.argument('name', default='my-stagehand-app', required=False)
-@click.option('--template', '-t', default='basic', help='Template to use (basic or GitHub examples: quickstart, example, agent-example)')
+@click.option('--template', '-t', default='basic', help='Template to use (basic or dynamically fetched from GitHub)')
 def main(name, template):
     """Start your Stagehand project with a single command"""
 
@@ -36,7 +36,11 @@ def main(name, template):
     console.print("[yellow]Stagehand[/yellow]")
     console.print("[dim]The AI Browser Framework[/dim]\n")
 
-    project_name = name
+    # If user specified a template but not a custom name, use the template name
+    if name == 'my-stagehand-app' and template != 'basic':
+        project_name = template
+    else:
+        project_name = name
 
     # Sanitize project name for directory
     project_dir = project_name.lower().replace(" ", "-").replace("_", "-")
@@ -54,42 +58,27 @@ def main(name, template):
         # Create directories
         project_path.mkdir(parents=True, exist_ok=True)
 
-        # Determine which template to use
-        template_content = TEMPLATE_BASIC
-
-        # If not using basic template, try to fetch from GitHub
+        # Fetch template files from GitHub (if not basic)
+        template_files = {}
         if template != 'basic':
             console.print(f"Fetching template [cyan]{template}[/cyan] from GitHub...")
-            template_info = get_template_by_name(template)
-            if template_info:
-                fetched_content = fetch_template_content(template_info)
-                if fetched_content:
-                    template_content = fetched_content
-                    console.print(f"[green]✓[/green] Using template from GitHub: {template}")
-                else:
-                    console.print(f"[yellow]⚠[/yellow] Could not fetch template, using basic template")
+            template_files = fetch_all_template_files(template)
+            if template_files:
+                console.print(f"[green]✓[/green] Fetched {len(template_files)} files from GitHub: {template}")
             else:
-                console.print(f"[yellow]⚠[/yellow] Template '{template}' not found, using basic template")
-
-        # Create main.py
-        main_file = project_path / "main.py"
-        main_file.write_text(template_content)
-
-        # Create requirements.txt
-        requirements_file = project_path / "requirements.txt"
-        requirements_file.write_text(TEMPLATE_REQUIREMENTS)
-
-        # Create .env.example
-        env_file = project_path / ".env.example"
-        env_file.write_text(TEMPLATE_ENV)
-
-        # Create README.md
-        readme_file = project_path / "README.md"
-        readme_file.write_text(TEMPLATE_README.format(project_name=project_name))
-
-        # Create .gitignore
-        gitignore_file = project_path / ".gitignore"
-        gitignore_file.write_text(TEMPLATE_GITIGNORE)
+                console.print(f"[yellow]⚠[/yellow] Could not fetch template, using basic template")
+
+        # Write all template files to the project
+        if template_files:
+            for filename, content in template_files.items():
+                (project_path / filename).write_text(content)
+        else:
+            # Fallback to basic template defaults
+            (project_path / "main.py").write_text(TEMPLATE_BASIC)
+            (project_path / "requirements.txt").write_text(TEMPLATE_REQUIREMENTS)
+            (project_path / "README.md").write_text(TEMPLATE_README.format(project_name=project_name))
+            (project_path / ".env.example").write_text(TEMPLATE_ENV)
+            (project_path / ".gitignore").write_text(TEMPLATE_GITIGNORE)
 
         # Success message
         console.print(f"[green]✓[/green] Find your project at [cyan]{project_path}[/cyan]\n")

create_browser_app/template_fetcher.py

@@ -8,50 +8,42 @@ class TemplateInfo:
     name: str
     url: str
 
-GITHUB_TEMPLATES: List[TemplateInfo] = [
-    TemplateInfo(
-        name="example",
-        url="https://raw.githubusercontent.com/browserbase/stagehand-python/main/examples/example.py",
-    ),
-    TemplateInfo(
-        name="cua-example",
-        url="https://raw.githubusercontent.com/browserbase/stagehand-python/main/examples/cua-example.py",
-    ),
-    TemplateInfo(
-        name="form-filling",
-        url="https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/form-filling/main.py",
-    ),
-    TemplateInfo(
-        name="gift-finder",
-        url="https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/gift-finder/main.py",
-    ),
-    TemplateInfo(
-        name="pickleball",
-        url="https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/pickleball/main.py",
-    ),
-    TemplateInfo(
-        name="license-verification",
-        url="https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/license-verification/main.py",
-    ),
-    TemplateInfo(
-        name="context",
-        url="https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/context/main.py",
-    ),
-    TemplateInfo(
-        name="proxies",
-        url="https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/proxies/main.py",
-    ),
-    TemplateInfo(
-        name="gemini-cua",
-        url="https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/gemini-cua/main.py",
-    ),
-]
+def fetch_github_templates() -> List[str]:
+    """Fetch list of Python templates from GitHub API."""
+    try:
+        response = requests.get(
+            "https://api.github.com/repos/browserbase/templates/contents/python",
+            headers={"User-Agent": "create-browser-app-py"},
+            timeout=10
+        )
+        
+        if response.status_code != 200:
+            raise Exception(f"GitHub API returned status {response.status_code}")
+        
+        data = response.json()
+        if not isinstance(data, list):
+            raise Exception("Invalid response format from GitHub API")
+        
+        # Filter directories and exclude README.md
+        template_names = []
+        for item in data:
+            if (isinstance(item, dict) and 
+                item.get("type") == "dir" and 
+                item.get("name") != "README.md"):
+                template_names.append(item["name"])
+        
+        return sorted(template_names)
+        
+    except Exception as e:
+        # Re-raise with more context for debugging
+        raise Exception(f"Failed to fetch templates from GitHub: {e}")
+
 def get_template_by_name(name: str) -> Optional[TemplateInfo]:
     """Get a specific template by name."""
-    for template in GITHUB_TEMPLATES:
-        if template.name == name:
-            return template
-    return None
+    # Normalize to lowercase to match GitHub directory naming convention
+    normalized_name = name.lower()
+    url = f"https://raw.githubusercontent.com/browserbase/templates/refs/heads/dev/python/{normalized_name}/main.py"
+    return TemplateInfo(name=normalized_name, url=url)
 
 def fetch_template_content(template: TemplateInfo) -> Optional[str]:
     """Fetch the content of a specific template from GitHub."""
@@ -67,11 +59,68 @@ def fetch_template_content(template: TemplateInfo) -> Optional[str]:
     except Exception:
         return None
 
+def fetch_template_files(template_name: str) -> List[Dict]:
+    """Fetch list of files in a template directory from GitHub API."""
+    normalized_name = template_name.lower()
+    try:
+        response = requests.get(
+            f"https://api.github.com/repos/browserbase/templates/contents/python/{normalized_name}",
+            headers={"User-Agent": "create-browser-app-py"},
+            timeout=10
+        )
+        if response.status_code != 200:
+            return []
+
+        data = response.json()
+        if not isinstance(data, list):
+            return []
+
+        # Return file info for each file (not directories)
+        files = []
+        for item in data:
+            if isinstance(item, dict) and item.get("type") == "file":
+                files.append({
+                    "name": item["name"],
+                    "download_url": item.get("download_url")
+                })
+        return files
+    except Exception:
+        return []
+
+def fetch_all_template_files(template_name: str) -> Dict[str, str]:
+    """Fetch all files from a template directory.
+
+    Returns dict mapping filename -> content.
+    """
+    files = fetch_template_files(template_name)
+    if not files:
+        return {}
+
+    result = {}
+    for file_info in files:
+        name = file_info["name"]
+        download_url = file_info.get("download_url")
+        if download_url:
+            try:
+                response = requests.get(
+                    download_url,
+                    headers={"User-Agent": "create-browser-app-py"},
+                    timeout=10
+                )
+                if response.status_code == 200:
+                    result[name] = response.text
+            except Exception:
+                pass  # Skip files that fail to download
+
+    return result
+
 def get_available_templates() -> List[str]:
-    """Get a list of available template names."""
-    default_templates = ["basic"]
-    github_templates = [t.name for t in GITHUB_TEMPLATES]
-    return default_templates + github_templates
+    """Get templates from GitHub API, fallback to basic only."""
+    try:
+        github_templates = fetch_github_templates()
+        return ["basic"] + github_templates
+    except Exception:
+        return ["basic"]  # Fallback when GitHub unreachable
 
 def list_templates() -> List[str]:
     """Get a list of template names."""

create_browser_app/templates.py

@@ -1,69 +1,84 @@
-TEMPLATE_BASIC = '''from stagehand import Stagehand
-import asyncio
+TEMPLATE_BASIC = '''import asyncio
 import os
+
 from dotenv import load_dotenv
 from pydantic import BaseModel
+from stagehand import AsyncStagehand
+
 load_dotenv()
 
+
+class StagehandDescription(BaseModel):
+    answer: str
+
+
+class ModelInfo(BaseModel):
+    name: str
+    provider: str = ""
+    accuracy: float = 0.0
+
+
 async def main():
     """Basic Stagehand browser automation example."""
 
-    # Check all the configurations in StagehandConfig
-    stagehand = Stagehand(
-        env="LOCAL", # or "BROWSERBASE"
-        verbose=1, # 0: silent, 1: info, 2: debug
-        model_name="gpt-4.1",
-        model_api_key=os.getenv("OPENAI_API_KEY"),
-    )
-    await stagehand.init()
-
-    page = stagehand.page
-    await page.goto("https://docs.stagehand.dev")
-    result = await page.extract("In a few words, what is Stagehand?")
-    print()
-    print(result)
-    print()
-
-    await page.act("click on models")
-    await page.act({
-        "method": "click",
-        "selector": "/html/body[1]/div[2]/div[2]/div[3]/div[2]/div[2]/a[1]",
-        "description": "the model evaluation card",
-        "args": []
-    })
-    elements = await page.observe("find the graph with the list of most accurate models")
-    print()
-    print(elements)
-    print()
-
-    class Model(BaseModel):
-        name: str
-        provider: str
-        accuracy: float
-        
-    extraction = await page.extract(
-        "the most accurate model",
-        schema=Model
+    client = AsyncStagehand(
+        browserbase_api_key=os.getenv("BROWSERBASE_API_KEY"),
+        browserbase_project_id=os.getenv("BROWSERBASE_PROJECT_ID"),
+        model_api_key=os.getenv("MODEL_API_KEY"),
     )
-    print()
-    print(extraction)
-    print()
+
+    # Start a Stagehand session
+    session = await client.sessions.create(model_name="google/gemini-3-flash-preview")
+    print(f"Session started: {session.id}")
+    print(f"View live: https://www.browserbase.com/sessions/{session.id}")
+
+    try:
+        # Navigate to a webpage
+        await session.navigate(url="https://docs.stagehand.dev")
+
+        # Extract data using Stagehand AI
+        result = await session.extract(
+            instruction="In a few words, what is Stagehand?",
+            schema=StagehandDescription.model_json_schema(),
+        )
+        print(f"Extracted: {result.data.result}")
+
+        # Perform AI-powered action
+        await session.act(input="click on models")
+
+        # Observe elements on the page
+        observe_response = await session.observe(
+            instruction="find the graph with the list of most accurate models",
+        )
+        print(f"Found {len(observe_response.data.result)} elements")
+
+        # Extract structured data
+        result = await session.extract(
+            instruction="the most accurate model",
+            schema=ModelInfo.model_json_schema(),
+        )
+        print(f"Most accurate model: {result.data.result}")
+
+    finally:
+        await session.end()
+        print("Session ended")
+
+
 if __name__ == "__main__":
     asyncio.run(main())
-
 '''
 
-TEMPLATE_REQUIREMENTS = '''stagehand
+TEMPLATE_REQUIREMENTS = '''stagehand>=3.4.0
+pydantic
 python-dotenv
 '''
 
-TEMPLATE_ENV = '''# Add your environment variables here
-# BROWSERBASE_API_KEY=your_api_key_here
-# BROWSERBASE_PROJECT_ID=your_project_id_here
-# Add your LLM key
-# OPENAI_API_KEY=your_api_key_here
-# ANTHROPIC=your_api_key_here
-# GOOGLE_API_KEY=your_api_key_here
+TEMPLATE_ENV = '''# Browserbase credentials (required)
+BROWSERBASE_API_KEY=your_api_key_here
+BROWSERBASE_PROJECT_ID=your_project_id_here
+
+# LLM API key (required - using Google Gemini by default)
+MODEL_API_KEY=your_api_key_here
 '''
 
 TEMPLATE_README = '''# {project_name}
@@ -81,6 +96,7 @@ pip install -r requirements.txt
 ```bash
 BROWSERBASE_API_KEY=your_api_key_here
 BROWSERBASE_PROJECT_ID=your_project_id_here
+MODEL_API_KEY=your_api_key_here
 ```
 
 3. Run the project:
@@ -90,8 +106,8 @@ python main.py
 
 ## About Stagehand
 
-Stagehand is a Python library for browser automation built on Playwright. Learn more at:
-- [Stagehand Documentation](https://github.com/browserbase/stagehand)
+Stagehand is the AI browser automation framework. Learn more at:
+- [Stagehand Documentation](https://docs.stagehand.dev/v3/sdk/python)
 - [Browserbase](https://browserbase.com)
 '''
 

{create_browser_app-0.1.12.dist-info → create_browser_app-0.1.14.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: create-browser-app
-Version: 0.1.12
+Version: 0.1.14
 Summary: Start your Stagehand project with a single command
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: stagehand
+Requires-Dist: stagehand>=3.4.0
 Requires-Dist: click>=8.0
 Requires-Dist: rich>=13.0
 Requires-Dist: inquirer>=3.0

create_browser_app-0.1.14.dist-info/RECORD

@@ -0,0 +1,9 @@
+create_browser_app/__init__.py,sha256=izlkaSvrxNqPnulbHlD8VYldR618NxRQRlKQFgQn0Gk,42
+create_browser_app/main.py,sha256=oU02yImNWoDAVyk5ULupJUJ4WDfnSreW2TCpjmHwkbw,4637
+create_browser_app/template_fetcher.py,sha256=vpt19HVQLZ8_3It9cEAJhZzAh1KLVj0-ajV4CvgKXBg,4297
+create_browser_app/templates.py,sha256=Vt1oGrWuYfpM0Z8QsY20B7d3x2qHVoUTS3poDihsTvM,3172
+create_browser_app-0.1.14.dist-info/METADATA,sha256=mRBPpJkk4QXzqn8X5C2fbQcc53T2hbkHn7lHmvIunf0,656
+create_browser_app-0.1.14.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+create_browser_app-0.1.14.dist-info/entry_points.txt,sha256=yyAKhFZs2kraGA_ixwnRZeStL0uxWvBCMzysg8S-0VE,68
+create_browser_app-0.1.14.dist-info/top_level.txt,sha256=sLW8imVtlXvOy1D5aGPBDxbMP3Yvonif7VxcLu_Izy4,19
+create_browser_app-0.1.14.dist-info/RECORD,,

{create_browser_app-0.1.12.dist-info → create_browser_app-0.1.14.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

create_browser_app-0.1.12.dist-info/RECORD

@@ -1,9 +0,0 @@
-create_browser_app/__init__.py,sha256=izlkaSvrxNqPnulbHlD8VYldR618NxRQRlKQFgQn0Gk,42
-create_browser_app/main.py,sha256=8n6Mot1AAM5Vwd7YhzUjresboJtwnt-hwFo1_jBI98Q,4864
-create_browser_app/template_fetcher.py,sha256=iu1hhs9RDZJrwpSwdyGJRaGwboMcfkLmzh5PecnYH_U,2690
-create_browser_app/templates.py,sha256=ipfp4zmfbD4-vhoggEW9M_EBXYvvyCzvmHew9tTgi7w,2636
-create_browser_app-0.1.12.dist-info/METADATA,sha256=U_lzsuZJnDfMe-PpL7Lrks0VyrVwa0ri7c2fVuMA7s0,649
-create_browser_app-0.1.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-create_browser_app-0.1.12.dist-info/entry_points.txt,sha256=yyAKhFZs2kraGA_ixwnRZeStL0uxWvBCMzysg8S-0VE,68
-create_browser_app-0.1.12.dist-info/top_level.txt,sha256=sLW8imVtlXvOy1D5aGPBDxbMP3Yvonif7VxcLu_Izy4,19
-create_browser_app-0.1.12.dist-info/RECORD,,