uusi projekti

network-poc/native-node/src/tui_dashboard.rs

@@ -35,6 +35,10 @@ pub struct DashboardState {
     pub last_tokens_sec: f64,
     pub network_active_nodes: usize,
     pub network_total_tasks: u64,
+    // Mallivalikko
+    pub model_picker_open: bool,
+    pub model_picker_items: Vec<String>,
+    pub model_picker_idx: usize,
 }
 
 impl DashboardState {
@@ -51,6 +55,9 @@ impl DashboardState {
             last_tokens_sec: 0.0,
             network_active_nodes: 1, // oletetaan itsemme
             network_total_tasks: 0,
+            model_picker_open: false,
+            model_picker_items: Vec::new(),
+            model_picker_idx: 0,
         }
     }
 
@@ -88,20 +95,53 @@ pub async fn run_dashboard(
             }
             ev = reader.next() => {
                 if let Some(Ok(Event::Key(key))) = ev {
-                    match key.code {
-                        KeyCode::Char('q') | KeyCode::Esc => {
-                            // Palautetaan näyttö ja suljetaan ohjelma
-                            disable_raw_mode()?;
-                            execute!(terminal.backend_mut(), LeaveAlternateScreen)?;
-                            std::process::exit(0);
+                    let picker_open = state.read().await.model_picker_open;
+
+                    if picker_open {
+                        // Mallivalikko auki — navigointi
+                        match key.code {
+                            KeyCode::Up | KeyCode::Char('k') => {
+                                let mut st = state.write().await;
+                                if st.model_picker_idx > 0 { st.model_picker_idx -= 1; }
+                            }
+                            KeyCode::Down | KeyCode::Char('j') => {
+                                let mut st = state.write().await;
+                                let max = st.model_picker_items.len().saturating_sub(1);
+                                if st.model_picker_idx < max { st.model_picker_idx += 1; }
+                            }
+                            KeyCode::Enter => {
+                                let mut st = state.write().await;
+                                let idx = st.model_picker_idx;
+                                if let Some(model) = st.model_picker_items.get(idx).cloned() {
+                                    st.model_picker_open = false;
+                                    st.push_log("System", format!("Vaihdetaan malliin: {}...", model), None);
+                                    let _ = cmd_tx.send(format!("change_model:{}", model));
+                                }
+                            }
+                            KeyCode::Esc | KeyCode::Char('m') | KeyCode::Char('M') => {
+                                state.write().await.model_picker_open = false;
+                            }
+                            _ => {}
                         }
-                        KeyCode::Char('p') | KeyCode::Char('P') => {
-                            let _ = cmd_tx.send("pause".to_string());
+                    } else {
+                        // Normaali tila
+                        match key.code {
+                            KeyCode::Char('q') | KeyCode::Esc => {
+                                disable_raw_mode()?;
+                                execute!(terminal.backend_mut(), LeaveAlternateScreen)?;
+                                std::process::exit(0);
+                            }
+                            KeyCode::Char('p') | KeyCode::Char('P') => {
+                                let _ = cmd_tx.send("pause".to_string());
+                            }
+                            KeyCode::Char('r') | KeyCode::Char('R') | KeyCode::Char('s') => {
+                                let _ = cmd_tx.send("resume".to_string());
+                            }
+                            KeyCode::Char('m') | KeyCode::Char('M') => {
+                                let _ = cmd_tx.send("fetch_models".to_string());
+                            }
+                            _ => {}
                         }
-                        KeyCode::Char('r') | KeyCode::Char('R') | KeyCode::Char('s') => {
-                            let _ = cmd_tx.send("resume".to_string());
-                        }
-                        _ => {}
                     }
                 }
             }
@@ -214,10 +254,43 @@ fn ui(f: &mut ratatui::Frame, st: &DashboardState) {
 
     // --- Footer / Status ---
     let status_color = if st.status == "ACTIVE" { Color::Green } else { Color::Yellow };
-    let status_text = format!(" Tila: {} | Komennot: [P] Pause / [R] Työhön / [Q] Sulje ", st.status);
+    let status_text = format!(" Tila: {} | [P] Pause [R] Työhön [M] Malli [Q] Sulje ", st.status);
     let footer = Paragraph::new(status_text)
         .style(Style::default().fg(status_color).add_modifier(Modifier::BOLD))
         .alignment(Alignment::Center)
         .block(Block::default().borders(Borders::ALL));
     f.render_widget(footer, chunks[2]);
+
+    // --- Mallivalikko-overlay ---
+    if st.model_picker_open && !st.model_picker_items.is_empty() {
+        let area = f.area();
+        let popup_h = (st.model_picker_items.len() as u16 + 4).min(area.height - 4);
+        let popup_w = 50.min(area.width - 4);
+        let popup = ratatui::layout::Rect::new(
+            (area.width - popup_w) / 2,
+            (area.height - popup_h) / 2,
+            popup_w,
+            popup_h,
+        );
+
+        // Tausta
+        f.render_widget(ratatui::widgets::Clear, popup);
+
+        let items: Vec<ratatui::text::Line> = st.model_picker_items.iter().enumerate().map(|(i, name)| {
+            if i == st.model_picker_idx {
+                ratatui::text::Line::from(format!(" ▸ {} ", name))
+                    .style(Style::default().fg(Color::Cyan).add_modifier(Modifier::BOLD))
+            } else {
+                ratatui::text::Line::from(format!("   {} ", name))
+                    .style(Style::default().fg(Color::White))
+            }
+        }).collect();
+
+        let picker = Paragraph::new(items)
+            .block(Block::default()
+                .title(" Vaihda malli [↑↓] Enter=valitse Esc=peruuta ")
+                .borders(Borders::ALL)
+                .style(Style::default().fg(Color::Cyan)));
+        f.render_widget(picker, popup);
+    }
 }