Update main.py

main.py

@@ -239,7 +239,7 @@ def transcription(stt_tokenizer, stt_model, summarizer, dia_pipeline, filename,
                     if filename.endswith((".mp3", ".mp4")):
                         myaudio, filename = convert_file_to_wav(myaudio, filename)
                     else:
-                        filename = "../data/" + filename
+                        filename = "whisper_app/data/" + filename
                         myaudio.export(filename, format="wav")
 
                     # Differentiate speakers process
@@ -259,7 +259,7 @@ def transcription(stt_tokenizer, stt_model, summarizer, dia_pipeline, filename,
                     if timestamps_token:
                         update_session_state("chosen mode", "NODIA_TS")
                     
-                    filename = "../data/" + filename
+                    filename = "whisper_app/data/" + filename
                     # Transcribe process with non Diarization Mode
                     save_result, txt_text, srt_text = transcription_non_diarization(filename, myaudio, start, end, diarization_token, timestamps_token, srt_token, summarize_token, stt_model, stt_tokenizer, min_space, max_space, save_result, txt_text, srt_text)
 
@@ -280,7 +280,7 @@ def transcription(stt_tokenizer, stt_model, summarizer, dia_pipeline, filename,
                                 txt_text += my_split_text
 
                 # Delete files
-                clean_directory("../data")  # clean folder that contains generated files
+                clean_directory("whisper_app/data")  # clean folder that contains generated files
 
                 # Display the final transcript
                 if txt_text != "":
@@ -820,7 +820,7 @@ def convert_file_to_wav(aud_seg, filename):
     :param filename: name of the file
     :return: name of the converted file
     """
-    filename = "../data/my_wav_file_" + filename[:-3] + "wav"
+    filename = "whisper_app/data/my_wav_file_" + filename[:-3] + "wav"
     aud_seg.export(filename, format="wav")
 
     newaudio = AudioSegment.from_file(filename)