flowchart TD
 %% Nodes Styling
 classDef user fill:#f9f,stroke:#333,stroke-width:2px,color:black;
 classDef system fill:#e1f5fe,stroke:#0277bd,stroke-width:2px,color:black;
 classDef external fill:#fff9c4,stroke:#fbc02d,stroke-width:2px,color:black;
 classDef error fill:#ffcdd2,stroke:#c62828,stroke-width:2px,color:black;
 Start([User Action]) --> InstallCheck{Is vibecheck installed?}
 class Start user
 %% Installation Branch
 InstallCheck -- No --> OS{Select OS}
 OS -- macOS/Linux --> Curl[Run curl command]
 OS -- Windows --> PS[Run PowerShell as Admin]
 OS -- macOS Brew --> Brew[Run brew install]
 Curl & PS & Brew --> Config[Setup API Keys]
 Config --> EnvVars[Export vars OR .env file]
 Config --> KeysCmd[vibecheck keys]
 KeysCmd --> KeysFile[Store in ~/.vibecheck_keys.json]
 %% Main Execution
 InstallCheck -- Yes --> Command{Run Command}
 EnvVars -.-> Command
 KeysFile -.-> Command
 %% Branch: UPGRADE
 Command -- "vibecheck upgrade" --> PermCheck{Protected Dir?}
 PermCheck -- Yes --> Sudo[Auto-run with Sudo] --> UpdateBin[Download & Replace Binary]
 PermCheck -- No --> UpdateBin
 UpdateBin --> End([Done])
 %% Branch: DASHBOARD/MODELS/KEYS
 Command -- "vibecheck dashboard" --> ShowDash[Read Logs] --> DisplayStats[Show Commits & $$ Saved] --> End
 Command -- "vibecheck models" --> ShowMods[List Supported Models] --> SwitchMod[Switch Model Preference] --> End
 Command -- "vibecheck keys" --> KeysUI[Interactive Keys Manager] --> KeysList[List Providers with Status] --> KeysEdit[Add/Edit/Delete Keys] --> KeysSave[Save to ~/.vibecheck_keys.json] --> End
 %% Branch: COMMIT (Core Feature)
 Command -- "vibecheck commit" --> GitCheck{Files Staged?}
 
 %% Git Check Logic
 GitCheck -- No --> ErrStage[Error: Stage files first!]:::error
 ErrStage --> End
 
 GitCheck -- Yes --> KeyCheck{API Key Found?}
 %% Key Lookup Priority
 KeyCheck --> CheckEnv{Check .env file}
 CheckEnv -- Found --> KeyFound[Use Key]
 CheckEnv -- Not Found --> CheckExport{Check export vars}
 CheckExport -- Found --> KeyFound
 CheckExport -- Not Found --> CheckKeysFile{Check vibecheck keys}
 CheckKeysFile -- Found --> KeyFound
 CheckKeysFile -- Not Found --> ErrKey[Error: Missing API Key]:::error
 ErrKey --> Config
 KeyFound --> ProviderSelect{Provider Flag?}
 
 %% Provider Logic
 ProviderSelect -- Default --> DefModel[Default Model]
 ProviderSelect -- "--provider X" --> SelectModel[Select Specific Provider]
 
 subgraph AI_Providers [External AI Cloud / Local]
 direction LR
 OpenAI
 Gemini
 Anthropic
 Groq
 XAI_Grok
 Kimi
 Qwen
 DeepSeek
 Perplexity
 Ollama_Local
 end
 class AI_Providers external
 %% Context Injection
 SelectModel & DefModel --> PromptCheck{Custom Prompt?}
 PromptCheck -- Yes --> InjectContext[Inject User Context]
 PromptCheck -- No --> GenPayload[Prepare Diff Payload]
 
 InjectContext & GenPayload --> AI_Providers
 
 %% Output
 AI_Providers --> Response[Receive Generated Message]
 Response --> Display[Output to Terminal]
 Display --> SaveLog[Update Dashboard Stats]
 SaveLog --> End