update command to ai run target_url

Author: root0zaki

README.md

@@ -9,23 +9,23 @@ AI security testing framework with terminal-first execution and GitHub Pages dem
 ## Primary command
 
 ```bash
-./ai run <url>
+ai run <target_url>
 ```
 
 Example sample endpoints:
 
 ```bash
-./ai run https://prompt-injection-lab.vercel.app/api/chat
-./ai run https://api-inference.huggingface.co/models/gpt2
-./ai run https://aihorde.net/api/v2
-./ai run https://chat.openai.com/backend-api/models
+ai run https://prompt-injection-lab.vercel.app/api/chat
+ai run https://api-inference.huggingface.co/models/gpt2
+ai run https://aihorde.net/api/v2
+ai run https://chat.openai.com/backend-api/models
 ```
 
 ## Technical structure
 
 ```text
 /Users/harshitaaa/Projects/AI-Pentest
-├── ai                              # launcher for ./ai run <url>
+├── ai                              # launcher for ai run <target_url>
 ├── main.py                         # command parser + run orchestration
 ├── demo/
 │   ├── __init__.py
@@ -48,7 +48,7 @@ Example sample endpoints:
 
 ## Runtime flow
 
-1. Parse `ai run <url>`.
+1. Parse `ai run <target_url>`.
 2. Print helper prelude (project, link, command, sample URLs).
 3. Discover testcase folders containing all required files:
    - `discovery.py`

demo/index.html

@@ -154,7 +154,7 @@
       line('AI Pentest REPL Demo', 'muted');
       line('Project: AI-Pentest', 'muted');
       line('GitHub: https://github.com/harshitaaa/AI-Pentest', 'muted');
-      line('Command: ai run <url>', 'muted');
+      line('Command: ai run <target_url>', 'muted');
       line('Sample URLs:', 'muted');
       line('  - https://prompt-injection-lab.vercel.app/api/chat', 'muted');
       line('  - https://api-inference.huggingface.co/models/gpt2', 'muted');
@@ -305,7 +305,7 @@
         line('Commands:', 'other');
         line('  help', 'other');
         line('  clear', 'other');
-        line('  ai run <url>', 'other');
+        line('  ai run <target_url>', 'other');
         spacer();
         return;
       }
@@ -317,7 +317,7 @@
       }
 
       if (parsed.kind === 'bad') {
-        line('Invalid command. Use: ai run <url>', 'vuln');
+        line('Invalid command. Use: ai run <target_url>', 'vuln');
         spacer();
         return;
       }

demo/ui.py

@@ -26,7 +26,7 @@ def print_banner() -> None:
 def show_repl_prelude() -> None:
     print(color("Project: AI-Pentest", GRAY))
     print(color("GitHub: https://github.com/harshitaaa/AI-Pentest", GRAY))
-    print(color("Command: ai run <url>", GRAY))
+    print(color("Command: ai run <target_url>", GRAY))
     print(color("Examples:", GRAY))
     print(color("  - ai run https://prompt-injection-lab.vercel.app/api/chat", GRAY))
     print(color("  - ai run https://api-inference.huggingface.co/models/gpt2", GRAY))

index.html

@@ -16,7 +16,7 @@
   <div class="wrap">
     <div class="card">
       <h1>AI PENTEST DEMO</h1>
-      <p>Primary command: <code>./ai run &lt;url&gt;</code></p>
+      <p>Primary command: <code>ai run &lt;url&gt;</code></p>
       <p>Live demo page: <a href="https://imharshitaa.github.io/AI-Pentest/demo/">https://imharshitaa.github.io/AI-Pentest/demo/</a></p>
     </div>
   </div>

main.py

@@ -15,8 +15,8 @@ def _build_parser() -> argparse.ArgumentParser:
         description="AI pentest testcase runner",
         epilog=(
             "Samples:\n"
-            "  ./ai run https://prompt-injection-lab.vercel.app/api/chat\n"
-            "  ./ai run https://api-inference.huggingface.co/models/gpt2 --report reports/hf.json\n"
+            "  ai run https://prompt-injection-lab.vercel.app/api/chat\n"
+            "  ai run https://api-inference.huggingface.co/models/gpt2 --report reports/hf.json\n"
             "  python3 main.py run offline://lab\n"
         ),
         formatter_class=argparse.RawDescriptionHelpFormatter,
@@ -28,9 +28,9 @@ def _build_parser() -> argparse.ArgumentParser:
         help="Run all pentest cases against a target URL",
         epilog=(
             "Examples:\n"
-            "  ./ai run https://prompt-injection-lab.vercel.app/api/chat\n"
-            "  ./ai run https://aihorde.net/api/v2\n"
-            "  ./ai run https://chat.openai.com/backend-api/models --quiet\n"
+            "  ai run https://prompt-injection-lab.vercel.app/api/chat\n"
+            "  ai run https://aihorde.net/api/v2\n"
+            "  ai run https://chat.openai.com/backend-api/models --quiet\n"
         ),
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )
@@ -63,7 +63,7 @@ def _interactive_args() -> argparse.Namespace:
     show_vulnerable_targets_hint(load_vulnerable_target_urls(history_path))
     command_line = input("Command (example: ai run https://target-ai.example): ").strip()
     if not command_line:
-        raise SystemExit("No command provided. Use `./ai --help` for usage.")
+        raise SystemExit("No command provided. Use `ai --help` for usage.")
     tokens = shlex.split(command_line)
     if tokens and tokens[0] == "ai":
         tokens = tokens[1:]
@@ -83,7 +83,7 @@ def main() -> None:
         print_banner()
 
     if args.command not in {"run", "pentest"}:
-        raise SystemExit("Unknown command. Use `./ai --help`.")
+        raise SystemExit("Unknown command. Use `ai --help`.")
 
     target = {
         "source": "any",