launch.py

#!/usr/bin/env python3
"""
launch.py  —  run this instead of `streamlit run src/ui/app.py`
 
    python /mnt/w/SecGraph-AI/launch.py
 
Sets PYTHONPATH at the OS environment level BEFORE the subprocess starts,
so Streamlit's own sys.path manipulation cannot remove it.
"""
import os
import sys
import subprocess
from pathlib import Path
 
SRC = Path(__file__).resolve().parent / "src"
 
env = os.environ.copy()
existing = env.get("PYTHONPATH", "")
env["PYTHONPATH"] = f"{SRC}:{existing}" if existing else str(SRC)
 
app = SRC / "ui" / "app.py"
 
print(f"PYTHONPATH = {env['PYTHONPATH']}")
print(f"Launching  : streamlit run {app}\n")
 
result = subprocess.run(
    [sys.executable, "-m", "streamlit", "run", str(app)] + sys.argv[1:],
    env=env,
)
sys.exit(result.returncode)