ai-security-toolkit 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

ai_security_toolkit/modules/backdoor_trigger_attack.py

@@ -1,10 +1,5 @@
-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
@@ -14,6 +9,17 @@ 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/modules/fgsm_mobilenet.py

@@ -1,16 +1,22 @@
-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/modules/label_flip_attack.py

@@ -1,8 +1,3 @@
-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
@@ -11,6 +6,17 @@ 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/modules/membership_inference_attack.py

@@ -1,4 +1,3 @@
-import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 from datetime import datetime
@@ -8,6 +7,13 @@ 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/modules/simulate_inversion.py

@@ -1,4 +1,3 @@
-import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 from datetime import datetime
@@ -9,6 +8,11 @@ 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/modules/steal_model.py

@@ -1,17 +1,21 @@
-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/modules/train_mnist_model.py

@@ -1,11 +1,15 @@
-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/run.py

@@ -2,6 +2,12 @@ import importlib
 import sys
 import os
 
+import platform
+
+if platform.system() == "Windows":
+    print("ℹ️ Running on Windows. If you see DLL errors, install:")
+    print("👉 https://aka.ms/vs/17/release/vc_redist.x64.exe")
+
 # Add project root to PYTHONPATH
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '.')))
 

{ai_security_toolkit-1.0.0.dist-info → ai_security_toolkit-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-security-toolkit
-Version: 1.0.0
+Version: 1.1.0
 Summary: A red-team AI security framework with adversarial attack modules
 Author: Rishit Goel
 License: MIT
@@ -11,6 +11,10 @@ Requires-Dist: numpy
 Requires-Dist: matplotlib
 Requires-Dist: pandas
 Requires-Dist: cleverhans
+Provides-Extra: tensorflow
+Requires-Dist: tensorflow; extra == "tensorflow"
+Provides-Extra: cleverhans
+Requires-Dist: cleverhans; extra == "cleverhans"
 Dynamic: requires-python
 
 # 🛡️ AI Security Toolkit
@@ -18,6 +22,7 @@ Dynamic: requires-python
 [![Made by Rishit Goel 💻](https://img.shields.io/badge/Made%20by-Rishit%20Goel-blueviolet?style=flat-square&logo=github)](https://github.com/rishit03)
 ![Python](https://img.shields.io/badge/Python-3.8+-blue?logo=python)
 ![License](https://img.shields.io/github/license/rishit03/ai-security-toolkit?style=flat)
+[![PyPI](https://img.shields.io/pypi/v/ai-security-toolkit?color=brightgreen)](https://pypi.org/project/ai-security-toolkit/)
 ![GitHub Repo stars](https://img.shields.io/github/stars/rishit03/ai-security-toolkit?style=social)
 ![GitHub last commit](https://img.shields.io/github/last-commit/rishit03/ai-security-toolkit?color=green)
 

ai_security_toolkit-1.1.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+ai_security_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_security_toolkit/run.py,sha256=glzg-wrt34IwyBIp8xi0YkMqaInW68ZoX_066JGbKsY,2090
+ai_security_toolkit/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_security_toolkit/modules/backdoor_trigger_attack.py,sha256=YYLqSu0XabP6krHJ0E0Bh8XLAjX9DxinTCi0Pa7DEN4,4573
+ai_security_toolkit/modules/fgsm_mobilenet.py,sha256=8E-X7Ny_sNMYM-8Uw_f4_h3wCwbuh6KyfLR-G_tAjFM,2586
+ai_security_toolkit/modules/label_flip_attack.py,sha256=xB3elmpFZmp_neFlDb25hQW7mFA7_BChNsd9ZCPKqME,3309
+ai_security_toolkit/modules/membership_inference_attack.py,sha256=zWfUWDC_R8OF2cV5an-aUWMWikKHU67cRxcluqsn_zM,2561
+ai_security_toolkit/modules/simulate_inversion.py,sha256=UrHbesrNpKtsiaCwwsDuEqNfEavpcTUUm65NfNPQmSk,2622
+ai_security_toolkit/modules/steal_model.py,sha256=1o_uOO47wwvf9XOb9fVa-NmFB39LgI-4lgt2TvNrcas,2795
+ai_security_toolkit/modules/train_mnist_model.py,sha256=I_fBLHLMkQrysB9rMKLZspHLQidATsRk3GCBBEv_Hd4,1799
+ai_security_toolkit/shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_security_toolkit/shared/log_utils.py,sha256=no2_W1BPTy7K4ETqpWbYXYm1RrnpSqQI2zsw_puY0JU,1413
+ai_security_toolkit-1.1.0.dist-info/METADATA,sha256=Lx0xFLEozx3x3H7cPF6Nl3VXqS18jwJ0p2g8GWC0DeU,2419
+ai_security_toolkit-1.1.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+ai_security_toolkit-1.1.0.dist-info/entry_points.txt,sha256=mE4BdYeHEwNrXf21cNPa3qMNB4VGUq3JlBGGeqYVFuA,60
+ai_security_toolkit-1.1.0.dist-info/top_level.txt,sha256=DPMp0X7MzVKm6szxcG6toEN5S3Glc30SkJkjM7XbYpw,20
+ai_security_toolkit-1.1.0.dist-info/RECORD,,

ai_security_toolkit-1.0.0.dist-info/RECORD

@@ -1,17 +0,0 @@
-ai_security_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_security_toolkit/run.py,sha256=2bKlLZs-yfcjBDZx-37TF18Zoyj5DckrpJZH9IV7HcI,1900
-ai_security_toolkit/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_security_toolkit/modules/backdoor_trigger_attack.py,sha256=FSE1LkPoLbXuEQXq2vReyM9R4tDbBZ76BxndE4i9DUs,4408
-ai_security_toolkit/modules/fgsm_mobilenet.py,sha256=6em43Ymcgg_vlHuSHvAtpnJv-PDSmu9pxG-5gZyCi-s,2429
-ai_security_toolkit/modules/label_flip_attack.py,sha256=YQ5OK2RQAMcRY3pTuxOHLeihR4XlbOSp-Xl_tgpmrNo,3144
-ai_security_toolkit/modules/membership_inference_attack.py,sha256=3UsR2-yf8YyoCNJa_HcaVw8cwFKSkAEU8O9hGFT3dOs,2428
-ai_security_toolkit/modules/simulate_inversion.py,sha256=dea4l13AvhcSY-a5wphQukjdMsmkep0EdsPylSpBTAk,2495
-ai_security_toolkit/modules/steal_model.py,sha256=-svDT9YFve6WbdXGzcQdGwssgcrWruRg5oQC7Ycf9UA,2628
-ai_security_toolkit/modules/train_mnist_model.py,sha256=e-yomDzN4PTiOBQCOU500fj3bpgc4siWXMnuCkSEQIE,1640
-ai_security_toolkit/shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_security_toolkit/shared/log_utils.py,sha256=no2_W1BPTy7K4ETqpWbYXYm1RrnpSqQI2zsw_puY0JU,1413
-ai_security_toolkit-1.0.0.dist-info/METADATA,sha256=7-UQ3PZj6m5iThxSByWECUeaQBIvXYcH6QL8Yf80A-4,2141
-ai_security_toolkit-1.0.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-ai_security_toolkit-1.0.0.dist-info/entry_points.txt,sha256=mE4BdYeHEwNrXf21cNPa3qMNB4VGUq3JlBGGeqYVFuA,60
-ai_security_toolkit-1.0.0.dist-info/top_level.txt,sha256=DPMp0X7MzVKm6szxcG6toEN5S3Glc30SkJkjM7XbYpw,20
-ai_security_toolkit-1.0.0.dist-info/RECORD,,