Team Ai
Apppublic

sameersyed/Defence_FrameWork

sourceHugging Faceupdated 8mo agoView on Hugging Face
0likes
analysis_engine.cpython-313.pyc102 linesDownload Raw Back to __pycache__
1�

2#iw+��<�SSKrSSKrSSKJrJrJr "SS5rg)�N)�Dict�List�Tuplec���\rSrSrS\4SjrS\S\4SjrS\S\4SjrS\S\4S	jr	S\S\3\4S4jrS\S\5\S\4Sjr
S\6\S\7\4S
jrS\8\S\9\4SjrS\S\10\S\11\S\4SjrSrSrg)�SecurityAnalysisEngine��knowledge_base_pathc��[US5n[R"U5UlSSS5 g!,(df   g=f)N�r)�open�json�load�kb)�selfr	�fs   �*D:\Security_app\backend\analysis_engine.py�__init__�SecurityAnalysisEngine.__init__s*��
�%�s�
+�q��i�i��l�D�G�,�
+�
+�s	�2�12A�13user_query�returnc��UR5nURU5nURU5nURU5nUR	X%5nUUUUUS.$)z5Extract intent from user query using keyword matching)�14model_type�domain�threats�query�15confidence)�lower�_detect_model_type�_detect_domain�_detect_threats�_calculate_confidence)rr�query_lowerrrrrs       r�analyze_intent�%SecurityAnalysisEngine.analyze_intent16ss�� �&�&�(���,�,�[�9�17��$�$�[�1���&�&�{�3���/�/��E�18�%����$�19�	20�rc�^�/SQ/SQ/SQ/SQ/SQS.nUR5H#up4[U4SjU55(dM!Us $ g)	z!Identify ML model type from query)�cnn�
convolutional�resnet�vggzimage classifier)�rnn�lstm�gru�	recurrent)�transformer�bert�gpt�	attention�llm)�gan�21generative�	generator�
discriminator)�
reinforcement�rlz22q-learning�policy)r'r+r/r4r8c3�,># �UH	oT;v� M g7f�N���.0�kwrs  �r�	<genexpr>�<SecurityAnalysisEngine._detect_model_type.<locals>.<genexpr>-�����2��2��;�����general��items�any)rr�model_keywords�model�keywordss `   rr�)SecurityAnalysisEngine._detect_model_type"sN���Q�6�M�F�L�23�� .�3�3�5�O�E��2��2�2�2��� 6�r%c�^�/SQ/SQ/SQ/SQ/SQS.nUR5H#up4[U4SjU55(dM!Us $ g)	zIdentify application domain)�image�visionzobject detection�facialzautonomous vehicle)�text�nlp�language�	sentiment�chatbot)�medical�	diagnosis�patient�24healthcare)�	financial�fraud�trading�banking)�iot�sensor�embeddedzedge device)�computer_visionrRrY�financer^c3�,># �UH	oT;v� M g7fr<r=r>s  �rrA�8SecurityAnalysisEngine._detect_domain.<locals>.<genexpr><rCrDrErF)rr�domainsrrKs `   rr�%SecurityAnalysisEngine._detect_domain1sL��� g�F�K�C�?�25��!(�
�
���F��2��2�2�2��
�!0�r%c�D^�/nURSR5HVup4USnUSn[U4SjU55(d[U4SjU55(dMEURU5 MX U(d[U4SjS55(aSS	/nU$)26z.Identify potential security threats from query�attack_typesrK�	scenariosc3�,># �UH	oT;v� M g7fr<r=r>s  �rrA�9SecurityAnalysisEngine._detect_threats.<locals>.<genexpr>IrCrDc3�,># �UH	oT;v� M g7fr<r=)r?�scrs  �rrArkIs����9Z�PY�"��+�PY�rDc3�,># �UH	oT;v� M g7fr<r=)r?�wordrs  �rrArkNs����d�-c�T�5�=�-c�rD)�secure�protect�safe�
vulnerability�risk�adversarial_attacks�data_poisoning)rrGrH�append)rr�detected�	threat_id�threat_datarKris `     rr �&SecurityAnalysisEngine._detect_threats@s������&*�g�g�n�&=�&C�&C�&E�"�I�"�:�.�H�#�K�0�I��2��2�2�2�c�9Z�PY�9Z�6Z�6Z����	�*�
'F���d�-c�d�d�d�1�3C�D���r%rc�&^�U(dgSnSnUHRnXPRS;dMURSUSnU[U5-
nU[U4SjU55-
nMT US:Xag[XC-S-5n[	S[SU55$)z3Calculate confidence score based on keyword matches�2rrhrKc3�6># �UHoT;dM27Sv� M g7f)�Nr=r>s  �rrA�?SecurityAnalysisEngine._calculate_confidence.<locals>.<genexpr>_s����'L�H�b�e����H�s�	�	�d�_)r�len�sum�int�min�max)rrr�total_keywords�matched_keywordsryrKrs `      rr!�,SecurityAnalysisEngine._calculate_confidenceSs���������� �I��G�G�N�3�3��7�7�>�2�9�=�j�I���#�h�-�/�� �C�'L�H�'L�$L�L� �	!��Q����*�;�s�B�C�28��2�s�2�z�*�+�+r%c	��/nUHLnX0RS;dMURSUnURUUSUSUSUSS.5 MN U$)z1Get detailed information about identified threatsrh�name�description�29risk_levelri)�idr�r�r�ri)rrw)rr�threat_detailsry�threats     r�classify_threats�'SecurityAnalysisEngine.classify_threatsgsp���� �I��G�G�N�3�3�����0��;���%�%�#�"�6�N�#)�-�#8�"(��"6�!'��!4�'��!��r%c	�~^�[5nUH=nX0RS;dMURSUSnURU5 M? /nUHLnX`RS;dMURSUnURUUSUSUSUSS.5 MN SS	S30S.mUR	U4SjS
S9 U$)z-Map threats to appropriate defense mechanismsrh�defensesr�r��
effectiveness�31complexity)r�r�r�r�r���r)�high�medium�lowc�.>�TRUSS5$)Nr�r)�get)�x�effectiveness_orders �r�<lambda>�;SecurityAnalysisEngine.recommend_defenses.<locals>.<lambda>�s���$7�$;�$;�A�o�<N�PQ�$Rr%T)�key�reverse)�setr�updaterw�sort)	rr�defense_idsry�threat_defensesr��32defense_id�defenser�s	        @r�recommend_defenses�)SecurityAnalysisEngine.recommend_defensesxs�����e�� �I��G�G�N�3�3�"&�'�'�.�"9�)�"D�Z�"P���"�"�?�3�!�33��%�J��W�W�Z�0�0��'�'�*�-�j�9�����$�#�F�O�#*�=�#9�%,�_�%=�")�,�"7�!��&�()�A�a�@���
�
�R�\`�
�a��r%�intentr�c34�	�/nU(dgURS5 URSUR5S35 URSUSRSS5R5S35 URS	US35R	5S35 URSURSS
5S35 URS5 [
SU55n[
SU55nUS:�aURSUS35 US:�aURSUS35 URS5 [US5GHupxUSS:XaSOUSS:XaSOSn	URU	SUSUS S35 URS!USR	5S35 URS"URS#S$5R	5S%35 URS&US'S%35 URS(5(aURS)US(S%35 US*(dM�S+RUS*S,S-5n36URS.U37S%35 GM URS5 U(a�URS/5 [US,S0S5H�up{US1S:XaS2OUS1S3:XaS4OS5nUS6S7:XaSOUS6S3:XaS8OS9n
URUSUS SUS35 URS:US1R5S;US6R5SU
S%35 URS<US'S%35 URS=5(dM�URS>US=S%35 M� URS5 URS?5 US:�aDURS@5 URSA5 URSB5 URSC5 URSD5 URSE5 URSF5 URSG5 URSH5 URSI5 URSJ5 URSK5 URS5 URSL5 URSM5 URSN5 URSO5 SPRU5$)Qz<Generate clean, structured response without markdown symbolsu�🔒 ShieldML Security Advisor38 39I can help you analyze security risks in ML/DL systems.40 41Ask me about:42• Adversarial attacks on your model43• Data poisoning and backdoor detection44• Privacy attacks (membership inference)45• Model extraction/stealing46• Physical-world attacks47• Defense strategies and implementations48 49Example: "How to protect my CNN from adversarial attacks?"50u"🔒 Real-Time Security Analysis51 52zAnalysis Time: �53zDomain: r�_� zModel Type: rzConfidence: r�Kz%54 55u�────────────────────────────────────────────────────────────56 57c3�:# �UHoSS:XdM
Sv� M g7f)r��criticalrNr=�r?�ts  rrA�;SecurityAnalysisEngine.generate_response.<locals>.<genexpr>�s���Q��1�\�?�j�3P�Q�Q�����	c3�:# �UHoSS:XdM
Sv� M g7f)r�r�rNr=r�s  rrAr��s���I�G�q���&�/H���G�r�ru🚨 ALERT: z CRITICAL threat(s) detected58 59u⚠️ WARNING: z HIGH-risk threat(s) detected60 61u⚠️ Detected Threats:62 63rr�r�u🚨r�u⚠️u⚡z. r�z   Risk Level: z   Attack Phase: �attack_phasezN/Az64 65z   Description: r��real_world_impactz   Real-World Impact: riz, N�z   Common in: u3🛡️ Defense Recommendations (Priority Order):66 67rr�u🟢r�u🟡u🔴r�r�u⚙️u🔧z   Effectiveness: z | Complexity: z   �implementationz   Implementation: u✅ Immediate Action Items:68 69zURGENT (Do Now):70u+• Halt production deployment immediately71u&• Audit training data for poisoning72u0• Implement critical defenses before release73 74zShort-term (1-2 weeks):75u)• Implement top 3 recommended defenses76u"• Set up monitoring and logging77u5• Test model robustness with adversarial examples78 79zLong-term (1-3 months):80u(• Establish security testing pipeline81u• Regular security audits82u.• Train team on ML security best practices83 84u📚 Need More Help?85 86u3• Upload CSV dataset for automated risk analysis87u5• Upload images to detect spam/adversarial content88u3• Ask follow-up questions for specific scenarios89�)90rw�_get_timestamp�replace�title�upperr�r��	enumerate�join�91capitalize)rr�rr��response_parts�critical_count�92high_count�ir��93risk_emojirir��	eff_emoji�94comp_emojis              r�generate_response�(SecurityAnalysisEngine.generate_response�s7�����
� 	���D�E������0C�0C�0E�/F�b�I�J�������)9�)A�)A�#�s�)K�)Q�)Q�)S�(T�TV�W�X�����V�L�-A�-G�-G�-I�,J�"�M�N�����V�Z�Z��b�-I�,J�%�P�Q����1�2��Q��Q�Q���I�G�I�I�95��A���!�!�L��0@�@`�"a�b���>��!�!�$4�Z�L�@a�"b�c�	���<�=�"�7�A�.�I�A�#)�,�#7�:�#E��W]�^j�Wk�ou�Wu�8�|A�J��!�!�Z�L��!��B�v�f�~�6F�b�"I�J��!�!�O�F�<�4H�4N�4N�4P�3Q�QS�"T�U��!�!�$5�f�j�j��QV�6W�6]�6]�6_�5`�`d�"e�f��!�!�$4�V�M�5J�4K�4�"P�Q��z�z�-�.�.��%�%�(>�v�FY�?Z�>[�[_�&`�a��k�"�"� �I�I�f�[�&9�"�1�&=�>�	��%�%��y�k��&F�G�/�	���1�2���!�!�"Y�Z�'���!��a�8�96��&-�o�&>�&�&H�F�X_�`o�Xp�t|�X|�f�CI�	�&-�l�&;�u�&D�U�V]�^j�Vk�ow�Vw�(�~D�97��%�%���2�g�f�o�->�a�	�{�"�&M�N��%�%�(:�7�?�;S�;^�;^�;`�:a�ap�qx�zF�rG�rR�rR�rT�qU�UV�Wa�Vb�bf�'g�h��%�%��G�M�,B�+C�4�&H�I��;�;�/�0�0�"�)�)�,?��HX�@Y�?Z�Z^�*_�`�9�	���1�2�	���?�@��A���!�!�"6�7��!�!�"P�Q��!�!�"K�L��!�!�"V�W����9�:����J�K����C�D����W�X����9�:����I�J����=�>����P�Q����1�2�	���8�9����T�U����V�W����T�U��w�w�~�&�&r%c�L�SSKJn UR5RS5$)zGet current timestampr)�datetimez%Y-%m-%d %H:%M:%S)r��now�strftime)rr�s  rr��%SecurityAnalysisEngine._get_timestamp�s��%��|�|�~�&�&�':�;�;r%)rN)�__name__�98__module__�__qualname__�__firstlineno__�strrrr#rrrr r�r!r�r�r�r��__static_attributes__r=r%rrrs���#�C�#�99��100��101�0
��
��
�
�C�
�C�
��S��T�#�Y��&,�3�,��c��,�s�,�(��S�	��d�4�j��"�$�s�)���T�102��6e'��e'�t�D�z�e'�T�RV�Z�e'�\_�e'�N<r%r)r
�re�typingrrrrr=r%r�<module>r�s���	�$�$�x<�x<r%