stacklok · yrobla · Feb 28, 2025 · Feb 19, 2025 · Feb 19, 2025 · Feb 19, 2025
diff --git a/README.md b/README.md
@@ -21,33 +21,54 @@
 
 ---
 
-# CodeGate: secure AI code generation
+# CodeGate: AI Development environments simplified.
 
-**By [Stacklok](https://stacklok.com)**
+**From [Stacklok](https://stacklok.com)**
 
-CodeGate is a **local gateway** that makes AI agents and coding assistants safer. It
-ensures AI-generated recommendations adhere to best practices while safeguarding
-your code's integrity and protecting your privacy. With CodeGate, you can
-confidently leverage AI in your development workflow without sacrificing
-security or productivity.
+CodeGate is the ultimate toolkit for developers using coding assistants to build AI applications. It streamlines multi-environment workflows, enabling you to consume AI with confidence from development to production. It empowers ongoing AI application management by optimizing model routing, prompt tracking and security enforcement, ensuring privacy, compliance, and robust operational integrity.
+
+---
+
+## CodeGate Architecture
 
 <picture>
   <source media="(prefers-color-scheme: dark)" srcset="./static/diagram-dark.png">
   <img alt="CodeGate dashboard" src="./static/diagram-light.png" width="1100px" style="max-width: 100%;">
 </picture>
 
----
-## ✨ Why choose CodeGate?
 
-AI coding assistants are powerful, but they can inadvertently introduce risks.
-CodeGate protects your development process by:
+## 🚀 Why Developers Love CodeGate
+
+AI unlocks new levels of productivity, but you need to use consume
+with confidence. CodeGate helps you do just that, by providing a suite of
+features that make AI development safe and efficient.
+
+### Key Features
+
+- **Workspaces**: Organize and personalize your AI tooling and environments.
+- **Prompt & Alert History**: Track AI suggestions and security insights in one place.
+- **Model Muxing**: Switch seamlessly between AI models per project or route
+  specfic file types to a particular model or provider.
+- **Custom Instructions**: Tailor your AI’s behavior to match your coding style,
+  by providing custom prompts or instructions per project or workspace.
+- **Prompt Database**: Leverage a growing collection of developer-contributed
+  prompts or create and store your own, to make them available across all your
+  projects and workspaces.
+
+🔒 Built-in Security, Zero Effort
 
-- 🔒 Preventing accidental exposure of secrets and sensitive data
-- 🛡️ Ensuring AI suggestions follow secure coding practices
-- ⚠️ Blocking recommendations of known malicious or deprecated libraries
-- 🔍 Providing real-time security analysis of AI suggestions
+- **Secrets Protection**: Prevent sensitive data from leaking to AI cloud
+  service providers.
+- **Malicious Package Detection**: Block risky dependencies before they reach
+  your code, using Stackloks free AI / ML inteligence threat detection pipeline.
+- **Command Execution Monitoring**: Stop AI-generated shell commands from running
+  unsafe actions within your agent or coding assistant.
+- **PII Protection**: Prevent personally identifiable information from being
+  exposed to AI cloud service providers, such as credit card numbers,
+  social security numbers, and more.
 
 ---
+
 ## 🚀 Quickstart
 
 ### Prerequisites
@@ -85,8 +106,11 @@ documentation.
 
 CodeGate includes a web dashboard that provides:
 
-- A view of **security risks** detected by CodeGate
-- A **history of interactions** between your AI coding assistant and your LLM
+- **Manage workspaces** and AI model / provider routing rules.
+- **Track security risks** detected by CodeGate.
+- **Manage project prompts** and apply them across all your projects and  
+  workspaces.
+- **History of interactions** between your AI coding assistant and your LLM.
 
 <picture>
   <source media="(prefers-color-scheme: dark)" srcset="./static/dashboard-dark.webp">
@@ -104,6 +128,48 @@ To learn more, visit the
 ---
 ## 🔐 Features
 
+### Workspaces
+
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="./static/workspace-light.png">
+  <img alt="CodeGate logo" src="./static/workspace-dark.png" width="800px" style="max-width: 100%;">
+</picture>
+
+Workspaces are a way to organize your AI tooling and environments. You can
+create multiple workspaces to switch between them as needed. 
+
+Workspaces can be used to:
+
+- Switch between different AI models or providers.
+- Apply different prompts or instructions to different projects.
+- Isolate projects by applying different settings or rules.
+
+Workspaces are then available within all the different AI coding assistants
+and tools that CodeGate supports (i.e. Aider, Cline, Continue, Copilot, Open-Interpreter, etc.)
+
+### Prompt Database
+
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="./static/prompts-light.png">
+  <img alt="CodeGate logo" src="./static/prompts-dark.png" width="800px" style="max-width: 100%;">
+</picture>
+
+The Prompt Database is a collection of prompts that you can use across all your
+projects and workspaces. You can also contribute your own prompts to the
+database.
+
+### Model Muxing
+
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="./static/mux-light.png">
+  <img alt="CodeGate logo" src="./static/mux-dark.png" width="800px" style="max-width: 100%;">
+</picture>
+
+
+Model Muxing is a feature that allows you to route different workspaces or
+to different AI models or providers, even down the level of a single file.
+
+
 ### Secrets encryption
 
 CodeGate helps you protect sensitive information from being accidentally exposed

diff --git a/static/mux-dark.png b/static/mux-dark.png
diff --git a/static/mux-light.png b/static/mux-light.png
diff --git a/static/prompts-dark.png b/static/prompts-dark.png
diff --git a/static/prompts-light.png b/static/prompts-light.png
diff --git a/static/workspace-dark.png b/static/workspace-dark.png
diff --git a/static/workspace-light.png b/static/workspace-light.png