| | |
| | """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() |
| |
|
