fix preprocessing

app/main.py

@@ -154,6 +154,7 @@ def preprocess_audio(path, save_dir):
     y_no_gaps = np.concatenate([y[start:end] for start, end in intervals])
 
     file_name_without_extension = os.path.basename(path).split('.')[0]
+    extension = os.path.basename(path).split('.')[1]
     y_trimmed, _ = librosa.effects.trim(y_no_gaps, top_db = 20) 
     D = librosa.stft(y)
     S_db = librosa.amplitude_to_db(np.abs(D), ref=np.max)
@@ -190,12 +191,11 @@ def preprocess_audio(path, save_dir):
 
     # Feature Extraction (example: MFCCs)
 #     mfccs = librosa.feature.mfcc(y=y_normalized, sr=target_sr, n_mfcc=20)
-
     
-    output_file_path = os.path.join(save_dir, f"{file_name_without_extension}.wav")
+    # output_file_path = os.path.join(save_dir, f"{file_name_without_extension}.{extension}")
 
     # Write the audio data to the output file in .wav format
-    sf.write(output_file_path, y_normalized, target_sr)
+    sf.write(path, y_normalized, target_sr)
     
     return 'success'