consciousness/src/user/compare.rs

// compare.rs — F7 compare screen: side-by-side test-model regen of
// every assistant response in the current context.

use ratatui::{
    layout::Rect,
    style::{Color, Modifier, Style},
    text::{Line, Span},
    widgets::{Block, Borders, List, ListItem, ListState, Paragraph, Wrap},
    Frame,
};
use ratatui::crossterm::event::{Event, KeyCode};

use super::{App, ScreenView, truncate, widgets};

pub use crate::subconscious::compare::CompareCandidate;

pub(crate) struct CompareScreen {
    list_state: ListState,
    mind_tx: tokio::sync::mpsc::UnboundedSender<crate::mind::MindCommand>,
}

impl CompareScreen {
    pub fn new(
        mind_tx: tokio::sync::mpsc::UnboundedSender<crate::mind::MindCommand>,
    ) -> Self {
        Self { list_state: ListState::default(), mind_tx }
    }
}

impl ScreenView for CompareScreen {
    fn label(&self) -> &'static str { "compare" }

    fn tick(&mut self, frame: &mut Frame, area: Rect,
            events: &[Event], app: &mut App) {
        widgets::handle_list_nav(events, &mut self.list_state,
            app.compare_candidates.len(), |code| match code {
                KeyCode::Char('c') | KeyCode::Enter => {
                    let _ = self.mind_tx.send(crate::mind::MindCommand::Compare);
                }
                _ => {}
            });

        let (settings_area, content_area, help_area) =
            widgets::candidate_frame(frame, area, "compare");

        let test_backend = crate::config::app().compare.test_backend.clone();
        let (label, color) = if test_backend.is_empty() {
            ("(unset — set compare.test_backend)".to_string(), Color::Red)
        } else {
            (test_backend, Color::Yellow)
        };
        frame.render_widget(Paragraph::new(Line::from(vec![
            Span::raw(" test model: "),
            Span::styled(label, Style::default().fg(color)),
        ])), settings_area);

        let candidates = &app.compare_candidates;
        if candidates.is_empty() {
            let err = app.mind_state.as_ref().and_then(|ms| ms.compare_error.as_deref());
            let mut lines = vec![Line::from(""),
                Line::styled("  Press c/Enter to compare against the configured test model.",
                    Style::default().fg(Color::DarkGray))];
            if let Some(e) = err {
                lines.push(Line::from(""));
                lines.push(Line::from(vec![
                    Span::raw("  "),
                    Span::styled(format!("error: {}", e), Style::default().fg(Color::Red)),
                ]));
            }
            frame.render_widget(Paragraph::new(lines), content_area);
        } else {
            let (list_area, detail_area) = widgets::list_detail_split(content_area);

            let items: Vec<ListItem> = candidates.iter().map(|c| ListItem::new(Line::from(vec![
                Span::styled(format!("#{:<3} ", c.entry_idx), Style::default().fg(Color::DarkGray)),
                Span::raw(truncate(&c.original_text, 30)),
            ]))).collect();
            frame.render_stateful_widget(
                List::new(items)
                    .block(Block::default().borders(Borders::RIGHT).title(" candidates "))
                    .highlight_style(Style::default().add_modifier(Modifier::REVERSED)),
                list_area, &mut self.list_state,
            );

            if let Some(c) = self.list_state.selected().and_then(|i| candidates.get(i)) {
                let mut text = String::new();
                if !c.prior_context.is_empty() {
                    text.push_str(&c.prior_context);
                    text.push_str("\n\n─── original ───\n\n");
                }
                text.push_str(&c.original_text);
                text.push_str("\n\n─── test model ───\n\n");
                text.push_str(&c.alternate_text);
                frame.render_widget(
                    Paragraph::new(text)
                        .block(Block::default().borders(Borders::TOP)
                            .title(format!(" entry {} ", c.entry_idx)))
                        .wrap(Wrap { trim: false }),
                    detail_area,
                );
            }
        }

        frame.render_widget(Paragraph::new(Line::from(vec![
            Span::styled(" j/k/\u{2191}\u{2193}", Style::default().fg(Color::Cyan)),
            Span::raw("=nav  "),
            Span::styled("c/Enter", Style::default().fg(Color::Green)),
            Span::raw("=run "),
        ])), help_area);
    }
}