Update video_processing.py

video_processing.py

@@ -155,18 +155,17 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
         try:
             X_posture = np.array([posture_scores_by_frame.get(frame, None) for frame in df['Frame']])
             X_posture = X_posture[X_posture != None].reshape(-1, 1)
-
+    
             if len(X_posture) == 0:
                 raise ValueError("No valid posture data found")
-
+    
             mse_embeddings, mse_posture = anomaly_detection(X_embeddings, X_posture)
-
+    
             # Extract audio from video
             video = AudioSegment.from_file(video_path, "mp4")
             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)
     
@@ -178,41 +177,24 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             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")
+            
+            # 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)
-
-            mse_heatmap_embeddings = plot_mse_heatmap(mse_embeddings, "Facial Features MSE Heatmap", df)
-
+    
             # Generate plots for voice
             mse_plot_voice, anomaly_segments_voice = plot_mse(df, mse_voice, "Voice",
                                                               color='green',
@@ -220,12 +202,12 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             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)}")
             import traceback
             traceback.print_exc()
-            return (f"Error in video processing: {str(e)}",) + (None,) * 14
+            return (f"Error in video processing: {str(e)}",) + (None,) * 20
 
         progress(1.0, "Preparing results")
         results = f"Number of persons detected: {num_clusters}\n\n"