feat: enhance user message styling and improve session banner rendering

Author: vinhnx

docs/project/TODO.md

@@ -280,19 +280,3 @@ The project aims for clean, maintainable, and testable code through these archit
 ---
 
 explore and use parking_lot for better performance in multithreaded scenarios. Replace standard library synchronization primitives (like Mutex and RwLock) with their parking_lot counterparts where appropriate. Ensure that the transition maintains thread safety and correctness while improving performance. Update documentation to reflect the use of parking_lot and any changes in synchronization behavior.
-
----
-
--   `/command <program> [args...]` — run a shell command via `run_terminal_cmd`
-    --> also add "!" to toggle command and normal mode.
-
-press ! -> command mode
-press ! again -> normal mode.
-
-note:
-
--   on command mode only execute `run_terminal_cmd` only no tool call
--   on normal mode, full chat agent loop with tools call
--   on command mode, update the info and visual of the repl
-
----

src/agent/runloop/gemini.rs

@@ -96,8 +96,8 @@ pub(crate) async fn run_single_agent_loop_gemini(
             }
             placeholder_shown = true;
         }
-        let prompt_style = theme::active_styles().primary;
-        renderer.inline_with_style(prompt_style, "❯ ")?;
+        let styles = theme::active_styles();
+        renderer.inline_with_style(styles.primary, "❯ ")?;
         let mut input = String::new();
         io::stdin().read_line(&mut input)?;
         let input = input.trim();
@@ -179,6 +179,8 @@ pub(crate) async fn run_single_agent_loop_gemini(
             }
         }
 
+        renderer.line(MessageStyle::User, input)?;
+
         conversation_history.push(Content::user_text(input));
         let _pruned_tools = prune_gemini_tool_responses(
             &mut conversation_history,

src/agent/runloop/ui.rs

@@ -1,5 +1,6 @@
 use anyhow::Result;
 use chrono::Local;
+use pathdiff::diff_paths;
 use sysinfo::System;
 use vtagent_core::config::types::AgentConfig as CoreAgentConfig;
 use vtagent_core::tool_policy::{ToolPolicy, ToolPolicyManager};
@@ -13,36 +14,19 @@ pub(crate) fn render_session_banner(
     config: &CoreAgentConfig,
     session_bootstrap: &SessionBootstrap,
 ) -> Result<()> {
-    const VT_ASCII: &[&str] = &[
-        r"__      _______          _____          _           ",
-        r"\ \    / /__   __|        / ____|        | |          ",
-        r" \ \  / /   | |   ______ | |     ___   __| | ___  ___ ",
-        r"  \ \/ /    | |  |______|| |    / _ \ / _` |/ _ \/ __|",
-        r"   \  /     | |          | |___| (_) | (_| |  __/\__ \",
-        r"    \/      |_|           \_____/\___/ \__,_|\___||___/",
-    ];
+    let styles = theme::active_styles();
+    renderer.line_with_style(styles.primary, "Welcome to VT Code!")?;
 
-    for line in VT_ASCII {
-        renderer.line(MessageStyle::Info, line)?;
-    }
-    renderer.line(MessageStyle::Output, "")?;
-
-    renderer.line(MessageStyle::Info, "Interactive chat (tools)")?;
-    renderer.line(MessageStyle::Output, &format!("Model: {}", config.model))?;
-    renderer.line(
-        MessageStyle::Output,
-        &format!("Workspace: {}", config.workspace.display()),
-    )?;
-    renderer.line(
-        MessageStyle::Output,
-        &format!("Theme: {}", theme::active_theme_label()),
-    )?;
+    let mut bullets = Vec::new();
+    bullets.push(format!("- Model: {}", config.model));
+    bullets.push(format!("- Workspace: {}", config.workspace.display()));
+    bullets.push(format!("- Theme: {}", theme::active_theme_label()));
 
     let now = Local::now();
-    renderer.line(
-        MessageStyle::Output,
-        &format!("Local time: {}", now.format("%Y-%m-%d %H:%M:%S %Z")),
-    )?;
+    bullets.push(format!(
+        "- Local time: {}",
+        now.format("%Y-%m-%d %H:%M:%S %Z")
+    ));
 
     let mut sys = System::new_all();
     sys.refresh_all();
@@ -55,17 +39,11 @@ pub(crate) fn render_session_banner(
         .max(1);
     let total_mem_gb = sys.total_memory() as f64 / 1024.0 / 1024.0;
     let used_mem_gb = sys.used_memory() as f64 / 1024.0 / 1024.0;
-    renderer.line(
-        MessageStyle::Output,
-        &format!("System: {} (kernel {})", os_label, kernel),
-    )?;
-    renderer.line(
-        MessageStyle::Output,
-        &format!(
-            "Resources: {:.1} GB used / {:.1} GB total, {} cores",
-            used_mem_gb, total_mem_gb, cpu_count
-        ),
-    )?;
+    bullets.push(format!("- System: {} · kernel {}", os_label, kernel));
+    bullets.push(format!(
+        "- Resources: {:.1}/{:.1} GB RAM · {} cores",
+        used_mem_gb, total_mem_gb, cpu_count
+    ));
 
     match ToolPolicyManager::new_with_workspace(&config.workspace) {
         Ok(manager) => {
@@ -80,28 +58,25 @@ pub(crate) fn render_session_banner(
                     ToolPolicy::Deny => deny += 1,
                 }
             }
-            let policy_line = format!(
-                "Tool policy: allow {}, prompt {}, deny {} ({})",
-                allow,
-                prompt,
-                deny,
-                manager.config_path().display()
-            );
-            renderer.line(MessageStyle::Output, &policy_line)?;
+            let policy_path = diff_paths(manager.config_path(), &config.workspace)
+                .and_then(|p| p.to_str().map(|s| s.to_string()))
+                .unwrap_or_else(|| manager.config_path().display().to_string());
+            bullets.push(format!(
+                "- Tool policy: allow {} · prompt {} · deny {} ({})",
+                allow, prompt, deny, policy_path
+            ));
         }
         Err(err) => {
-            renderer.line(
-                MessageStyle::Error,
-                &format!("Tool policy unavailable: {}", err),
-            )?;
+            bullets.push(format!("- Tool policy: unavailable ({})", err));
         }
     }
 
     if let Some(summary) = session_bootstrap.language_summary.as_deref() {
-        renderer.line(
-            MessageStyle::Output,
-            &format!("Detected languages: {}", summary),
-        )?;
+        bullets.push(format!("- Languages: {}", summary));
+    }
+
+    for line in bullets {
+        renderer.line(MessageStyle::Output, &line)?;
     }
 
     renderer.line(MessageStyle::Output, "")?;

src/agent/runloop/unified.rs

@@ -90,12 +90,16 @@ pub(crate) async fn run_single_agent_loop_unified(
             }
             placeholder_shown = true;
         }
-        let prompt_style = theme::active_styles().primary;
-        renderer.inline_with_style(prompt_style, "❯ ")?;
+        let styles = theme::active_styles();
+        renderer.inline_with_style(styles.primary, "❯ ")?;
         let mut input = String::new();
         io::stdin().read_line(&mut input)?;
         let input = input.trim();
 
+        if input.is_empty() {
+            continue;
+        }
+
         match input {
             "" => continue,
             "exit" | "quit" => {
@@ -173,6 +177,8 @@ pub(crate) async fn run_single_agent_loop_unified(
             }
         }
 
+        renderer.line(MessageStyle::User, input)?;
+
         let refined_user = refine_user_prompt_if_enabled(input, config, vt_cfg).await;
         conversation_history.push(uni::Message::user(refined_user));
         let _pruned_tools = prune_unified_tool_responses(

vtagent-core/src/ui/theme.rs

@@ -62,6 +62,12 @@ impl ThemePalette {
             MIN_CONTRAST,
             &[lighten(text_color, 0.15), fallback_light],
         );
+        let user_color = ensure_contrast(
+            lighten(primary, 0.25),
+            background,
+            MIN_CONTRAST,
+            &[lighten(secondary, 0.15), info_color, text_color],
+        );
         let alert_color = ensure_contrast(
             self.alert,
             background,
@@ -75,6 +81,7 @@ impl ThemePalette {
             output: Self::style_from(text_color, false),
             response: Self::style_from(response_color, false),
             tool: Self::style_from(tool_color, true),
+            user: Self::style_from(user_color, false),
             primary: Self::style_from(primary, false),
             secondary: Self::style_from(secondary, false),
             background: Color::Rgb(background),
@@ -91,6 +98,7 @@ pub struct ThemeStyles {
     pub output: Style,
     pub response: Style,
     pub tool: Style,
+    pub user: Style,
     pub primary: Style,
     pub secondary: Style,
     pub background: Color,

vtagent-core/src/utils/ansi.rs

@@ -13,6 +13,7 @@ pub enum MessageStyle {
     Output,
     Response,
     Tool,
+    User,
 }
 
 impl MessageStyle {
@@ -24,6 +25,7 @@ impl MessageStyle {
             Self::Output => styles.output,
             Self::Response => styles.response,
             Self::Tool => styles.tool,
+            Self::User => styles.user,
         }
     }