1#!/usr/bin/env python3
2"""
3NotebookLM Skill Scripts Package
4Provides automatic environment management for all scripts
5"""
6 
7import os
8import sys
9import subprocess
10from pathlib import Path
11 
12 
13def ensure_venv_and_run():
14    """
15    Ensure virtual environment exists and run the requested script.
16    This is called when any script is imported or run directly.
17    """
18    # Only do this if we're not already in the skill's venv
19    skill_dir = Path(__file__).parent.parent
20    venv_dir = skill_dir / ".venv"
21 
22    # Check if we're in a venv
23    in_venv = hasattr(sys, 'real_prefix') or (
24        hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix
25    )
26 
27    # Check if it's OUR venv
28    if in_venv:
29        venv_path = Path(sys.prefix)
30        if venv_path == venv_dir:
31            # We're already in the correct venv
32            return
33 
34    # We need to set up or switch to our venv
35    if not venv_dir.exists():
36        print("🔧 First-time setup detected...")
37        print("   Creating isolated environment for NotebookLM skill...")
38        print("   This ensures clean dependency management...")
39 
40        # Create venv
41        import venv
42        venv.create(venv_dir, with_pip=True)
43 
44        # Install requirements
45        requirements_file = skill_dir / "requirements.txt"
46        if requirements_file.exists():
47            if os.name == 'nt':  # Windows
48                pip_exe = venv_dir / "Scripts" / "pip.exe"
49            else:
50                pip_exe = venv_dir / "bin" / "pip"
51 
52            print("   Installing dependencies in isolated environment...")
53            subprocess.run(
54                [str(pip_exe), "install", "-q", "-r", str(requirements_file)],
55                check=True
56            )
57 
58            # Also install patchright's chromium
59            print("   Setting up browser automation...")
60            if os.name == 'nt':
61                python_exe = venv_dir / "Scripts" / "python.exe"
62            else:
63                python_exe = venv_dir / "bin" / "python"
64 
65            subprocess.run(
66                [str(python_exe), "-m", "patchright", "install", "chromium"],
67                check=True,
68                capture_output=True
69            )
70 
71        print("✅ Environment ready! All dependencies isolated in .venv/")
72 
73    # If we're here and not in the venv, we should recommend using the venv
74    if not in_venv:
75        print("\n⚠️  Running outside virtual environment")
76        print("   Recommended: Use scripts/run.py to ensure clean execution")
77        print("   Or activate: source .venv/bin/activate")
78 
79 
80# Check environment when module is imported
81ensure_venv_and_run()