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: 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

app.py

@@ -4,71 +4,70 @@ from pathlib import Path
 from huggingface_hub import snapshot_download
 
 PRIVATE_REPO = "cybercentinel/ai-red-teaming"
-CACHE = Path("cache")
+CACHE_DIR = Path("/tmp/sentitrust_cache")
 
-def load_toolkit():
-    """Load the full SentiTrust AI Red-Teaming Toolkit from private repository."""
-    try:
-        print("Loading SentiTrust AI Security Toolkit...")
-        CACHE.mkdir(exist_ok=True)
-        
-        # Download the complete toolkit from private space
-        snapshot_download(
-            repo_id=PRIVATE_REPO,
-            repo_type="space",
-            local_dir=CACHE,
-            token=os.environ.get("HF_TOKEN")
-        )
-        
-        # Add cache to path and import the main app
-        sys.path.insert(0, str(CACHE))
-        
-        # 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
+# Download the private space
+print("Loading SentiTrust AI Security Toolkit...")
+CACHE_DIR.mkdir(parents=True, exist_ok=True)
+
+try:
+    snapshot_download(
+        repo_id=PRIVATE_REPO,
+        repo_type="space",
+        local_dir=CACHE_DIR,
+        token=os.environ.get("HF_TOKEN")
+    )
+    
+    print("Starting AI Red-Teaming Toolkit...")
+    
+    # Add to path and execute the downloaded app
+    sys.path.insert(0, str(CACHE_DIR))
+    
+    # Execute the app.py file directly
+    app_path = CACHE_DIR / "app.py"
+    with open(app_path) as f:
+        exec(f.read(), {"__name__": "__main__"})
         
-        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;">
-                <h1 style="color: #dc2626; font-size: 2.5em; margin-bottom: 20px;">
-                    SentiTrust AI Security Toolkit
-                </h1>
-                <p style="color: rgba(255,255,255,0.8); font-size: 1.2em; margin-bottom: 30px;">
-                    The toolkit is temporarily unavailable. Please try again in a moment.
-                </p>
-                <p style="color: rgba(255,255,255,0.6); font-size: 0.95em; margin-bottom: 40px;">
-                    If the issue persists, please contact our team.
-                </p>
-                <div style="margin-top: 30px;">
-                    <a href="https://sentitrust.adogent.com" target="_blank" 
-                       style="display: inline-block; background: #dc2626; color: white; padding: 16px 32px; 
-                              border-radius: 8px; text-decoration: none; font-weight: 700; margin: 8px; font-size: 1.1em;">
-                        Visit SentiTrust.com →
-                    </a>
-                    <a href="https://calendly.com/sentitrust/free-security-scan" target="_blank" 
-                       style="display: inline-block; background: white; color: #0a0a0a; padding: 16px 32px; 
-                              border-radius: 8px; text-decoration: none; font-weight: 700; margin: 8px; font-size: 1.1em;">
-                        Book Free Scan →
-                    </a>
-                </div>
-                <p style="color: rgba(255,255,255,0.4); font-size: 0.85em; margin-top: 40px;">
-                    hello@adogent.com
-                </p>
+except Exception as e:
+    # Show professional error page if loading fails
+    print(f"Error loading toolkit: {e}")
+    import traceback
+    traceback.print_exc()
+    
+    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;">
+            <h1 style="color: #dc2626; font-size: 2.5em; margin-bottom: 20px;">
+                SentiTrust AI Security Toolkit
+            </h1>
+            <p style="color: rgba(255,255,255,0.8); font-size: 1.2em; margin-bottom: 30px;">
+                The toolkit is temporarily unavailable. Please try again in a moment.
+            </p>
+            <p style="color: rgba(255,255,255,0.6); font-size: 0.95em; margin-bottom: 40px;">
+                If the issue persists, please contact our team.
+            </p>
+            <div style="margin-top: 30px;">
+                <a href="https://sentitrust.adogent.com" target="_blank" 
+                   style="display: inline-block; background: #dc2626; color: white; padding: 16px 32px; 
+                          border-radius: 8px; text-decoration: none; font-weight: 700; margin: 8px; font-size: 1.1em;">
+                    Visit SentiTrust.com →
+                </a>
+                <a href="https://calendly.com/sentitrust/free-security-scan" target="_blank" 
+                   style="display: inline-block; background: white; color: #0a0a0a; padding: 16px 32px; 
+                          border-radius: 8px; text-decoration: none; font-weight: 700; margin: 8px; font-size: 1.1em;">
+                    Book Free Scan →
+                </a>
             </div>
-            """)
-        
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False
-        )
-
-if __name__ == "__main__":
-    load_toolkit()
+            <p style="color: rgba(255,255,255,0.4); font-size: 0.85em; margin-top: 40px;">
+                hello@adogent.com
+            </p>
+        </div>
+        """)
+    
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )

requirements.txt

@@ -1,2 +1,2 @@
-gradio==4.19.0
-huggingface-hub<0.23.0
+gradio==5.49.1
+huggingface-hub>=0.30.0