Upload 3 files

README.md

@@ -1,12 +1,24 @@
----
-title: Sentitrust Toolkit
-emoji: 📊
-colorFrom: yellow
-colorTo: blue
-sdk: gradio
-sdk_version: 5.49.1
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+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 Toolkit
+
+Professional AI security testing platform.
+
+## Features
+- Jailbreak Testing
+- Prompt Injection Detection
+- Vulnerability Classification
+- Professional Reporting
+
+## Links
+- Website: https://sentitrust.adogent.com
+- Book Free Scan: https://calendly.com/sentitrust/free-security-scan

app.py

@@ -0,0 +1,31 @@
+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")
+
+def load_toolkit():
+    try:
+        print("Loading...")
+        CACHE.mkdir(exist_ok=True)
+        snapshot_download(
+            repo_id=PRIVATE_REPO,
+            repo_type="space",
+            local_dir=CACHE,
+            token=os.environ.get("HF_TOKEN")
+        )
+        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)
+    except Exception as e:
+        import gradio as gr
+        with gr.Blocks() as demo:
+            gr.Markdown(f"Error: {e}")
+        demo.launch()
+
+if __name__ == "__main__":
+    load_toolkit()

requirements.txt

@@ -0,0 +1,2 @@
+gradio==4.19.0
+huggingface-hub>=0.20.0