Skip to content

Commit

Permalink
Change temp file name for server application
Browse files Browse the repository at this point in the history
Avoid issue of removing file if it exists in the current working
directory
  • Loading branch information
felrock committed Nov 21, 2023
1 parent 9befab5 commit 817f8d5
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions examples/server/server.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -453,21 +453,23 @@ int main(int argc, char ** argv) {
std::vector<float> pcmf32; // mono-channel F32 PCM
std::vector<std::vector<float>> pcmf32s; // stereo-channel F32 PCM

// write file to temporary file
std::ofstream temp_file{filename, std::ios::binary};
// write to temporary file
const std::string temp_filename = "whisper_server_temp_file.wav";
std::ofstream temp_file{temp_filename, std::ios::binary};
temp_file << audio_file.content;
temp_file.close();

// read wav content into pcmf32
if (!::read_wav(filename, pcmf32, pcmf32s, params.diarize)) {
fprintf(stderr, "error: failed to read WAV file '%s'\n", filename.c_str());
if (!::read_wav(temp_filename, pcmf32, pcmf32s, params.diarize)) {
fprintf(stderr, "error: failed to read WAV file '%s'\n", temp_filename.c_str());
const std::string error_resp = "{\"error\":\"failed to read WAV file\"}";
res.set_content(error_resp, "application/json");
std::remove(temp_filename.c_str());
whisper_mutex.unlock();
return;
}
// remove temp file
std::remove(filename.c_str());
std::remove(temp_filename.c_str());

printf("Successfully loaded %s\n", filename.c_str());

Expand Down

0 comments on commit 817f8d5

Please sign in to comment.