warren/momentry_playground
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(overlay): update ASR loader

Browse Source
This commit is contained in:
Warren Lo
2026-03-19 01:26:16 +08:00
parent 19d60de245
commit 55bb5d62f1
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/overlay/asr.rs
View File

@@ -27,18 +27,19 @@ impl AsrLoader {
pub fn load(path: &Path) -> Result<Self> { pub fn load(path: &Path) -> Result<Self> {
let content = std::fs::read_to_string(path) let content = std::fs::read_to_string(path)
.with_context(|| format!("Failed to read ASR file: {:?}", path))?; .with_context(|| format!("Failed to read ASR file: {:?}", path))?;
let data: AsrData = serde_json::from_str(&content) let data: AsrData =
.with_context(|| "Failed to parse ASR JSON")?; serde_json::from_str(&content).with_context(|| "Failed to parse ASR JSON")?;
Ok(Self { data }) Ok(Self { data })
} }
pub fn get_segment_at(&self, time_ms: f64) -> Option<&AsrSegment> { pub fn get_segment_at(&self, time_ms: f64) -> Option<&AsrSegment> {
let time_sec = time_ms / 1000.0; let time_sec = time_ms / 1000.0;
self.data.segments.iter().find(|seg| { self.data
time_sec >= seg.start && time_sec < seg.end .segments
}) .iter()
.find(|seg| time_sec >= seg.start && time_sec < seg.end)
} }
pub fn get_text_at(&self, time_ms: f64) -> Option<String> { pub fn get_text_at(&self, time_ms: f64) -> Option<String> {