Update app.py

app.py

@@ -15,12 +15,13 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
 
         if isinstance(results[0], str) and results[0].startswith("Error"):
             print(f"Error occurred: {results[0]}")
-            return [results[0]] + [None] * 18
+            return [results[0]] + [None] * 23  # Increased number of None values
 
-        exec_time, results_summary, df, mse_embeddings, mse_posture, \
+        exec_time, results_summary, df, mse_embeddings, mse_posture, mse_voice, \
             mse_plot_embeddings, mse_histogram_embeddings, \
             mse_plot_posture, mse_histogram_posture, \
-            mse_heatmap_embeddings, mse_heatmap_posture, \
+            mse_plot_voice, mse_histogram_voice, \
+            mse_heatmap_embeddings, mse_heatmap_posture, mse_heatmap_voice, \
             face_samples_frequent, \
             anomaly_faces_embeddings, anomaly_frames_posture_images, \
             aligned_faces_folder, frames_folder, \
@@ -33,14 +34,14 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
 
         output = [
             exec_time, results_summary,
-            df, mse_embeddings, mse_posture,
-            mse_plot_embeddings, mse_plot_posture,
-            mse_histogram_embeddings, mse_histogram_posture,
-            mse_heatmap_embeddings, mse_heatmap_posture,
+            df, mse_embeddings, mse_posture, mse_voice,
+            mse_plot_embeddings, mse_plot_posture, mse_plot_voice,
+            mse_histogram_embeddings, mse_histogram_posture, mse_histogram_voice,
+            mse_heatmap_embeddings, mse_heatmap_posture, mse_heatmap_voice,
             anomaly_faces_embeddings_pil, anomaly_frames_posture_pil,
             face_samples_frequent,
             aligned_faces_folder, frames_folder,
-            mse_embeddings, mse_posture,
+            mse_embeddings, mse_posture, mse_voice,
             heatmap_video_path
         ]
 
@@ -51,14 +52,14 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
         print(error_message)
         import traceback
         traceback.print_exc()
-        return [error_message] + [None] * 18
+        return [error_message] + [None] * 23  # Increased number of None values
 
 with gr.Blocks() as iface:
     gr.Markdown("""
     # Multimodal Behavioral Anomalies Detection
 
-    This tool detects anomalies in facial expressions and body language over the timeline of a video.
-    It extracts faces and postures from video frames, and analyzes them to identify anomalies using time series analysis and a variational autoencoder (VAE) approach.
+    This tool detects anomalies in facial expressions, body language, and voice over the timeline of a video.
+    It extracts faces, postures, and voice from video frames, and analyzes them to identify anomalies using time series analysis and a variational autoencoder (VAE) approach.
     """)
 
     with gr.Row():
@@ -86,30 +87,37 @@ with gr.Blocks() as iface:
             mse_posture_heatmap = gr.Plot(label="MSE Heatmap: Body Posture")
             anomaly_frames_posture = gr.Gallery(label="Anomaly Frames (Body Posture)", columns=6, rows=2, height="auto")
 
+        with gr.Tab("Voice"):
+            mse_voice_plot = gr.Plot(label="MSE: Voice")
+            mse_voice_hist = gr.Plot(label="MSE Distribution: Voice")
+            mse_voice_heatmap = gr.Plot(label="MSE Heatmap: Voice")
+
         with gr.Tab("Video with Heatmap"):
             heatmap_video = gr.Video(label="Video with Anomaly Heatmap")
 
     df_store = gr.State()
     mse_features_store = gr.State()
     mse_posture_store = gr.State()
+    mse_voice_store = gr.State()
     aligned_faces_folder_store = gr.State()
     frames_folder_store = gr.State()
     mse_heatmap_embeddings_store = gr.State()
     mse_heatmap_posture_store = gr.State()
+    mse_heatmap_voice_store = gr.State()
 
     process_btn.click(
         process_and_show_completion,
         inputs=[video_input, anomaly_threshold, fps_slider],
         outputs=[
             execution_time, results_text, df_store,
-            mse_features_store, mse_posture_store,
-            mse_features_plot, mse_posture_plot,
-            mse_features_hist, mse_posture_hist,
-            mse_features_heatmap, mse_posture_heatmap,
+            mse_features_store, mse_posture_store, mse_voice_store,
+            mse_features_plot, mse_posture_plot, mse_voice_plot,
+            mse_features_hist, mse_posture_hist, mse_voice_hist,
+            mse_features_heatmap, mse_posture_heatmap, mse_voice_heatmap,
             anomaly_frames_features, anomaly_frames_posture,
             face_samples_most_frequent,
             aligned_faces_folder_store, frames_folder_store,
-            mse_heatmap_embeddings_store, mse_heatmap_posture_store,
+            mse_heatmap_embeddings_store, mse_heatmap_posture_store, mse_heatmap_voice_store,
             heatmap_video
         ]
     ).then(