Update visualization.py

visualization.py

@@ -216,7 +216,7 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
     plt.close()
     return fig
 
-def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, output_folder, desired_fps, largest_cluster, progress=None):
+def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, output_folder, desired_fps, largest_cluster):
     print(f"Creating heatmap video. Output folder: {output_folder}")
     
     os.makedirs(output_folder, exist_ok=True)
@@ -302,25 +302,8 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
     final_clip = VideoClip(combine_video_and_heatmap, duration=video.duration)
     final_clip = final_clip.set_audio(video.audio)
 
-    # Define a logger to track progress
-    class ProgressBarLogger(logging.Logger):
-        def __init__(self, name, progress_callback):
-            super().__init__(name)
-            self.progress_callback = progress_callback
-
-        def debug(self, msg, *args, **kwargs):
-            if 't=' in msg and self.progress_callback:
-                progress_percent = float(msg.split(' ')[-1].replace('%', '')) / 100
-                self.progress_callback(progress_percent, msg)
-
-        def __call__(self, msg, *args, **kwargs):
-            self.debug(msg)
-
-    logging.basicConfig(level=logging.DEBUG)
-    progress_logger = ProgressBarLogger('progress_logger', progress)
-
     # Write the final video
-    final_clip.write_videofile(heatmap_video_path, codec='libx264', audio_codec='aac', fps=video.fps, logger=progress_logger)
+    final_clip.write_videofile(heatmap_video_path, codec='libx264', audio_codec='aac', fps=video.fps)
     
     # Close the video clips
     video.close()