Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 import time
 from video_processing import process_video
+from visualization import create_kdeplot, create_jointplot
 from PIL import Image
 import matplotlib
+
 matplotlib.rcParams['figure.dpi'] = 500
 matplotlib.rcParams['savefig.dpi'] = 500
 
@@ -14,7 +16,7 @@ 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] * 20
 
         exec_time, results_summary, df, mse_embeddings, mse_posture, \
             mse_plot_embeddings, mse_histogram_embeddings, \
@@ -29,7 +31,10 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
 
         face_samples_frequent = [Image.open(path) for path in face_samples_frequent]
         face_samples_other = [Image.open(path) for path in face_samples_other]
-
+        
+        kde_plot = create_kdeplot(df, mse_embeddings, mse_posture)
+        joint_plot = create_jointplot(df, mse_embeddings, mse_posture)
+        
         output = [
             exec_time, results_summary,
             df, mse_embeddings, mse_posture,
@@ -40,6 +45,7 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
             face_samples_frequent, face_samples_other,
             aligned_faces_folder, frames_folder,
             mse_embeddings, mse_posture,
+            kde_plot, joint_plot
         ]
 
         return output
@@ -49,7 +55,7 @@ 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] * 20
 
 with gr.Blocks() as iface:
     gr.Markdown("""
@@ -83,6 +89,11 @@ 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("Combined"):
+            with gr.Row():
+                kde_plot = gr.Plot(label="KDE Plot: Facial Features vs Body Posture MSE")
+                joint_plot = gr.Plot(label="Joint Plot: Facial Features vs Body Posture MSE")
+
         with gr.Tab("Face Samples"):
             face_samples_most_frequent = gr.Gallery(label="Most Frequent Person Samples (Target)", columns=6, rows=2, height="auto")
             face_samples_others = gr.Gallery(label="Other Persons Samples", columns=6, rows=1, height="auto")
@@ -107,7 +118,8 @@ with gr.Blocks() as iface:
             anomaly_frames_features, anomaly_frames_posture,
             face_samples_most_frequent, face_samples_others,
             aligned_faces_folder_store, frames_folder_store,
-            mse_heatmap_embeddings_store, mse_heatmap_posture_store
+            mse_heatmap_embeddings_store, mse_heatmap_posture_store,
+            kde_plot, joint_plot
         ]
     ).then(
         lambda: gr.Group(visible=True),