diff --git a/MANIFEST.in b/MANIFEST.in index f595173..5320e4b 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -4,7 +4,6 @@ include README.md include *.py recursive-include wazp *.py recursive-include sample_project *.avi -recursive-include sample_project *.png recursive-include sample_project_2 *.avi recursive-include sample_project_2 *.h5 diff --git a/sample_project/videos/roi_frames/jwaspE_nectar-open-close_frame-0.png b/sample_project/videos/roi_frames/jwaspE_nectar-open-close_frame-0.png deleted file mode 100644 index d6391cd..0000000 Binary files a/sample_project/videos/roi_frames/jwaspE_nectar-open-close_frame-0.png and /dev/null differ diff --git a/sample_project/videos/roi_frames/jwaspK_nectar-open-close_frame-0.png b/sample_project/videos/roi_frames/jwaspK_nectar-open-close_frame-0.png deleted file mode 100644 index 040845b..0000000 Binary files a/sample_project/videos/roi_frames/jwaspK_nectar-open-close_frame-0.png and /dev/null differ diff --git a/wazp/callbacks.py b/wazp/callbacks.py index 16811d8..f85a426 100644 --- a/wazp/callbacks.py +++ b/wazp/callbacks.py @@ -590,8 +590,8 @@ def update_frame_graph( video_path = pl.Path(video_path) video_name = video_path.name - # Get directory and path for roi frame - frames_dir = video_path.parent / "roi_frames" + # Cache frames in a .WAZP folder in the home directory + frames_dir = pl.Path.home() / ".WAZP" / "roi_frames" frames_dir.mkdir(parents=True, exist_ok=True) frame_path = frames_dir / f"{video_path.stem}_frame-{frame_idx}.png" @@ -621,7 +621,9 @@ def update_frame_graph( ) if frame_path.exists(): - alert_msg = f"Found cached frame {frame_idx} from {video_name}" + alert_msg = ( + f"Showing cached frame {frame_idx} from {video_name}" + ) alert_color = "success" alert_is_open = True else: