Update transcription_diarization.py

transcription_diarization.py

@@ -3,7 +3,7 @@ import time
 import json
 import os
 import urllib.parse
-from moviepy.editor import VideoFileClip
+from moviepy.editor import VideoFileClip, AudioFileClip
 import requests
 from botocore.exceptions import ClientError
 from config import aws_access_key_id, aws_secret_access_key
@@ -15,10 +15,21 @@ def convert_to_mono_audio(video_path, output_format='wav'):
     try:
         video = VideoFileClip(video_path)
         audio = video.audio
-        audio = audio.set_channels(1)  # Convert to mono
-        audio.write_audiofile(output_path)
+        
+        # Convert stereo to mono by averaging the two channels
+        mono_audio = AudioFileClip(video_path).to_soundarray()
+        mono_audio = np.mean(mono_audio, axis=1)
+        
+        # Create a new AudioFileClip with the mono audio
+        mono_clip = AudioFileClip(video_path).set_array_audio(mono_audio)
+        
+        # Write the mono audio to file
+        mono_clip.write_audiofile(output_path, fps=audio.fps)
+        
         video.close()
         audio.close()
+        mono_clip.close()
+        
         return output_path
     except Exception as e:
         print(f"Error during audio conversion: {str(e)}")