Update app.py

app.py

@@ -22,7 +22,7 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
             mse_heatmap_embeddings, mse_heatmap_posture, \
             face_samples_frequent, face_samples_other, \
             anomaly_faces_embeddings, anomaly_frames_posture_images, \
-            aligned_faces_folder, frames_folder = results
+            aligned_faces_folder, frames_folder, annotated_video_path = results
 
         anomaly_faces_embeddings_pil = [Image.fromarray(face) for face in anomaly_faces_embeddings]
         anomaly_frames_posture_pil = [Image.fromarray(frame) for frame in anomaly_frames_posture_images]
@@ -39,7 +39,7 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
             anomaly_faces_embeddings_pil, anomaly_frames_posture_pil,
             face_samples_frequent, face_samples_other,
             aligned_faces_folder, frames_folder,
-            mse_embeddings, mse_posture,
+            mse_embeddings, mse_posture, annotated_video_path
         ]
 
         return output
@@ -86,6 +86,9 @@ 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("Annotated Video"):
+            annotated_video_output = gr.Video(label="Annotated Video")
+        
         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")
@@ -119,7 +122,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,
+            annotated_video_output
         ]
     ).then(
         lambda: gr.Group(visible=True),