mirrors/obs-localvocal
1
0
Fork 0
mirror of https://github.com/occ-ai/obs-localvocal synced 2024-11-07 18:57:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

sorting

Browse Source
This commit is contained in:
Roy Shilkrot 2023-08-13 23:17:45 +03:00
parent 1b86990e91
commit 788b232ef1
2 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/transcription-filter.cpp
View File

@ -7,6 +7,8 @@
#include "whisper-language.h" #include "whisper-language.h"
#include "model-utils/model-downloader.h" #include "model-utils/model-downloader.h"
#include <algorithm>
inline enum speaker_layout convert_speaker_layout(uint8_t channels) inline enum speaker_layout convert_speaker_layout(uint8_t channels)
{ {
switch (channels) { switch (channels) {
@ -452,22 +454,19 @@ obs_properties_t *transcription_filter_properties(void *data)
obs_property_t *whisper_language_select_list = obs_property_t *whisper_language_select_list =
obs_properties_add_list(whisper_params_group, "whisper_language_select", "Language", obs_properties_add_list(whisper_params_group, "whisper_language_select", "Language",
OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING); OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING);
// sort the languages by name using a sorting function // sort the languages by flipping the map
std::vector<std::pair<std::string, std::string>> whisper_available_lang_vector; std::map<std::string, std::string> whisper_available_lang_flip;
for (auto const &pair : whisper_available_lang) { for (auto const &pair : whisper_available_lang) {
whisper_available_lang_vector.push_back(pair); whisper_available_lang_flip[pair.second] = pair.first;
} }
std::sort(whisper_available_lang_vector.begin(), whisper_available_lang_vector.end(),
[](const std::pair<std::string, std::string> &a,
const std::pair<std::string, std::string> &b) { return a.second < b.second; });
// iterate over all available languages and add them to the list // iterate over all available languages and add them to the list
for (auto const &pair : whisper_available_lang_vector) { for (auto const &pair : whisper_available_lang_flip) {
// Capitalize the language name // Capitalize the language name
std::string language_name = pair.second; std::string language_name = pair.first;
language_name[0] = toupper(language_name[0]); language_name[0] = (char)toupper(language_name[0]);
obs_property_list_add_string(whisper_language_select_list, language_name.c_str(), obs_property_list_add_string(whisper_language_select_list, language_name.c_str(),
pair.first.c_str()); pair.second.c_str());
} }
obs_property_t *whisper_sampling_method_list = obs_properties_add_list( obs_property_t *whisper_sampling_method_list = obs_properties_add_list(

4
src/whisper-language.h
View File

@ -5,6 +5,10 @@
#include <string> #include <string>
static const std::map<std::string, std::string> whisper_available_lang = { static const std::map<std::string, std::string> whisper_available_lang = {
{
"auto",
"auto",
},
{ {
"en", "en",
"english", "english",