Skip to content

Repository files navigation

title AI Pentest API
emoji πŸ›‘οΈ
colorFrom purple
colorTo blue
sdk docker
app_port 7860
pinned false

πŸ›‘οΈ AI Pentest Scanner

Live Demo License

An autonomous, AI-powered penetration testing tool that automates vulnerability discovery and reporting

πŸ“‹ Table of Contents

πŸ“– About

This folder contains the clean, standalone version of the AI Pentest Scanner. It is separated from legacy bot components for easier management.

The AI Pentest Scanner leverages multi-agent AI systems to perform comprehensive security scanning, including reconnaissance, vulnerability detection, and automated report generation.

✨ Features

  • πŸ€– Multi-Agent AI System - Specialized agents for different pentesting tasks
  • 🎯 Automated Reconnaissance - Intelligent target enumeration and information gathering
  • πŸ” Vulnerability Scanning - Automated detection of security weaknesses
  • πŸ“Š Auto-Report Generation - Professional reports in Markdown, HTML, JSON, and PDF
  • πŸ–₯️ Web UI - User-friendly interface for managing scans
  • πŸ’» CLI Support - Run scans directly from terminal

πŸ› οΈ Tech Stack

Frontend:

  • React (Vite)
  • Tailwind CSS

Backend:

  • Python Flask API
  • Groq API (LLM)
  • Gemini API (LLM)

Tools:

  • Conda (Environment Management)
  • npm (Package Manager)

πŸ“¦ Installation

Prerequisites

  • Python 3.8+
  • Node.js 16+
  • Conda

Setup

  1. Clone the repository:
git clone https://github.com/samarth70/AI-Pentest-Scanner.git
cd AI-Pentest-Scanner
  1. Install backend dependencies:
conda create -n agenticAi python=3.10
conda activate agenticAi
pip install -r requirements.txt
  1. Install frontend dependencies:
cd frontend
npm install
  1. Configure environment variables: Create a .env file in the root directory:
GROQ_API_KEY=your_groq_api_key
GEMINI_API_KEY=your_gemini_api_key

πŸš€ Usage

Option 1: Web Interface (Recommended)

  1. Start the Backend API:
conda activate agenticAi
python api_server.py

The API will start at http://localhost:5000.

  1. Start the Frontend: Open another terminal:
cd frontend
npm run dev

The UI will start at http://localhost:5173.

  1. Access the Application: Open your browser and navigate to http://localhost:5173

Option 2: CLI Mode

Run scans directly from the terminal:

python main.py --target example.com

Additional CLI options:

python main.py --target example.com --scan-type full --output-format pdf

πŸ“š Documentation

For detailed architecture, design decisions, and technical specifications, see:

πŸ“ Directory Structure

AI-Pentest-Scanner/
β”œβ”€β”€ api_server.py          # Flask backend for the web UI
β”œβ”€β”€ main.py                # CLI entry point
β”œβ”€β”€ agents/                # AI agents for different pentesting tasks
β”‚   β”œβ”€β”€ recon_agent.py    # Reconnaissance agent
β”‚   β”œβ”€β”€ vuln_agent.py     # Vulnerability scanning agent
β”‚   └── report_agent.py   # Report generation agent
β”œβ”€β”€ core/                  # Core configurations and utilities
β”‚   β”œβ”€β”€ config.py         # Configuration settings
β”‚   └── security.py       # Security utilities
β”œβ”€β”€ frontend/              # Vite + React + Tailwind frontend
β”‚   β”œβ”€β”€ src/              # React source code
β”‚   β”œβ”€β”€ public/           # Static assets
β”‚   └── package.json      # Frontend dependencies
β”œβ”€β”€ reports/               # Generated scan results
β”‚   β”œβ”€β”€ markdown/         # Markdown reports
β”‚   β”œβ”€β”€ html/             # HTML reports
β”‚   β”œβ”€β”€ json/             # JSON reports
β”‚   └── pdf/              # PDF reports
β”œβ”€β”€ .env                   # Environment variables (API keys)
└── requirements.txt       # Python dependencies

πŸ”’ Security Note

This tool is intended for educational purposes and authorized security testing only. Always obtain proper authorization before scanning targets.

πŸ“„ License

This project is licensed under the MIT License - see the LICENSE file for details.

πŸ‘€ Author

Samarth Agarwal

Acknowledgments

  • Groq for providing fast LLM inference
  • Google Gemini for AI capabilities
  • The open-source security community

⚠️ Disclaimer: Use responsibly and only on systems you own or have explicit permission to test.

About

Autonomous multi-agent AI penetration-testing tool β€” recon, vulnerability detection and auto-reporting via Groq/Gemini agents with Nmap/Shodan. React + Flask + Cloudflare.

Topics

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages