import typer
import sys
from pathlib import Path

from plume.utils import lazy_module
# from streamlit import cli as stcli

stcli = lazy_module('streamlit.cli')
app = typer.Typer()


@app.command()
def annotation(data_dir: Path, dump_fname: Path = "ui_dump.json", task_id: str = ""):
    annotation_lit_path = Path(__file__).parent / Path("annotation.py")
    if task_id:
        sys.argv = [
            "streamlit",
            "run",
            str(annotation_lit_path),
            "--",
            str(data_dir),
            "--task-id",
            task_id,
            "--dump-fname",
            dump_fname,
        ]
    else:
        sys.argv = [
            "streamlit",
            "run",
            str(annotation_lit_path),
            "--",
            str(data_dir),
            "--dump-fname",
            dump_fname,
        ]
    sys.exit(stcli.main())


@app.command()
def preview(manifest_path: Path):
    annotation_lit_path = Path(__file__).parent / Path("preview.py")
    sys.argv = [
        "streamlit",
        "run",
        str(annotation_lit_path),
        "--",
        str(manifest_path)
    ]
    sys.exit(stcli.main())


@app.command()
def collection(data_dir: Path, task_id: str = ""):
    # TODO: Implement web ui for data collection
    pass


def main():
    app()


if __name__ == "__main__":
    main()