ai-security-toolkit 1.1.0__tar.gz → 1.1.1__tar.gz

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-security-toolkit
-Version: 1.1.0
+Version: 1.1.1
 Summary: A red-team AI security framework with adversarial attack modules
 Author: Rishit Goel
 License: MIT

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/backdoor_trigger_attack.py

@@ -1,5 +1,10 @@
+import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
+from tensorflow.keras.datasets import mnist
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
+from tensorflow.keras.utils import to_categorical
 import os
 import random
 from datetime import datetime
@@ -9,17 +14,6 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from ai_security_toolkit.shared.log_utils import append_report_row, save_plot
 
 def main():
-
-    try:
-        import tensorflow as tf
-        from tensorflow.keras.datasets import mnist
-        from tensorflow.keras.models import Sequential
-        from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
-        from tensorflow.keras.utils import to_categorical
-    except ImportError:
-        print("❌ TensorFlow not found. Run: pip install tensorflow")
-        return
-    
     # Parameters
     trigger_label_target = 7
     trigger_class_source = 1

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/fgsm_mobilenet.py

@@ -1,22 +1,16 @@
+import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 from datetime import datetime
+from tensorflow.keras.applications.mobilenet_v2 import MobileNetV2, preprocess_input, decode_predictions
+from tensorflow.keras.preprocessing import image
+from cleverhans.tf2.attacks.fast_gradient_method import fast_gradient_method
 import sys
 import os
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from ai_security_toolkit.shared.log_utils import append_report_row, save_plot
 
 def main():
-
-    try:
-        import tensorflow as tf
-        from tensorflow.keras.applications.mobilenet_v2 import MobileNetV2, preprocess_input, decode_predictions
-        from tensorflow.keras.preprocessing import image
-        from cleverhans.tf2.attacks.fast_gradient_method import fast_gradient_method
-    except ImportError:
-        print("❌ TensorFlow not found. Run: pip install tensorflow")
-        return
-    
     # Load MobileNetV2 pretrained on ImageNet
     model = MobileNetV2(weights='imagenet')
     model.trainable = False

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/label_flip_attack.py

@@ -1,3 +1,8 @@
+import tensorflow as tf
+from tensorflow.keras.datasets import mnist
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
+from tensorflow.keras.utils import to_categorical
 import numpy as np
 from datetime import datetime
 import sys
@@ -6,17 +11,6 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from ai_security_toolkit.shared.log_utils import append_report_row, log_metrics
 
 def main():
-
-    try:
-        import tensorflow as tf
-        from tensorflow.keras.datasets import mnist
-        from tensorflow.keras.models import Sequential
-        from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
-        from tensorflow.keras.utils import to_categorical
-    except ImportError:
-        print("❌ TensorFlow not found. Run: pip install tensorflow")
-        return
-    
     # Build a simple CNN
     def build_model():
         model = Sequential([

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/membership_inference_attack.py

@@ -1,3 +1,4 @@
+import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 from datetime import datetime
@@ -7,13 +8,6 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from ai_security_toolkit.shared.log_utils import save_plot, append_report_row, log_metrics
 
 def main():
-
-    try:
-        import tensorflow as tf
-    except ImportError:
-        print("❌ TensorFlow not found. Run: pip install tensorflow")
-        return
-    
     # Load model
     model = tf.keras.models.load_model("shared/models/mnist_cnn_model.keras")
     print("✅ Loaded model from .keras file.")

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/simulate_inversion.py

@@ -1,3 +1,4 @@
+import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 from datetime import datetime
@@ -8,11 +9,6 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from ai_security_toolkit.shared.log_utils import append_report_row, save_plot
 
 def main():
-    try:
-        import tensorflow as tf
-    except ImportError:
-        print("❌ TensorFlow not found. Run: pip install tensorflow")
-        return
     # Load trained model
     model = tf.keras.models.load_model("shared/models/mnist_cnn_model.keras")
     model.trainable = False

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/steal_model.py

@@ -1,21 +1,17 @@
+import tensorflow as tf
 import numpy as np
 from datetime import datetime
+from tensorflow.keras.datasets import mnist
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
+from tensorflow.keras.utils import to_categorical
+from tensorflow.keras.models import load_model
 import sys
 import os
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from ai_security_toolkit.shared.log_utils import append_report_row, log_metrics
 
 def main():
-    try:
-        import tensorflow as tf
-        from tensorflow.keras.datasets import mnist
-        from tensorflow.keras.models import Sequential
-        from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
-        from tensorflow.keras.utils import to_categorical
-        from tensorflow.keras.models import load_model
-    except ImportError:
-        print("❌ TensorFlow not found. Run: pip install tensorflow")
-        return
     # Step 1: Load original (victim) model
     victim_model = load_model("shared/models/mnist_cnn_model.keras")
     victim_model.trainable = False

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/train_mnist_model.py

@@ -1,15 +1,11 @@
+import tensorflow as tf
+from tensorflow.keras.datasets import mnist
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
+from tensorflow.keras.utils import to_categorical
 import os
 
 def main():
-    try:
-        import tensorflow as tf
-        from tensorflow.keras.datasets import mnist
-        from tensorflow.keras.models import Sequential
-        from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
-        from tensorflow.keras.utils import to_categorical
-    except ImportError:
-        print("❌ TensorFlow not found. Run: pip install tensorflow")
-        return
     # Load and preprocess MNIST data
     (x_train, y_train), (x_test, y_test) = mnist.load_data()
 

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/run.py

@@ -1,7 +1,6 @@
 import importlib
 import sys
 import os
-
 import platform
 
 if platform.system() == "Windows":
@@ -32,8 +31,9 @@ def print_menu():
 def run_selected_module(choice_idx):
     try:
         label = list(available_modules.keys())[choice_idx - 1]
-        module_name = f"modules.{available_modules[label]}"
+        module_name = f"ai_security_toolkit.modules.{available_modules[label]}"
         print(f"\n🔍 Running: {label} ({module_name})...\n")
+
         mod = importlib.import_module(module_name)
 
         if hasattr(mod, "main"):
@@ -42,8 +42,11 @@ def run_selected_module(choice_idx):
             print("⚠️ No 'main()' found — running file as script...")
             exec(open(mod.__file__).read())
 
+    except ModuleNotFoundError:
+        print(f"❌ Could not find module: {module_name}")
+        print("💡 Make sure the package is installed correctly via pip.")
     except Exception as e:
-        print(f"❌ Error: {e}")
+        print(f"❌ Unexpected error: {e}")
 
 def main():
     while True:

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-security-toolkit
-Version: 1.1.0
+Version: 1.1.1
 Summary: A red-team AI security framework with adversarial attack modules
 Author: Rishit Goel
 License: MIT

{ai_security_toolkit-1.1.0 → ai_security_toolkit-1.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ai-security-toolkit"
-version = "1.1.0"
+version = "1.1.1"
 description = "A red-team AI security framework with adversarial attack modules"
 readme = "README.md"
 authors = [
@@ -28,3 +28,7 @@ ai-toolkit = "ai_security_toolkit.run:main"
 tensorflow = ["tensorflow"]
 cleverhans = ["cleverhans"]
 
+[tool.setuptools.packages.find]
+include = ["ai_security_toolkit*"]
+
+