Add GitStatusChecker to detect if branch is behind upstream.Show current branch status and also check 'main' if not on it (#48)

Author: DeepKnowledge1

anodet/general.py

@@ -2,6 +2,8 @@
 import os
 import matplotlib.pyplot as plt
 import torch
+import subprocess
+import sys
 
 
 
@@ -148,4 +150,67 @@ def get_avg_time_ms(self, num_operations):
         """
         if num_operations > 0:
             return (self.accumulated_time / num_operations) * 1000
-        return 0.0
+        return 0.0
+
+
+class GitStatusChecker:
+    def __init__(self, branch: str = "HEAD"):
+        self.branch = branch
+
+    def _run(self, cmd: list[str]) -> str:
+        result = subprocess.run(cmd, capture_output=True, text=True, check=True)
+        return result.stdout.strip()
+
+    def is_repo(self) -> bool:
+        try:
+            out = self._run(["git", "rev-parse", "--is-inside-work-tree"])
+            return out == "true"
+        except subprocess.CalledProcessError:
+            return False
+
+    def get_current_branch(self) -> str:
+        try:
+            return self._run(["git", "rev-parse", "--abbrev-ref", "HEAD"])
+        except subprocess.CalledProcessError:
+            return "UNKNOWN"
+
+    def check_branch_status(self, branch_name: str) -> None:
+        print(f"\n🌿 Checking branch: {branch_name}")
+
+        # Make sure branch exists locally
+        try:
+            self._run(["git", "rev-parse", "--verify", branch_name])
+        except subprocess.CalledProcessError:
+            print(f"⚠️ Branch '{branch_name}' does not exist locally.")
+            return
+
+        # Fetch latest changes from remote
+        self._run(["git", "fetch"])
+
+        # Compare with upstream
+        try:
+            counts = self._run(["git", "rev-list", "--left-right", "--count", f"{branch_name}...{branch_name}@{{u}}"])
+            ahead, behind = map(int, counts.split())
+        except subprocess.CalledProcessError:
+            print(f"⚠️ No upstream set for '{branch_name}'. Use:")
+            print(f"   git branch --set-upstream-to origin/{branch_name} {branch_name}")
+            return
+
+        if behind > 0:
+            print(f"⬇️ Branch '{branch_name}' is behind by {behind} commit(s). Run `git pull`.")
+        elif ahead > 0:
+            print(f"⬆️ Branch '{branch_name}' is ahead by {ahead} commit(s). Run `git push`.")
+        else:
+            print(f"✅ Branch '{branch_name}' is up-to-date with the remote.")
+
+    def check_status(self) -> None:
+        if not self.is_repo():
+            print("❌ Not a git repository. Navigate to a repo and try again.")
+            sys.exit(1)
+
+        current_branch = self.get_current_branch()
+        self.check_branch_status(current_branch)
+
+        # If not on main, also check main
+        if current_branch != "main":
+            self.check_branch_status("main")

detect.py

@@ -83,12 +83,12 @@ def parse_args():
     # Visualization parameters
     parser.add_argument(
         "--enable_visualization",
-        action="store_true",
+        action="store_false",
         help="Enable visualization of results.",
     )
     parser.add_argument(
         "--save_visualizations",
-        action="store_true",
+        action="store_false",
         help="Save visualization images to disk.",
     )
     parser.add_argument(

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "AnomaVision"
-version = "2.0.37"
+version = "2.0.38"
 description = "Deep learnIng Anomaly Detection EnvironMent [AnomaVision] is a deep learning library that aims to collect state-of-the-art anomaly detection algorithms for benchmarking on both public and private datasets. PaDimOpti provides several ready-to-use implementations of anomaly detection algorithms described in the recent literature, as well as a set of tools that facilitate the development and implementation of custom models. The library has a strong focus on image-based anomaly detection, where the goal of the algorithm is to identify anomalous images, or anomalous pixel regions within images in a dataset. PaDimOpti is constantly being updated with new algorithms and training/inference extensions, so stay tuned!!"
 authors = ["Deep Knowledge <Deepp.Knowledge@gmail.com>"]
 readme = "README.md"

train.py

@@ -11,10 +11,13 @@
 
 import anodet
 from anodet.utils import get_logger, save_args_to_yaml, setup_logging
-from anodet.general import increment_path
+from anodet.general import increment_path, GitStatusChecker
 
 # pre-commit run trailing-whitespace --files .\anodet\utils.py
 
+checker = GitStatusChecker()
+checker.check_status()
+
 def parse_args():
     parser = argparse.ArgumentParser(
         description="Train a PaDiM model for anomaly detection."