🛡️ Zero-Trust GenAI Architect (RAG-Sec)

An Autonomous, Multi-Cloud (AWS/Azure) Posture Management System powered by Graph Machine Learning and Retrieval-Augmented Generation (RAG).

👨‍💻 Author

Designed and Developed by: Vamshi Batthula Email: batthulavamshi740@gmail.com

🚀 The 1-Crore FAANG Architecture

This project fundamentally re-architects how Enterprise Cloud Security functions. It replaces manual security audits with a decoupled, asynchronous Graph ML & GenAI Orchestration Pipeline capable of predicting and auto-patching toxic cloud configurations before they are exploited.

The Problem with Cloud Security

Scanning thousands of JSON/Terraform files manually is impossible. Sending massive Infrastructure-as-Code (IaC) states directly to an LLM (like GPT-4) causes extreme hallucination and incurs massive API token costs.

The RAG & Graph ML Solution

This architecture implements Predictive Zero-Trust by splitting the workload:

graph TD
    A[Cloud Infrastructure JSON/Terraform] -->|Ingestion| B(Graph ML Engine)
    B -->|NetworkX Edge Traversal| C{Detects Toxic Combination?}
    C -->|No| D[Drop Event]
    C -->|Yes| E[Alert Triggered]
    E --> F[RAG Vector Database]
    F -->|Retrieves Enterprise Policy| G[GenAI Auto-Patcher]
    G -->|Generates IaC Fix| H[Deploy Zero-Trust Patch]
    H --> I((Slack Webhook Alert))
    
    classDef secure fill:#10b981,stroke:#047857,stroke-width:2px,color:#fff;
    classDef alert fill:#ef4444,stroke:#b91c1c,stroke-width:2px,color:#fff;
    classDef ai fill:#8b5cf6,stroke:#6d28d9,stroke-width:2px,color:#fff;
    
    class H secure;
    class E alert;
    class B,G,F ai;

🕸️ Graph ML Scanner (NetworkX / Scikit-Learn)
- Parses AWS and Azure JSON payloads and converts the cloud infrastructure into a mathematical Directed Graph.
- Instantly traverses edges to detect Toxic Combinations (e.g., A Public S3 bucket directly attached to an Admin IAM Role).
- Drops 99.9% of normal network configurations locally on CPU for free.
🧠 RAG Security Engine (Vector Database)
- Triggered only when the Graph ML flags a toxic combination.
- Uses Retrieval-Augmented Generation (RAG) to query a local Vector DB containing strict Enterprise AWS/Azure Security Baselines.
- Prevents LLM hallucination by anchoring the prompt to approved corporate policies.
⚡ GenAI Auto-Patcher (Langchain)
- Dynamically writes the exact Infrastructure-as-Code (IaC) patch required to neutralize the threat.
- Automatically generates AWS Boto3 Lambda functions or Azure Management SDK scripts to enforce Least Privilege.

🖥️ The Threat Graph UI

The entire pipeline operates asynchronously via FastAPI and pushes live telemetry to a React dashboard via WebSockets. This creates a "Live Telemetry" UI where security teams can watch the Graph ML detect vulnerabilities and the GenAI deploy patches in real-time.

📁 Repository Structure

main.py: The Asynchronous FastAPI Gateway, Graph ML, and RAG logic.
simulate_scans.py: The CI/CD Pipeline Simulator (Generates mock AWS/Azure Infrastructure JSON).
/dashboard: The React + Vite WebSockets telemetry dashboard.
/terraform: AWS ECS Fargate deployment scripts (main.tf and Dockerfile).
INTERVIEW_MASTERCLASS.md: An exhaustive guide on how to explain this architecture for AI Engineer, ML Engineer, and Cloud Security roles at Google, Meta, Microsoft, Accenture, and Infosys.

⚙️ How to Run Locally

1. Start the FastAPI Backend

python -m venv venv
source venv/bin/activate  # (or venv\Scripts\activate on Windows)
pip install -r requirements.txt
uvicorn main:app --port 8000

2. Start the React Dashboard

cd dashboard
npm install
npm run dev -- --port 5174

3. Run the Infrastructure Simulator

python simulate_scans.py

Open http://localhost:5174 to watch the AI Engine detect and patch vulnerabilities in real-time.

FilesExpand file tree

README.md

Latest commit

History