Upload 3 files

README.md

@@ -1,62 +1,62 @@
----
-title: SentiTrust AI Security Toolkit
-emoji: 🛡️
-colorFrom: red
-colorTo: gray
-sdk: gradio
-sdk_version: 5.49.1
-app_file: app.py
-pinned: false
----
-
-# SentiTrust AI Red-Teaming Toolkitst AI Security Toolkit
-emoji: 🛡️
-colorFrom: red
-colorTo: gray
-sdk: gradio
-sdk_version: 4.19.0
-app_file: app.py
-pinned: false
----: SentiTrust AI Security Toolkit
-emoji: �
-colorFrom: red
-colorTo: gray
-sdk: gradio
-sdk_version: 4.19.0
-app_file: app.py
-pinned: false
----
-
-# SentiTrust AI Red-Teaming Toolkit
-
-Professional AI security testing platform for enterprise applications.
-
-## Capabilities
-
-**Framework-Aligned Testing**
-- Jailbreak vulnerability detection using industry-standard attack patterns
-- Prompt injection analysis with OWASP LLM Top 10 mapping
-- Automated vulnerability classification (MITRE ATLAS, NIST AI RMF)
-- Professional risk reports with CVSS scoring
-
-**Attack Techniques**
-- 9 jailbreak methods (DAN, AIM, role-play, system leak, constraint breaking)
-- 8 injection patterns (direct, indirect, context switching, delimiter escape)
-- Multi-stage attack simulation
-- Evidence tracking and reproducible testing
-
-## Professional Services
-
-Need expert security assessments for your AI application?
-
-**Express Audit** ($2,500) - 3-day rapid assessment  
-**Professional Assessment** ($8,500) - Comprehensive 2-week security review  
-**Enterprise Red Team** ($25,000+) - Full adversarial testing program
-
-■ [Visit SentiTrust.com](https://sentitrust.adogent.com)  
-■ [Book Free 15-Min Security Scan](https://calendly.com/sentitrust/free-security-scan)
-
----
-
-**Built by SentiTrust** - Professional AI security assessments for startups and enterprises.  
-Trusted by 50+ AI companies • Prevented $20M+ in potential breaches
+---
+title: SentiTrust AI Security Toolkit
+emoji: 🛡️
+colorFrom: red
+colorTo: gray
+sdk: gradio
+sdk_version: 4.19.0
+app_file: app.py
+pinned: false
+---
+
+# SentiTrust AI Red-Teaming Toolkitst AI Security Toolkit
+emoji: 🛡️
+colorFrom: red
+colorTo: gray
+sdk: gradio
+sdk_version: 4.19.0
+app_file: app.py
+pinned: false
+---: SentiTrust AI Security Toolkit
+emoji: �
+colorFrom: red
+colorTo: gray
+sdk: gradio
+sdk_version: 4.19.0
+app_file: app.py
+pinned: false
+---
+
+# SentiTrust AI Red-Teaming Toolkit
+
+Professional AI security testing platform for enterprise applications.
+
+## Capabilities
+
+**Framework-Aligned Testing**
+- Jailbreak vulnerability detection using industry-standard attack patterns
+- Prompt injection analysis with OWASP LLM Top 10 mapping
+- Automated vulnerability classification (MITRE ATLAS, NIST AI RMF)
+- Professional risk reports with CVSS scoring
+
+**Attack Techniques**
+- 9 jailbreak methods (DAN, AIM, role-play, system leak, constraint breaking)
+- 8 injection patterns (direct, indirect, context switching, delimiter escape)
+- Multi-stage attack simulation
+- Evidence tracking and reproducible testing
+
+## Professional Services
+
+Need expert security assessments for your AI application?
+
+**Express Audit** ($2,500) - 3-day rapid assessment  
+**Professional Assessment** ($8,500) - Comprehensive 2-week security review  
+**Enterprise Red Team** ($25,000+) - Full adversarial testing program
+
+■ [Visit SentiTrust.com](https://sentitrust.adogent.com)  
+■ [Book Free 15-Min Security Scan](https://calendly.com/sentitrust/free-security-scan)
+
+---
+
+**Built by SentiTrust** - Professional AI security assessments for startups and enterprises.  
+Trusted by 50+ AI companies • Prevented $20M+ in potential breaches

app.py

@@ -2,7 +2,6 @@ import os
 import sys
 from pathlib import Path
 from huggingface_hub import snapshot_download
-import importlib.util
 
 PRIVATE_REPO = "cybercentinel/ai-red-teaming"
 CACHE = Path("cache")
@@ -21,15 +20,20 @@ def load_toolkit():
             token=os.environ.get("HF_TOKEN")
         )
         
-        # Import and execute the main application
+        # Add cache to path and import the main app
         sys.path.insert(0, str(CACHE))
-        spec = importlib.util.spec_from_file_location("app", CACHE / "app.py")
-        app = importlib.util.module_from_spec(spec)
-        spec.loader.exec_module(app)
+        
+        # Import and run the app
+        print("Starting AI Red-Teaming Toolkit...")
+        import app as main_app
+        
+        # The app should auto-launch when imported since it has demo.launch() at module level
         
     except Exception as e:
         # Show professional error page if loading fails
+        print(f"Error loading toolkit: {e}")
         import gradio as gr
+        
         with gr.Blocks() as demo:
             gr.HTML("""
             <div style="background: #0a0a0a; padding: 60px 30px; border-radius: 12px; text-align: center; color: white; max-width: 800px; margin: 40px auto;">
@@ -59,7 +63,12 @@ def load_toolkit():
                 </p>
             </div>
             """)
-        demo.launch()
+        
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False
+        )
 
 if __name__ == "__main__":
     load_toolkit()