Update video_processing.py

video_processing.py

@@ -15,7 +15,7 @@ import pandas as pd
 from facenet_pytorch import MTCNN
 import torch
 import mediapipe as mp
-from voice_analysis import process_audio, cluster_voices, get_most_frequent_voice
+from voice_analysis import process_audio
 from pydub import AudioSegment
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -167,13 +167,11 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             video.export(audio_path, format="wav")
         
             # Process audio
-            voice_embeddings = process_audio(audio_path)
-            voice_clusters = cluster_voices(voice_embeddings)
-            most_frequent_voice = get_most_frequent_voice(voice_embeddings, voice_clusters)
+            most_frequent_voice, voice_features, voice_clusters = process_audio(audio_path)
         
             # Perform anomaly detection on voice
             X_voice = np.array(most_frequent_voice)
-            mse_voice = anomaly_detection(X_voice, X_voice)  # Using the same function as for facial features
+            mse_voice = anomaly_detection(X_voice, X_voice)
 
 
             progress(0.95, "Generating plots")