diff --git a/src/powerpwn/cli/arguments.py b/src/powerpwn/cli/arguments.py index 3b7fdc8..57b019e 100644 --- a/src/powerpwn/cli/arguments.py +++ b/src/powerpwn/cli/arguments.py @@ -161,6 +161,11 @@ def module_copilot(command_subparsers: argparse.ArgumentParser): dump.add_argument("-d", "--directory", type=str, required=True, help="Path to whoami output directory") dump.add_argument("-g", "--gui", action="store_true", help="Run local server for gui.") + gui = copilot_subparsers.add_parser( + "gui", description="Browse data in a local server", help="Browse collected data in a simple gui on a local server" + ) + gui.add_argument("-d", "--directory", type=str, required=True, help="Data directory") + def copilot_modules(parser): parser.add_argument("-u", "--user", required=True, type=str, help="User email to connect.") diff --git a/src/powerpwn/cli/runners.py b/src/powerpwn/cli/runners.py index 6d43d64..8024727 100644 --- a/src/powerpwn/cli/runners.py +++ b/src/powerpwn/cli/runners.py @@ -178,6 +178,10 @@ def run_phishing_command(args): def run_copilot_chat_command(args): + if args.copilot_subcommand == "gui": + CopilotGui().run(args.directory) + return + parsed_args = ChatArguments( user=args.user, password=args.password,