Added --gui functionality

This commit is contained in:
Paul Gauthier 2024-04-28 14:08:25 -07:00
parent d60b343274
commit 542c091cf3
2 changed files with 33 additions and 0 deletions

View file

@ -364,5 +364,12 @@ def get_parser(default_config_files, git_root):
" or home directory)" " or home directory)"
), ),
) )
group.add_argument(
"--gui",
"--browser",
action="store_true",
help="Run aider in your browser",
default=False,
)
return parser return parser

View file

@ -5,6 +5,8 @@ from pathlib import Path
import git import git
import litellm import litellm
from click.testing import CliRunner
from streamlit.web import cli
from aider import models from aider import models
from aider.args import get_parser from aider.args import get_parser
@ -141,6 +143,26 @@ def scrub_sensitive_info(args, text):
return text return text
def launch_gui(args):
from aider import gui
target = gui.__file__
# from streamlit.web import bootstrap
# bootstrap.load_config_options(flag_options={})
# cli.main_run(target, args)
# sys.argv = ['streamlit', 'run', '--'] + args
runner = CliRunner()
st_args = [
"run",
target,
"--global.developmentMode=false",
"--",
]
runner.invoke(cli.main, st_args + args)
def main(argv=None, input=None, output=None, force_git_root=None, return_coder=False): def main(argv=None, input=None, output=None, force_git_root=None, return_coder=False):
if argv is None: if argv is None:
argv = sys.argv[1:] argv = sys.argv[1:]
@ -163,6 +185,10 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
parser = get_parser(default_config_files, git_root) parser = get_parser(default_config_files, git_root)
args = parser.parse_args(argv) args = parser.parse_args(argv)
if args.gui and not return_coder:
launch_gui(argv)
return
if args.dark_mode: if args.dark_mode:
args.user_input_color = "#32FF32" args.user_input_color = "#32FF32"
args.tool_error_color = "#FF3333" args.tool_error_color = "#FF3333"