| |
| """A CLI for building an running RealChar project locally.""" |
| import click |
| import os |
| import subprocess |
|
|
|
|
| @click.group() |
| def cli(): |
| pass |
|
|
|
|
| @click.command() |
| @click.option('--name', default="realtime-ai-character", help='The name to give to your Docker image.') |
| @click.option('--rebuild', is_flag=True, help='Flag to indicate whether to rebuild the Docker image.') |
| def docker_build(name, rebuild): |
| if rebuild or not image_exists(name): |
| click.secho(f"Building Docker image: {name}...", fg='green') |
| if (image_exists(name)): |
| subprocess.run(["docker", "rmi", "-f", name]) |
| subprocess.run(["docker", "build", "-t", name, "."]) |
| else: |
| click.secho( |
| f"Docker image: {name} already exists. Skipping build. To rebuild, use --rebuild option", fg='yellow') |
|
|
|
|
| @click.command() |
| @click.option('--name', default="realtime-ai-character", help='The name of the Docker image to run.') |
| @click.option('--db-file', default=None, help='Path to the database file to mount inside the container.') |
| def docker_run(name, db_file): |
| click.secho(f"Running Docker image: {name}...", fg='green') |
| if not os.path.isfile('.env'): |
| click.secho( |
| "Warning: .env file not found. Running without environment variables.", fg='yellow') |
| |
| subprocess.run(["docker", "rm", "-f", name], |
| stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) |
| if db_file: |
| subprocess.run(["docker", "run", "--env-file", ".env", "--name", name, "-p", "7860:7860", |
| "-v", f"{os.path.abspath(db_file)}:/realtime_ai_character/test.db", name]) |
| else: |
| subprocess.run(["docker", "run", "--env-file", ".env", |
| "--name", name, "-p", "7860:7860", name]) |
|
|
|
|
| @click.command() |
| @click.option('--name', default="realtime-ai-character", help='The name of the Docker image to delete.') |
| def docker_delete(name): |
| if image_exists(name): |
| click.secho(f"Deleting Docker image: {name}...", fg='green') |
| subprocess.run(["docker", "rmi", "-f", name]) |
| else: |
| click.secho(f"Docker image: {name} does not exist.", fg='yellow') |
|
|
|
|
| @click.command(context_settings={"ignore_unknown_options": True}) |
| @click.argument('args', nargs=-1, type=click.UNPROCESSED) |
| def run_uvicorn(args): |
| click.secho("Running uvicorn server...", fg='green') |
| subprocess.run(["uvicorn", "realtime_ai_character.main:app", |
| "--ws-ping-interval", "60", "--ws-ping-timeout", "60", "--timeout-keep-alive", "60"] + list(args)) |
|
|
|
|
| def image_exists(name): |
| result = subprocess.run( |
| ["docker", "image", "inspect", name], capture_output=True, text=True) |
| return result.returncode == 0 |
|
|
|
|
| cli.add_command(docker_build) |
| cli.add_command(docker_run) |
| cli.add_command(docker_delete) |
| cli.add_command(run_uvicorn) |
|
|
|
|
| if __name__ == '__main__': |
| cli() |
|
|
