sameersyed/Defence_FrameWork
0
1�
2 77i� � �@ � S SK rS SKJrJr S SKrS SKr " S S5 rg)� N)�Image� ImageFilterc �R � \ rS rSrSrS rS S jrS
S jrSS jrSS jr S r3S r S4r g )�DefenseSimulator� z9Simulate defense mechanisms and show before/after resultsc �V � U R U R U R S.U l g )N��jpeg_compression�
gaussian_blur�bit_depth_reduction)�_jpeg_compression�_gaussian_blur�_bit_depth_reduction�defenses)�selfs �,D:\Security_app\backend\defense_simulator.py�__init__�DefenseSimulator.__init__ s& � � $� 6� 6� !� 0� 0�#'�#<�#<�5��
� c �* � SU; a UR S5 S n[ R " U5 n[ R " [6 R " U5 5 R S5 nU R U " U5 n[7 R " 5 nUR USS9 [ R " UR 5 5 R 5 n[ R " U5 n[ R " U5 n [ R " X�-8 S- 5 n9S[ R " SU10S - - 5 - n S11US U 3S U 3[# U125 [# U 5 S .U R% U5 S
.$ ! [& a n S[) U 5 S.s Sn A $ Sn A ff = f)z#Apply defense and return comparison�,� �RGB�PNG)�format� �13 i� g�����|�=Tzdata:image/png;base64,)�mse�psnr)�success� defense_type�original_image�defended_image�metrics�
effectivenessF)r �errorN)�split�base64� b64decoder �open�io�BytesIO�convertr �save� b64encode�getvalue�decode�np�array�mean�log10�float�_estimate_effectiveness� Exception�str)
r �14image_datar! � image_bytes�imager# �buffered�defended_base64�original_array�defended_arrayr r �es
r �
apply_defense� DefenseSimulator.apply_defense sg � �# 7��j� � '� -� -�c� 2�1� 5�15� �*�*�:�6�K��J�J�r�z�z�+� 6�7�?�?��F�E� "�]�]�<� 8�� ?�N� �z�z�|�H� � � ��� � 7� $� .� .�x�/@�/@�/B� C� J� J� L�O� �X�X�e�_�N� �X�X�n� 5�N��'�'�>�:�q�@�A�C������3��;�!7�8�8�D� � ,�$:�:�,�"G�$:�?�:K�"L� ��:� !�$�K� � "&�!=�!=�l�!K�16� 17
�� � 7� $�s�1�v�6� 6�� 7�s �E-E0 �018F�:
F
�F�
Fc � � [ R " 5 nUR USUS9 UR S5 [ R19 " U5 R
S5 $ )zJPEG compression defense�JPEG)r �qualityr r )r+ r, r. �seekr r* r- )r r< rF r= s r r
�"DefenseSimulator._jpeg_compression7 sE � ��:�:�<��
�20�21�8�F�G�22�<��
�
�a���z�z�(�#�+�+�E�2�2r c �H � UR [ R " US95 $ )zGaussian blur defense)�radius)�filterr � GaussianBlur)r r< rJ s r r � DefenseSimulator._gaussian_blur> s � ��|�|�K� 4� 4�F� C�D�Dr c � � [ R " U5 nSSU-23 - nX4- U- n[ R " UR [ R24 5 5 $ )zBit depth reduction defenser � )r2 r3 r � fromarray�astype�uint8)r r< �bits� img_array�factor�reduceds r r �%DefenseSimulator._bit_depth_reductionB sE � ��H�H�U�O� ��q�4�x� �� �&�&�0�����w�~�~�b�h�h� 7�8�8r c �J � SSS.SSS.SSS.S.nUR US SS.5 $ )25z Estimate defense effectiveness�\ �A )�accuracy_retention�attack_mitigation�X �F �U �< r �Z )�get)r r! r% s r r7 �(DefenseSimulator._estimate_effectivenessI s@ � � 8:�PR� S�46�R� P�:<�SU�#V�26�
�27 � � � �b�_a�/b�c�cr c � � US ( d SUS 3$ / nUR S5 UR S5 SSSS .nUR S28UR US US 5 S 35 UR S
5 UR SUS S S S35 UR SUS S S S35 US nUR S5 UR SUS S35 UR SUS S 35 UR S5 UR S 5 US S :� a UR S 5 OUR S!5 US S":� a UR S#5 OUR S$5 UR S%5 UR S&5 UR S'5 UR S(5 S)R U5 $ )*z"Generate defense comparison reportr u ❌ Defense simulation failed: r& u& 🛡️ Defense Mechanism Simulation29 30u� ────────────────────────────────────────────────────────────31 32z JPEG Compression (Quality=75)zGaussian Blur (Radius=1)zBit Depth Reduction (6-bit)r u 🔧 Applied Defense: r! z33 34u 📊 Image Quality Metrics:35u • MSE (Mean Squared Error): r$ r z.2f�36u' • PSNR (Peak Signal-to-Noise Ratio): r z dB37 38r% u 📈 Defense Effectiveness:39u • Accuracy Retention: r[ z%40u • Attack Mitigation: r\ z%41 42u 💡 Analysis:43ra uB • ✅ High accuracy retention - minimal impact on clean samples44uB • ⚠️ Moderate accuracy retention - some clean accuracy loss45rZ uJ • ✅ Strong attack mitigation - effective against adversarial examples46u@ • ⚠️ Moderate attack mitigation - partial protection only47u 48🎯 Recommendation:49u4 • Combine multiple defenses for better robustness50u5 • Use adversarial training alongside preprocessing51u8 • Monitor accuracy-robustness trade-off in production52� )�appendrb �join)r �result�report�
defense_names�effs r �generate_report� DefenseSimulator.generate_reportR s� � ��i� �4�V�G�_�4E�F� F����
�
�@�A��
�
�)�*� !@� 7�#@�53�
� �
�
�.�}�/@�/@���AW�Y_�`n�Yo�/p�.q�qu�v�w��
�
�5�6��
�
�6�v�i�7H��7O�PS�6T�TV�W�X��
�
�?��y�@Q�RX�@Y�Z]�?^�^e�f�g��_�%���
�
�5�6��
�
�0��5I�1J�0K�3�O�P��
�
�/��4G�0H�/I��O�P��
�
�)�*��
�
�(�)� �#� $�r� )� �M�M�_� `� �M�M�_� `� �"� #�b� (� �M�M�g� h� �M�M�]� ^��
�
�0�1��
�
�M�N��
�
�N�O��
�
�Q�R��w�w�v�� r )r N)r54 )�K )r )r )
�__name__�55__module__� __qualname__�__firstlineno__�__doc__r rB r
r r r7 rm �__static_attributes__� r r r r s+ � �C�56�%7�N3�E�9�d�, r r )�numpyr2 �PILr r r+ r( r rv r r �<module>ry s �� � "� �
�x � x r