Update video_processing.py

video_processing.py

@@ -167,41 +167,27 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             video.export(audio_path, format="wav")
     
             # Process audio
-            most_frequent_voice, voice_features, voice_clusters = process_audio(audio_path)
+            most_frequent_voice, voice_features, voice_clusters = process_audio(audio_path, desired_fps, video_duration)
     
             if len(voice_features) < 2:
                 print("Not enough voice segments for processing. Skipping voice analysis.")
-                raise ValueError("Insufficient voice data")
+                mse_voice = None
+                mse_plot_voice = None
+                mse_histogram_voice = None
+                mse_heatmap_voice = None
+                anomaly_segments_voice = None
+            else:
+                # 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)
-    
-            progress(0.95, "Generating plots")
-            
-            # Generate plots for facial features
-            mse_plot_embeddings, anomaly_frames_embeddings = plot_mse(df, mse_embeddings, "Facial Features",
-                                                                      color=GRAPH_COLORS['facial_embeddings'],
-                                                                      anomaly_threshold=anomaly_threshold)
-            mse_histogram_embeddings = plot_mse_histogram(mse_embeddings, "MSE Distribution: Facial Features",
-                                                          anomaly_threshold, color=GRAPH_COLORS['facial_embeddings'])
-            mse_heatmap_embeddings = plot_mse_heatmap(mse_embeddings, "Facial Features MSE Heatmap", df)
-    
-            # Generate plots for body posture
-            mse_plot_posture, anomaly_frames_posture = plot_mse(df, mse_posture, "Body Posture",
-                                                                color=GRAPH_COLORS['body_posture'],
-                                                                anomaly_threshold=anomaly_threshold)
-            mse_histogram_posture = plot_mse_histogram(mse_posture, "MSE Distribution: Body Posture",
-                                                       anomaly_threshold, color=GRAPH_COLORS['body_posture'])
-            mse_heatmap_posture = plot_mse_heatmap(mse_posture, "Body Posture MSE Heatmap", df)
-    
-            # 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)
+                # 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 details: {str(e)}")