Update visualization.py

visualization.py

@@ -291,8 +291,6 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
         seconds = t
         timecode = f"{int(seconds//3600):02d}:{int((seconds%3600)//60):02d}:{int(seconds%60):02d}"
         
-        # We'll use PIL for text as it's easier to work with in this context
-        from PIL import Image, ImageDraw, ImageFont
         pil_img = Image.fromarray(combined_frame)
         draw = ImageDraw.Draw(pil_img)
         font = ImageFont.load_default()
@@ -301,7 +299,7 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
         return np.array(pil_img)
 
     # Create the final video
-    final_clip = VideoFileClip(video_path).fl(make_frame)
+    final_clip = video.fl_image(make_frame)
     final_clip = final_clip.set_duration(video.duration)
 
     # Write the final video
@@ -317,6 +315,4 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
         return heatmap_video_path
     else:
         print(f"Failed to create heatmap video at: {heatmap_video_path}")
-        return None
-
-
+        return None