Enables cg as a global command (#1229)

Author: jayhack

pyproject.toml

@@ -70,6 +70,7 @@ keywords = [
 ]
 [project.scripts]
 codegen = "codegen.cli.cli:main"
+cg = "codegen.cli.cli:main"
 
 [project.optional-dependencies]
 types = []

src/codegen/cli/auth/login.py

@@ -23,6 +23,10 @@ def login_routine(token: str | None = None) -> str:
  typer.Exit: If login fails
 
  """
+ # Display header like in the main TUI
+ print("\033[38;2;82;19;217m" + "/" * 20 + " Codegen\033[0m")
+ print()
+
  # Try environment variable first
  token = token or global_env.CODEGEN_USER_ACCESS_TOKEN
 
@@ -39,7 +43,7 @@ def login_routine(token: str | None = None) -> str:
  try:
  token_manager = TokenManager()
  token_manager.authenticate_token(token)
- rich.print(f"[green]✓ Stored token and profile to:[/green] {token_manager.token_file}")
+ rich.print(f"[dim]✓ Stored token and profile to:[/dim] [#ffca85]{token_manager.token_file}[/#ffca85]")
 
  # Show organization selector if multiple organizations available
  organizations = get_cached_organizations()
@@ -69,6 +73,12 @@ def login_routine(token: str | None = None) -> str:
  except Exception as e:
  rich.print(f"[yellow]Warning: Could not set default organization: {e}[/yellow]")
 
+ # After successful login, launch the TUI
+ print() # Add some space
+ from codegen.cli.tui.app import run_tui
+
+ run_tui()
+
  return token
  except AuthError as e:
  rich.print(f"[red]Error:[/red] {e!s}")

src/codegen/cli/tui/app.py

@@ -402,7 +402,7 @@ def _create_background_agent(self, prompt: str):
  input("Press Enter to continue...")
  return
 
- print(f"\n🔄 Creating agent run with prompt: '{prompt[:50]}{'...' if len(prompt) > 50 else ''}'")
+ print(f"\n\033[90mCreating agent run with prompt: '{prompt[:50]}{'...' if len(prompt) > 50 else ''}'\033[0m")
 
  try:
  payload = {"prompt": prompt.strip()}
@@ -420,10 +420,10 @@ def _create_background_agent(self, prompt: str):
  status = agent_run_data.get("status", "Unknown")
  web_url = self._generate_agent_url(run_id)
 
- print("\n✅ Agent run created successfully!")
- print(f" Run ID: {run_id}")
- print(f" Status: {status}")
- print(f" Web URL: {web_url}")
+ print("\n\033[90mAgent run created successfully!\033[0m")
+ print(f"\033[90m Run ID: {run_id}\033[0m")
+ print(f"\033[90m Status: {status}\033[0m")
+ print(f"\033[90m Web URL: \033[38;2;255;202;133m{web_url}\033[0m")
 
  # Clear the input
  self.prompt_input = ""
@@ -441,8 +441,8 @@ def _show_post_creation_menu(self, web_url: str):
  """Show menu after successful agent creation."""
  from codegen.cli.utils.inplace_print import inplace_print
 
- print("\nWhat would you like to do next?")
- options = ["open in web preview", "go to recent"]
+ print("\n\033[90mWhat would you like to do next?\033[0m")
+ options = ["Open Trace ↗", "Go to Recent"]
  selected = 0
  prev_lines = 0
 
@@ -470,15 +470,15 @@ def build_lines():
  selected = (selected + 1) % len(options)
  prev_lines = inplace_print(build_lines(), prev_lines)
  elif key == "\r" or key == "\n": # Enter - select option
- if selected == 0: # open in web preview
+ if selected == 0: # Open Trace
  try:
  import webbrowser
 
  webbrowser.open(web_url)
  except Exception as e:
  print(f"\n❌ Failed to open browser: {e}")
  input("Press Enter to continue...")
- elif selected == 1: # go to recent
+ elif selected == 1: # Go to Recent
  self.current_tab = 0 # Switch to recent tab
  self.input_mode = False
  self._load_agent_runs() # Refresh the data
@@ -884,8 +884,16 @@ def _clear_and_redraw(self):
  def run(self):
  """Run the minimal TUI."""
  if not self.is_authenticated:
- print("⚠️ Not authenticated. Please run 'codegen login' first.")
- return
+ # Automatically start login flow for first-time users
+ from codegen.cli.auth.login import login_routine
+
+ try:
+ login_routine()
+ # login_routine will launch TUI after successful authentication
+ return
+ except Exception:
+ # If login fails, just exit gracefully
+ return
 
  # Show UI immediately
  self._clear_and_redraw()

src/codegen/cli/utils/simple_selector.py

@@ -77,7 +77,7 @@ def signal_handler(signum, frame):
  for i, option in enumerate(options):
  display_text = str(option.get(display_key, f"Option {i + 1}"))
  if i == selected:
- print(f" \033[34m→ {display_text}\033[0m")
+ print(f" \033[37m→ {display_text}\033[0m") # White for selected
  else:
  print(f" \033[90m {display_text}\033[0m")
 
@@ -100,7 +100,7 @@ def signal_handler(signum, frame):
  for i, option in enumerate(options):
  display_text = str(option.get(display_key, f"Option {i + 1}"))
  if i == selected:
- print(f" \033[34m→ {display_text}\033[0m\033[K") # Clear to end of line
+ print(f" \033[37m→ {display_text}\033[0m\033[K") # White for selected, clear to end of line
  else:
  print(f" \033[90m {display_text}\033[0m\033[K") # Clear to end of line
  if show_help:
@@ -115,7 +115,7 @@ def signal_handler(signum, frame):
  for i, option in enumerate(options):
  display_text = str(option.get(display_key, f"Option {i + 1}"))
  if i == selected:
- print(f" \033[34m→ {display_text}\033[0m\033[K") # Clear to end of line
+ print(f" \033[37m→ {display_text}\033[0m\033[K") # White for selected, clear to end of line
  else:
  print(f" \033[90m {display_text}\033[0m\033[K") # Clear to end of line
  if show_help: