Update video_processing.py

video_processing.py

@@ -166,29 +166,32 @@ 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")
         
-            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)  # 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
-
+        try:
+            # Process audio
+            most_frequent_voice, voice_features, voice_clusters = process_audio(audio_path)
+    
+            if len(voice_features) < 2:
+                print("Not enough voice segments for processing. Skipping voice analysis.")
+                raise ValueError("Insufficient voice data")
+    
+            # Perform anomaly detection on voice
+            X_voice = np.array(most_frequent_voice)
+            mse_voice = anomaly_detection(X_voice, X_voice)
+    
+            # 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")