1#!/usr/bin/env python3
2"""
3Universal runner for NotebookLM skill scripts
4Ensures all scripts run with the correct virtual environment
5"""
6 
7import os
8import sys
9import subprocess
10from pathlib import Path
11 
12 
13def get_venv_python():
14    """Get the virtual environment Python executable"""
15    skill_dir = Path(__file__).parent.parent
16    venv_dir = skill_dir / ".venv"
17 
18    if os.name == 'nt':  # Windows
19        venv_python = venv_dir / "Scripts" / "python.exe"
20    else:  # Unix/Linux/Mac
21        venv_python = venv_dir / "bin" / "python"
22 
23    return venv_python
24 
25 
26def ensure_venv():
27    """Ensure virtual environment exists"""
28    skill_dir = Path(__file__).parent.parent
29    venv_dir = skill_dir / ".venv"
30    setup_script = skill_dir / "scripts" / "setup_environment.py"
31 
32    # Check if venv exists
33    if not venv_dir.exists():
34        print("🔧 First-time setup: Creating virtual environment...")
35        print("   This may take a minute...")
36 
37        # Run setup with system Python
38        result = subprocess.run([sys.executable, str(setup_script)])
39        if result.returncode != 0:
40            print("❌ Failed to set up environment")
41            sys.exit(1)
42 
43        print("✅ Environment ready!")
44 
45    return get_venv_python()
46 
47 
48def main():
49    """Main runner"""
50    if len(sys.argv) < 2:
51        print("Usage: python run.py <script_name> [args...]")
52        print("\nAvailable scripts:")
53        print("  ask_question.py    - Query NotebookLM")
54        print("  notebook_manager.py - Manage notebook library")
55        print("  session_manager.py  - Manage sessions")
56        print("  auth_manager.py     - Handle authentication")
57        print("  cleanup_manager.py  - Clean up skill data")
58        sys.exit(1)
59 
60    script_name = sys.argv[1]
61    script_args = sys.argv[2:]
62 
63    # Handle both "scripts/script.py" and "script.py" formats
64    if script_name.startswith('scripts/'):
65        # Remove the scripts/ prefix if provided
66        script_name = script_name[8:]  # len('scripts/') = 8
67 
68    # Ensure .py extension
69    if not script_name.endswith('.py'):
70        script_name += '.py'
71 
72    # Get script path
73    skill_dir = Path(__file__).parent.parent
74    script_path = skill_dir / "scripts" / script_name
75 
76    if not script_path.exists():
77        print(f"❌ Script not found: {script_name}")
78        print(f"   Working directory: {Path.cwd()}")
79        print(f"   Skill directory: {skill_dir}")
80        print(f"   Looked for: {script_path}")
81        sys.exit(1)
82 
83    # Ensure venv exists and get Python executable
84    venv_python = ensure_venv()
85 
86    # Build command
87    cmd = [str(venv_python), str(script_path)] + script_args
88 
89    # Run the script
90    try:
91        result = subprocess.run(cmd)
92        sys.exit(result.returncode)
93    except KeyboardInterrupt:
94        print("\n⚠️ Interrupted by user")
95        sys.exit(130)
96    except Exception as e:
97        print(f"❌ Error: {e}")
98        sys.exit(1)
99 
100 
101if __name__ == "__main__":
102    main()