Update video_processing.py

video_processing.py

@@ -166,12 +166,29 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             audio_path = os.path.join(temp_dir, "audio.wav")
             video.export(audio_path, format="wav")
         
-            # Process audio
-            most_frequent_voice, voice_features, voice_clusters = process_audio(audio_path)
+            try:
+                # Process audio
+                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)
+                # 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
+        
+                # Generate plots for voice
+                mse_plot_voice, anomaly_segments_voice = plot_mse(df, mse_voice, "Voice",
+                                                                  color='green',
+                                                                  anomaly_threshold=anomaly_threshold)
+                mse_histogram_voice = plot_mse_histogram(mse_voice, "MSE Distribution: Voice",
+                                                         anomaly_threshold, color='green')
+                mse_heatmap_voice = plot_mse_heatmap(mse_voice, "Voice MSE Heatmap", df)
+            except Exception as e:
+                print(f"Error in voice processing: {str(e)}")
+                mse_voice = None
+                mse_plot_voice = None
+                mse_histogram_voice = None
+                mse_heatmap_voice = None
+                anomaly_segments_voice = None
+
 
 
             progress(0.95, "Generating plots")