extract-tracker 0.2.3__tar.gz → 0.3.0__tar.gz

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: extract-tracker
-Version: 0.2.3
+Version: 0.3.0
 Summary: Local-first experiment tracking for deep learning
 Author: Phil Oh
 Keywords: experiment-tracking,machine-learning,deep-learning,sqlite,tui,mcp

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "extract-tracker"
-version = "0.2.3"
+version = "0.3.0"
 description = "Local-first experiment tracking for deep learning"
 readme = "README.md"
 requires-python = ">=3.10"

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/python/src/extract_tracker.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: extract-tracker
-Version: 0.2.3
+Version: 0.3.0
 Summary: Local-first experiment tracking for deep learning
 Author: Phil Oh
 Keywords: experiment-tracking,machine-learning,deep-learning,sqlite,tui,mcp

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/rust/Cargo.lock

@@ -592,7 +592,7 @@ dependencies = [
 
 [[package]]
 name = "extract-tui"
-version = "0.2.3"
+version = "0.3.0"
 dependencies = [
  "chrono",
  "clap",

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/rust/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "extract-tui"
-version = "0.2.3"
+version = "0.3.0"
 edition = "2024"
 
 [dependencies]

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/rust/src/app.rs

@@ -306,6 +306,7 @@ pub struct RunBrowserState {
     pub filtered: Vec<usize>,
     pub cursor: usize,
     pub search_query: Option<String>,
+    pub rename_buffer: Option<String>,
     pub scroll_offset: usize,
 }
 
@@ -318,6 +319,7 @@ impl RunBrowserState {
             filtered,
             cursor: 0,
             search_query: None,
+            rename_buffer: None,
             scroll_offset: 0,
         }
     }

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/rust/src/db.rs

@@ -854,6 +854,19 @@ impl Db {
         Ok(())
     }
 
+    /// Rename a run. Empty string clears the name. Opens a writable connection.
+    pub fn rename_run(db_path: &Path, id: &str, name: &str) -> Result<()> {
+        let conn = Connection::open(db_path)?;
+        conn.execute_batch("PRAGMA journal_mode=WAL; PRAGMA foreign_keys=ON;")?;
+        let trimmed = name.trim();
+        let value: Option<&str> = if trimmed.is_empty() { None } else { Some(trimmed) };
+        conn.execute(
+            "UPDATE runs SET name = ? WHERE id = ?",
+            params![value, id],
+        )?;
+        Ok(())
+    }
+
     /// Set status on a single run or experiment. Opens a writable connection.
     pub fn set_status(db_path: &Path, table: &str, id: &str, status: &str) -> Result<()> {
         assert!(table == "runs" || table == "experiments");
@@ -1314,6 +1327,18 @@ mod tests {
         assert_eq!(run.status, "failed");
     }
 
+    #[test]
+    fn test_rename_run() {
+        let tdb = test_db_with_path();
+        Db::rename_run(&tdb.path, "r1", "renamed").unwrap();
+        let run = tdb.db.get_run("r1").unwrap().unwrap();
+        assert_eq!(run.name.as_deref(), Some("renamed"));
+
+        Db::rename_run(&tdb.path, "r1", "   ").unwrap();
+        let run = tdb.db.get_run("r1").unwrap().unwrap();
+        assert!(run.name.is_none());
+    }
+
     #[test]
     fn test_archive_experiment() {
         let tdb = test_db_with_path();

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/rust/src/ui/layout.rs

@@ -82,7 +82,14 @@ impl AppLayout {
         // focused panel's input handler see every keystroke unmodified.
         let in_text_input = state.tag_picker.is_some()
             || state.note_input.is_some()
-            || state.todo_input.is_some();
+            || state.todo_input.is_some()
+            || state
+                .run_picker
+                .as_ref()
+                .is_some_and(|picker| picker.search_query.is_some())
+            || state.run_browser.as_ref().is_some_and(|browser| {
+                browser.search_query.is_some() || browser.rename_buffer.is_some()
+            });
 
         if !in_text_input {
             // Global keys: gg/G, ?, work in all views

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/rust/src/ui/popup.rs

@@ -1,4 +1,5 @@
-use crossterm::event::{KeyCode, KeyEvent};
+use chrono::{DateTime, Local};
+use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
 use ratatui::layout::Rect;
 use ratatui::style::{Modifier, Style};
 use ratatui::symbols::border;
@@ -216,6 +217,70 @@ impl PopupRenderer {
         };
 
         let is_searching = browser.search_query.is_some();
+        let is_renaming = browser.rename_buffer.is_some();
+
+        // Rename mode
+        if is_renaming {
+            match key.code {
+                KeyCode::Esc => {
+                    browser.rename_buffer = None;
+                    return false;
+                }
+                KeyCode::Enter => {
+                    if let Some(&run_idx) = browser.filtered.get(browser.cursor) {
+                        if let Some(run) = browser.runs.get(run_idx) {
+                            let run_id = run.id.clone();
+                            let new_name = browser.rename_buffer.take().unwrap_or_default();
+                            let db_path = state.store_root.join("extract.db");
+                            match crate::db::Db::rename_run(&db_path, &run_id, &new_name) {
+                                Ok(()) => {
+                                    let trimmed = new_name.trim();
+                                    let value = if trimmed.is_empty() {
+                                        None
+                                    } else {
+                                        Some(trimmed.to_string())
+                                    };
+                                    if let Some(run) = browser.runs.get_mut(run_idx) {
+                                        run.name = value.clone();
+                                    }
+                                    if let Some(state_idx) = state.runs.iter().position(|r| r.id == run_id) {
+                                        state.runs[state_idx].name = value;
+                                    }
+                                    let _ = state.refresh_selection_summary();
+                                    state.notify(crate::app::NotifyLevel::Success, "Run renamed");
+                                }
+                                Err(err) => {
+                                    state.notify(
+                                        crate::app::NotifyLevel::Error,
+                                        format!("Rename failed: {err}"),
+                                    );
+                                }
+                            }
+                        } else {
+                            browser.rename_buffer = None;
+                        }
+                    } else {
+                        browser.rename_buffer = None;
+                    }
+                    return false;
+                }
+                KeyCode::Backspace => {
+                    if let Some(ref mut name) = browser.rename_buffer {
+                        name.pop();
+                    }
+                    return false;
+                }
+                KeyCode::Char(c) => {
+                    if accepts_text_modifiers(key) {
+                        if let Some(ref mut name) = browser.rename_buffer {
+                            name.push(c);
+                        }
+                    }
+                    return false;
+                }
+                _ => return false,
+            }
+        }
 
         // Search mode
         if is_searching {
@@ -290,6 +355,15 @@ impl PopupRenderer {
             return false;
         }
 
+        if is_rename_key(key) {
+            if let Some(&run_idx) = browser.filtered.get(browser.cursor) {
+                if let Some(run) = browser.runs.get(run_idx) {
+                    browser.rename_buffer = Some(run.name.clone().unwrap_or_default());
+                }
+            }
+            return false;
+        }
+
         if keys::matches(key, keys::SELECT) {
             if let Some(&filtered_idx) = browser.filtered.get(browser.cursor) {
                 if let Some(run) = browser.runs.get(filtered_idx) {
@@ -499,6 +573,7 @@ impl PopupRenderer {
         browser: &mut RunBrowserState,
     ) {
         let is_searching = browser.search_query.is_some();
+        let is_renaming = browser.rename_buffer.is_some();
         let width = POPUP_WIDTH.min(area.width.saturating_sub(4));
         let height = POPUP_HEIGHT.min(area.height.saturating_sub(4));
         let popup_area = centered_rect(width, height, area);
@@ -508,12 +583,16 @@ impl PopupRenderer {
         let title = format!(" {} — runs ", browser.experiment_name);
         let footer_spans = if is_searching {
             search_footer_spans(&self.theme)
+        } else if is_renaming {
+            rename_footer_spans(&self.theme)
         } else {
             vec![
                 Span::styled("j/k", Style::default().fg(self.theme.accent)),
                 Span::styled(" nav  ", Style::default().fg(self.theme.accent_dim)),
                 Span::styled("Enter", Style::default().fg(self.theme.accent)),
                 Span::styled(" select  ", Style::default().fg(self.theme.accent_dim)),
+                Span::styled("R", Style::default().fg(self.theme.accent)),
+                Span::styled(" rename  ", Style::default().fg(self.theme.accent_dim)),
                 Span::styled("/", Style::default().fg(self.theme.accent)),
                 Span::styled(" search  ", Style::default().fg(self.theme.accent_dim)),
                 Span::styled("x", Style::default().fg(self.theme.accent)),
@@ -570,11 +649,19 @@ impl PopupRenderer {
                 Style::default()
             };
 
-            let label = run
-                .name
-                .as_deref()
-                .map(|n| format!("{} ", n))
-                .unwrap_or_default();
+            let label = if is_cursor {
+                browser
+                    .rename_buffer
+                    .as_ref()
+                    .map(|name| format!("{name}_ "))
+                    .or_else(|| run.name.as_deref().map(|n| format!("{} ", n)))
+                    .unwrap_or_default()
+            } else {
+                run.name
+                    .as_deref()
+                    .map(|n| format!("{} ", n))
+                    .unwrap_or_default()
+            };
             let date = format_date(run);
             let config_summary = differing_config_summary(run, &diff_keys);
 
@@ -624,14 +711,24 @@ fn compute_scroll(cursor: usize, current_offset: usize, list_height: usize) -> u
     }
 }
 
-/// Format a run's date for display (first 19 chars of ended_at or started_at).
+/// Format a run timestamp for display in the user's local timezone.
 fn format_date(run: &Run) -> String {
-    run.ended_at
-        .as_deref()
-        .unwrap_or(&run.started_at)
-        .chars()
-        .take(19)
-        .collect()
+    let raw = run.ended_at.as_deref().unwrap_or(&run.started_at);
+    format_local_timestamp(raw).unwrap_or_else(|| raw.chars().take(19).collect())
+}
+
+fn format_local_timestamp(raw: &str) -> Option<String> {
+    DateTime::parse_from_rfc3339(raw)
+        .ok()
+        .map(|dt| dt.with_timezone(&Local).format("%Y-%m-%d %H:%M:%S").to_string())
+}
+
+fn is_rename_key(key: &KeyEvent) -> bool {
+    matches!(key.code, KeyCode::Char('R')) && accepts_text_modifiers(key)
+}
+
+fn accepts_text_modifiers(key: &KeyEvent) -> bool {
+    key.modifiers == KeyModifiers::NONE || key.modifiers == KeyModifiers::SHIFT
 }
 
 /// Build search input line with blinking cursor.
@@ -655,6 +752,17 @@ fn search_footer_spans(theme: &Theme) -> Vec<Span<'static>> {
     ]
 }
 
+fn rename_footer_spans(theme: &Theme) -> Vec<Span<'static>> {
+    vec![
+        Span::styled("Type", Style::default().fg(theme.accent)),
+        Span::styled(" name  ", Style::default().fg(theme.accent_dim)),
+        Span::styled("Enter", Style::default().fg(theme.accent)),
+        Span::styled(" save  ", Style::default().fg(theme.accent_dim)),
+        Span::styled("Esc", Style::default().fg(theme.accent)),
+        Span::styled(" cancel", Style::default().fg(theme.accent_dim)),
+    ]
+}
+
 /// Compare configs across all runs and return only the keys whose values differ.
 fn differing_config_keys(runs: &[Run]) -> Vec<String> {
     use std::collections::HashMap;

{extract_tracker-0.2.3 → extract_tracker-0.3.0}/rust/src/ui/summary.rs

@@ -1,3 +1,4 @@
+use chrono::{DateTime, Local};
 use ratatui::buffer::Buffer;
 use ratatui::layout::Rect;
 use ratatui::style::{Color, Modifier, Style};
@@ -219,7 +220,7 @@ impl SummaryRenderer {
             }
 
             let run = &data.runs[i];
-            let date = run.started_at.get(..10).unwrap_or(&run.started_at);
+            let date = format_run_time(run);
             let label = run.name.clone().unwrap_or_else(|| {
                 let id = &run.id;
                 if id.len() > 8 { id[id.len() - 8..].to_string() } else { id.clone() }
@@ -632,6 +633,13 @@ impl SummaryRenderer {
     }
 }
 
+fn format_run_time(run: &Run) -> String {
+    DateTime::parse_from_rfc3339(&run.started_at)
+        .ok()
+        .map(|dt| dt.with_timezone(&Local).format("%Y-%m-%d %H:%M:%S").to_string())
+        .unwrap_or_else(|| run.started_at.chars().take(19).collect())
+}
+
 /// Catmull-Rom spline interpolation.
 /// Generates `num_points` evenly spaced points along the spline that passes
 /// through all input points. Requires at least 3 input points.