diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 216419d5..40195246 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -232,8 +232,10 @@ def chat_with_code( if file_path.lower().endswith( ".mp4" ) or file_path.lower().endswith(".mov"): - ndarray = extract_frames(media) - save_video([frame for frame, _ in ndarray], file_path) + video_frames = extract_frames(media) + save_video( + [frame for frame, _ in video_frames], file_path + ) else: ndarray = load_image(media) save_image(ndarray, file_path)