Update app.py

app.py

@@ -15,7 +15,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] * 24
+            return [results[0]] + [None] * 26
 
         exec_time, results_summary, df, mse_embeddings, mse_posture, mse_voice, \
             mse_plot_embeddings, mse_plot_posture, mse_plot_voice, \
@@ -24,7 +24,8 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
             face_samples_frequent, \
             anomaly_faces_embeddings, anomaly_frames_posture_images, \
             aligned_faces_folder, frames_folder, \
-            heatmap_video_path, correlation_heatmap = results
+            heatmap_video_path, correlation_heatmap,
+            combined_mse_plot, combined_heatmap = results
 
         anomaly_faces_embeddings_pil = [Image.fromarray(face) for face in anomaly_faces_embeddings] if anomaly_faces_embeddings is not None else []
         anomaly_frames_posture_pil = [Image.fromarray(frame) for frame in anomaly_frames_posture_images] if anomaly_frames_posture_images is not None else []
@@ -42,7 +43,8 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
             aligned_faces_folder, frames_folder,
             mse_embeddings, mse_posture, mse_voice,
             heatmap_video_path,
-            correlation_heatmap
+            correlation_heatmap, 
+            combined_mse_plot, combined_heatmap
         ]
 
         return output
@@ -52,7 +54,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] * 24
+        return [error_message] + [None] * 26
 
 def show_results(outputs):
     return gr.Group(visible=True)
@@ -97,6 +99,8 @@ with gr.Blocks() as iface:
 
             with gr.TabItem("Combined"):
                 heatmap_video = gr.Video(label="Video with Anomaly Heatmap")
+                combined_mse_plot = gr.Plot(label="Combined MSE Plot")
+                combined_heatmap = gr.Plot(label="Combined Heatmap")
                 correlation_heatmap_plot = gr.Plot(label="Correlation Heatmap")
                 
     df_store = gr.State()
@@ -122,7 +126,7 @@ with gr.Blocks() as iface:
             face_samples_most_frequent,
             aligned_faces_folder_store, frames_folder_store,
             mse_heatmap_embeddings_store, mse_heatmap_posture_store, mse_heatmap_voice_store,
-            heatmap_video, correlation_heatmap_plot
+            heatmap_video, correlation_heatmap_plot, combined_mse_plot, combined_heatmap
         ]
     ).then(
         show_results,