lobo/stt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

look for models in a better directory

Browse source
This commit is contained in:
Emil Lerch 2025-10-09 09:25:23 -07:00
parent 0262ca6659
commit 186651dce0
Signed by: lobo
GPG key ID: A7B62D657EF764F8
2 changed files with 16 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
build.zig
View file

@ -23,7 +23,7 @@ pub fn build(b: *std.Build) void {
// Install the model to the output directory // Install the model to the output directory
const install_model = b.addInstallDirectory(.{ const install_model = b.addInstallDirectory(.{
.source_dir = model_step.getOutputPath(), .source_dir = model_step.getOutputPath(),
.install_dir = .bin, .install_dir = .{ .custom = "share/vosk/models" },
.install_subdir = "vosk-model-small-en-us-0.15", .install_subdir = "vosk-model-small-en-us-0.15",
}); });
install_model.step.dependOn(&model_step.step); install_model.step.dependOn(&model_step.step);

22
src/main.zig
View file

@ -337,12 +337,11 @@ pub fn main() !void {
_ = try stdout.writeAll(" --help, -h Show this help message\n\n"); _ = try stdout.writeAll(" --help, -h Show this help message\n\n");
_ = try stdout.writeAll("EXAMPLES:\n"); _ = try stdout.writeAll("EXAMPLES:\n");
_ = try stdout.writeAll(" stt\n"); _ = try stdout.writeAll(" stt\n");
_ = try stdout.writeAll(" stt --model=vosk-model-small-en-us-0.15\n"); _ = try stdout.writeAll(" stt --model=../share/vosk/models/vosk-model-small-en-us-0.15\n");
_ = try stdout.writeAll(" stt --exec=echo\n\n"); _ = try stdout.writeAll(" stt --exec=echo\n\n");
_ = try stdout.writeAll("The application will search for models in these locations:\n"); _ = try stdout.writeAll("The application will search for models in these locations:\n");
_ = try stdout.writeAll(" vosk-model-small-en-us-0.15\n"); _ = try stdout.writeAll(" vosk-model-small-en-us-0.15\n");
_ = try stdout.writeAll(" zig-out/bin/vosk-model-small-en-us-0.15\n"); _ = try stdout.writeAll(" <binary_dir>/../share/vosk/models/vosk-model-small-en-us-0.15\n");
_ = try stdout.writeAll(" /usr/share/vosk/models/vosk-model-small-en-us-0.15\n");
return; return;
} else if (std.mem.startsWith(u8, arg, "--model=")) { } else if (std.mem.startsWith(u8, arg, "--model=")) {
model_path = arg[8..]; // Skip "--model=" model_path = arg[8..]; // Skip "--model="
@ -365,10 +364,16 @@ pub fn main() !void {
}; };
// If no model specified, try default locations // If no model specified, try default locations
var exe_path_buf: [std.fs.max_path_bytes]u8 = undefined;
const exe_path = std.fs.selfExePath(&exe_path_buf) catch "stt";
const exe_dir = std.fs.path.dirname(exe_path) orelse ".";
const xdg_model_path = try std.fmt.allocPrint(allocator, "{s}/../share/vosk/models/vosk-model-small-en-us-0.15", .{exe_dir});
defer allocator.free(xdg_model_path);
const default_paths = [_][]const u8{ const default_paths = [_][]const u8{
"vosk-model-small-en-us-0.15", "vosk-model-small-en-us-0.15",
"zig-out/bin/vosk-model-small-en-us-0.15", xdg_model_path,
"/usr/share/vosk/models/vosk-model-small-en-us-0.15",
}; };
if (model_path == null) { if (model_path == null) {
@ -384,8 +389,11 @@ pub fn main() !void {
_ = try stderr.writeAll("Error: Vosk model not found.\n\n"); _ = try stderr.writeAll("Error: Vosk model not found.\n\n");
_ = try stderr.writeAll("Usage: stt [--model=<path>] [--exec=<program>]\n\n"); _ = try stderr.writeAll("Usage: stt [--model=<path>] [--exec=<program>]\n\n");
_ = try stderr.writeAll("Locations searched:\n"); _ = try stderr.writeAll("Locations searched:\n");
inline for (default_paths) |path| for (default_paths) |path| {
_ = try stderr.writeAll("\t" ++ path ++ "\n"); _ = try stderr.writeAll("\t");
_ = try stderr.writeAll(path);
_ = try stderr.writeAll("\n");
}
_ = try stderr.writeAll("Please download the model. A fine model can be downloaded from:\n"); _ = try stderr.writeAll("Please download the model. A fine model can be downloaded from:\n");
_ = try stderr.writeAll("\thttps://alphacephei.com/vosk/models/vosk-model-small-en-us-0.15.zip\n"); _ = try stderr.writeAll("\thttps://alphacephei.com/vosk/models/vosk-model-small-en-us-0.15.zip\n");
std.process.exit(1); std.process.exit(1);