Update app.py

app.py

@@ -22,8 +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, \
-            anomaly_sentences_features, anomaly_sentences_posture = results
+            aligned_faces_folder, frames_folder = 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]
@@ -31,12 +30,6 @@ 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]
 
-        anomaly_sentences_features, anomaly_sentences_posture = results[-2:]
-
-        # Format anomaly sentences output
-        sentences_features_output = format_anomaly_sentences(anomaly_sentences_features, "Facial Features")
-        sentences_posture_output = format_anomaly_sentences(anomaly_sentences_posture, "Body Posture")
-
         output = [
             exec_time, results_summary,
             df, mse_embeddings, mse_posture,
@@ -47,7 +40,6 @@ 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,
-            sentences_features_output, sentences_posture_output
         ]
 
         return output
@@ -57,7 +49,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] * 20
+        return [error_message] + [None] * 18
 
 with gr.Blocks() as iface:
     gr.Markdown("""
@@ -94,18 +86,6 @@ 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("Sentences"):
-            with gr.Row():
-                anomaly_sentences_features_output = gr.Textbox(label="Sentences before Facial Feature Anomalies",
-                                                               lines=10)
-                anomaly_frames_features = gr.Gallery(label="Anomaly Frames (Facial Features)", columns=6, rows=2,
-                                                     height="auto")
-
-            with gr.Row():
-                anomaly_sentences_posture_output = gr.Textbox(label="Sentences before Body Posture Anomalies", lines=10)
-                anomaly_frames_posture = gr.Gallery(label="Anomaly Frames (Body Posture)", columns=6, rows=2,
-                                                    height="auto")
-
         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")
@@ -139,8 +119,7 @@ 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,
-            anomaly_sentences_features_output, anomaly_sentences_posture_output
+            mse_heatmap_embeddings_store, mse_heatmap_posture_store
         ]
     ).then(
         lambda: gr.Group(visible=True),