Enhance TUI with modern design and improved status rendering

- Updated status bar rendering in `status.rs` to feature a segmented design with clear visual segments for app name, status, and metadata.
- Refined color theme in `theme.rs` to adopt a modern dark palette with neon accents, improving visual hierarchy and readability.
- Revamped workflow panel in `workflow.rs` to display agent statuses as compact cards with state badges, enhancing clarity and user experience.
- Improved overall styling consistency across components, ensuring a cohesive look and feel throughout the TUI.
This commit is contained in:
asepharyana
2026-07-13 05:42:17 +07:00
parent 2310c2df7f
commit 3f5f27c339
6 changed files with 977 additions and 554 deletions
+141 -100
View File
@@ -1,22 +1,22 @@
//! Workflow status panel rendering.
//! Workflow status panel rendering — agent cards with state badges.
//!
//! Flow: `draw_workflow_panel` reads `state.workflow_engine` and renders a
//! rich panel showing agent statuses, findings count, session counters,
//! and usage hints.
//! panel showing agent statuses, findings count, session counters, and
//! usage hints.
//!
//! Division-aware: when the workflow is a company pipeline, shows the
//! division pipeline header with visual arrows between stages.
//! Design: agents are shown as compact cards with state-colored badges.
//! The division pipeline mode adds a visual pipeline flow with arrows.
use ratatui::layout::Rect;
use ratatui::style::{Style, Modifier};
use ratatui::text::{Line, Span};
use ratatui::widgets::{Block, Borders, List, ListItem, Paragraph, Wrap};
use ratatui::widgets::{Block, Borders, Paragraph, Wrap};
use ratatui::Frame;
use super::theme::Theme;
use crate::app::workflow::engine::AgentState;
/// Icons for division states in the company pipeline.
fn div_icon(state: AgentState) -> &'static str {
/// Icons for agent states.
fn state_icon(state: AgentState) -> &'static str {
match state {
AgentState::Idle => "",
AgentState::Running => "",
@@ -25,13 +25,29 @@ fn div_icon(state: AgentState) -> &'static str {
}
}
/// Detect if the current workflow looks like a company pipeline by
/// checking agent names for division keywords.
fn state_label(state: AgentState) -> &'static str {
match state {
AgentState::Idle => "Idle",
AgentState::Running => "Running",
AgentState::Completed => "Done",
AgentState::Failed => "Failed",
}
}
fn state_color(state: AgentState) -> Color {
match state {
AgentState::Idle => Theme::TEXT_DIM,
AgentState::Running => Theme::WARNING,
AgentState::Completed => Theme::SUCCESS,
AgentState::Failed => Theme::ERROR,
}
}
/// Detect if the current workflow looks like a company pipeline.
fn is_company_pipeline(agents: &[crate::app::workflow::engine::WorkflowAgent]) -> bool {
if agents.is_empty() {
return false;
}
// Company pipeline agents have names like "Strategy", "Engineering", etc.
let division_keywords = ["Strategy", "Engineering", "Quality", "Security", "Documentation"];
agents.iter().any(|a| {
division_keywords.iter().any(|k| a.name.contains(k))
@@ -44,77 +60,63 @@ pub fn draw_workflow_panel(frame: &mut Frame, area: Rect, state: &crate::app::st
let is_company = is_company_pipeline(&state.workflow_engine.agents);
let title = if is_company {
Span::styled(" 🏢 Pipeline ", Style::default().fg(Theme::PRIMARY).add_modifier(Modifier::BOLD))
} else {
Span::styled(" ⚙ Workflow ", Style::default().fg(Theme::PRIMARY).add_modifier(Modifier::BOLD))
};
let block = Block::default()
.borders(Borders::ALL)
.border_style(Style::default().fg(Theme::PRIMARY))
.title({
if is_company {
Span::styled(" 🏢 Company Pipeline ", Style::default().fg(Theme::PRIMARY).add_modifier(Modifier::BOLD))
} else {
Span::styled(" ⚙ Workflow ", Style::default().fg(Theme::PRIMARY).add_modifier(Modifier::BOLD))
}
});
.border_style(Style::default().fg(Theme::BORDER))
.title(title);
let inner = block.inner(area);
frame.render_widget(block, area);
// Split inner into header (hints) and body (agent list / status)
// Split inner into header and body
let chunks = Layout::default()
.direction(Direction::Vertical)
.constraints([
Constraint::Length(3), // header / hints
Constraint::Min(4), // agent list or placeholder
Constraint::Length(3),
Constraint::Min(4),
])
.split(inner);
// ── Header ─────────────────────────────────────────────────────────
let mut header_lines = vec![];
// ── Header area ──────────────────────────────────────────────────────
let mut header_lines: Vec<Line> = Vec::new();
if is_company {
// Show the division pipeline header with visual arrows
// Division pipeline overview
let agents = &state.workflow_engine.agents;
let mut pipeline_spans: Vec<Span> = Vec::new();
let mut pipe_spans: Vec<Span> = Vec::new();
for (i, agent) in agents.iter().enumerate() {
if i > 0 {
pipeline_spans.push(Span::styled("", Style::default().fg(Theme::DIM)));
pipe_spans.push(Span::styled(
" ",
Style::default().fg(Theme::TEXT_DIM),
));
}
let icon = div_icon(agent.status.state);
let (color, modif) = match agent.status.state {
AgentState::Idle => (Theme::DIM, Modifier::empty()),
AgentState::Running => (Theme::WARNING, Modifier::BOLD),
AgentState::Completed => (Theme::SUCCESS, Modifier::BOLD),
AgentState::Failed => (Theme::ERROR, Modifier::BOLD),
let icon = state_icon(agent.status.state);
let color = state_color(agent.status.state);
let modif = match agent.status.state {
AgentState::Idle => Modifier::empty(),
_ => Modifier::BOLD,
};
pipeline_spans.push(Span::styled(
format!("{} {} ", icon, agent.name.chars().take(12).collect::<String>()),
pipe_spans.push(Span::styled(
format!("{} {} ", icon, agent.name.chars().take(10).collect::<String>()),
Style::default().fg(color).add_modifier(modif),
));
if i < agents.len().saturating_sub(1) {
pipe_spans.push(Span::styled(
"",
Style::default().fg(Theme::TEXT_DIM),
));
}
}
header_lines.push(Line::from(pipeline_spans));
header_lines.push(Line::from(pipe_spans));
header_lines.push(Line::from(vec![
Span::styled("Status: ", Style::default().fg(Theme::DIM)),
if state.turn_in_flight() {
Span::styled("● Pipeline Running", Style::default().fg(Theme::WARNING).add_modifier(Modifier::BOLD))
} else {
Span::styled("● Pipeline Complete", Style::default().fg(Theme::SUCCESS))
},
Span::raw(" "),
Span::styled(
format!("Divisions: {} Findings: {}",
state.workflow_engine.agents.len(),
state.workflow_engine.findings.len(),
),
Style::default().fg(Theme::DIM),
),
]));
} else {
header_lines.push(Line::from(vec![
Span::styled("/workflow run ", Style::default().fg(Theme::PRIMARY).add_modifier(Modifier::BOLD)),
Span::styled("<prompt>", Style::default().fg(Theme::DIM)),
Span::styled(" · Esc to close", Style::default().fg(Theme::DIM)),
]));
header_lines.push(Line::from(vec![
Span::styled("Status: ", Style::default().fg(Theme::DIM)),
Span::styled("Status: ", Style::default().fg(Theme::TEXT_DIM)),
if state.turn_in_flight() {
Span::styled("● Running", Style::default().fg(Theme::WARNING).add_modifier(Modifier::BOLD))
} else {
@@ -122,70 +124,107 @@ pub fn draw_workflow_panel(frame: &mut Frame, area: Rect, state: &crate::app::st
},
Span::raw(" "),
Span::styled(
format!("Agents: {} Findings: {}",
format!("Agents: {} | Findings: {}",
state.workflow_engine.agents.len(),
state.workflow_engine.findings.len(),
),
Style::default().fg(Theme::DIM),
Style::default().fg(Theme::TEXT_DIM),
),
]));
} else {
header_lines.push(Line::from(vec![
Span::styled("/workflow run ", Style::default().fg(Theme::PRIMARY).add_modifier(Modifier::BOLD)),
Span::styled("<prompt>", Style::default().fg(Theme::TEXT_DIM)),
Span::styled(" · Esc to close", Style::default().fg(Theme::TEXT_DIM)),
]));
header_lines.push(Line::from(vec![
Span::styled("Status: ", Style::default().fg(Theme::TEXT_DIM)),
if state.turn_in_flight() {
Span::styled("● Running", Style::default().fg(Theme::WARNING).add_modifier(Modifier::BOLD))
} else {
Span::styled("● Idle", Style::default().fg(Theme::SUCCESS))
},
Span::raw(" "),
Span::styled(
format!("Agents: {} | Findings: {}",
state.workflow_engine.agents.len(),
state.workflow_engine.findings.len(),
),
Style::default().fg(Theme::TEXT_DIM),
),
]));
}
let header = Paragraph::new(header_lines);
frame.render_widget(header, chunks[0]);
// ── Body: agent/division list ──────────────────────────────────────
// ── Body: agent cards ────────────────────────────────────────────────
if state.workflow_engine.agents.is_empty() {
let session_lines = build_session_lines(state);
let placeholder = Paragraph::new(session_lines).wrap(Wrap { trim: false });
frame.render_widget(placeholder, chunks[1]);
} else {
let items: Vec<ListItem> = state.workflow_engine.agents.iter().map(|agent| {
let (state_str, state_color) = match agent.status.state {
AgentState::Idle => ("○ Idle", Theme::DIM),
AgentState::Running => ("▶ Running…", Theme::WARNING),
AgentState::Completed => ("✓ Done", Theme::SUCCESS),
AgentState::Failed => ("✗ Failed", Theme::ERROR),
};
let mut card_lines: Vec<Line> = Vec::new();
for agent in &state.workflow_engine.agents {
let color = state_color(agent.status.state);
let icon = state_icon(agent.status.state);
let label = state_label(agent.status.state);
let duration_str = match (agent.status.started_at, agent.status.completed_at) {
(Some(s), Some(e)) => format!(" {}ms", e.saturating_sub(s)),
(Some(_), None) => " (running)".to_string(),
_ => String::new(),
};
ListItem::new(Line::from(vec![
Span::styled(
format!(" {:12} ", state_str),
Style::default().fg(state_color).add_modifier(Modifier::BOLD),
),
Span::styled(
format!("{}{}", agent.name, duration_str),
Style::default().fg(Theme::TEXT),
),
if let Some(ref err) = agent.status.error {
Span::styled(format!("{}", err), Style::default().fg(Theme::ERROR))
} else if let Some(ref prog) = agent.status.progress {
Span::styled(
format!(" ({})", prog),
Style::default().fg(Theme::DIM),
)
} else {
Span::raw("")
},
]))
}).collect();
let list = List::new(items)
.highlight_style(Style::default().add_modifier(Modifier::BOLD));
// Agent card header
card_lines.push(Line::from(vec![
Span::styled(
format!(" {} ", icon),
Style::default().fg(color).add_modifier(Modifier::BOLD),
),
Span::styled(
format!(" {}", agent.name),
Style::default().fg(Theme::TEXT).add_modifier(Modifier::BOLD),
),
Span::styled(
format!(" [{}]", label),
Style::default().fg(color),
),
Span::styled(
duration_str,
Style::default().fg(Theme::TEXT_DIM),
),
]));
// Agent details (progress / error)
if let Some(ref err) = agent.status.error {
card_lines.push(Line::from(vec![
Span::styled("", Style::default().fg(Theme::ERROR)),
Span::styled(err.clone(), Style::default().fg(Theme::ERROR)),
]));
} else if let Some(ref prog) = agent.status.progress {
card_lines.push(Line::from(vec![
Span::styled(" ", Style::default()),
Span::styled(prog.clone(), Style::default().fg(Theme::TEXT_DIM).add_modifier(Modifier::ITALIC)),
]));
}
// Card separator
card_lines.push(Line::from(Span::raw("")));
}
let list = Paragraph::new(card_lines);
frame.render_widget(list, chunks[1]);
}
}
/// Build a compact list of session counters for the placeholder view.
/// Build compact session info for the placeholder view.
fn build_session_lines(state: &crate::app::state::rest::AppStateRest) -> Vec<Line<'static>> {
let mut lines: Vec<Line<'static>> = Vec::new();
lines.push(Line::from(Span::styled(
" No workflow running.",
Style::default().fg(Theme::DIM),
Style::default().fg(Theme::TEXT_DIM),
)));
lines.push(Line::from(Span::raw("")));
@@ -196,37 +235,39 @@ fn build_session_lines(state: &crate::app::state::rest::AppStateRest) -> Vec<Lin
let msg_count = rt.messages.len();
lines.push(Line::from(vec![
Span::styled(" Messages ", Style::default().fg(Theme::DIM)),
Span::styled(" Messages ", Style::default().fg(Theme::TEXT_DIM)),
Span::styled(msg_count.to_string(), Style::default().fg(Theme::INFO).add_modifier(Modifier::BOLD)),
]));
lines.push(Line::from(vec![
Span::styled(" Tool calls", Style::default().fg(Theme::DIM)),
Span::styled(" Tool calls", Style::default().fg(Theme::TEXT_DIM)),
Span::styled(format!(" {}", tool_count), Style::default().fg(Theme::SUCCESS)),
]));
if pending > 0 {
lines.push(Line::from(vec![
Span::styled(" Pending ", Style::default().fg(Theme::DIM)),
Span::styled(" Pending ", Style::default().fg(Theme::TEXT_DIM)),
Span::styled(format!(" {}", pending), Style::default().fg(Theme::WARNING)),
]));
}
if bash_count > 0 {
lines.push(Line::from(vec![
Span::styled(" Bash jobs ", Style::default().fg(Theme::DIM)),
Span::styled(" Bash jobs ", Style::default().fg(Theme::TEXT_DIM)),
Span::styled(format!(" {}", bash_count), Style::default().fg(Theme::WARNING)),
]));
}
} else {
lines.push(Line::from(Span::styled(
" (no active session)",
Style::default().fg(Theme::DIM),
Style::default().fg(Theme::TEXT_DIM),
)));
}
lines.push(Line::from(Span::raw("")));
lines.push(Line::from(Span::styled(
" Complex tasks auto-delegate to the company pipeline.",
Style::default().fg(Theme::DIM).add_modifier(Modifier::ITALIC),
Style::default().fg(Theme::TEXT_DIM).add_modifier(Modifier::ITALIC),
)));
lines
}
use ratatui::style::Color;