| # AI-Contain-Checker |
|
|
| A modular AI content detection system with support for **image classification**, **image edit detection**, **Nepali text classification**, and **general text classification**. Built for performance and extensibility, it is ideal for detecting AI-generated content in both visual and textual forms. |
|
|
|
|
| ## 🌟 Features |
|
|
| ### 🖼️ Image Classifier |
|
|
| * **Purpose**: Classifies whether an image is AI-generated or a real-life photo. |
| * **Model**: Fine-tuned **InceptionV3** CNN. |
| * **Dataset**: Custom curated dataset with **\~79,950 images** for binary classification. |
| * **Location**: [`features/image_classifier`](features/image_classifier) |
| * **Docs**: [`docs/features/image_classifier.md`](docs/features/image_classifier.md) |
|
|
| ### 🖌️ Image Edit Detector |
|
|
| * **Purpose**: Detects image tampering or post-processing. |
| * **Techniques Used**: |
|
|
| * **Error Level Analysis (ELA)**: Visualizes compression artifacts. |
| * **Fast Fourier Transform (FFT)**: Detects unnatural frequency patterns. |
| * **Location**: [`features/image_edit_detector`](features/image_edit_detector) |
| * **Docs**: |
|
|
| * [ELA](docs/detector/ELA.md) |
| * [FFT](docs/detector/fft.md ) |
| * [Metadata Analysis](docs/detector/meta.md) |
| * [Backend Notes](docs/detector/note-for-backend.md) |
|
|
| ### 📝 Nepali Text Classifier |
|
|
| * **Purpose**: Determines if Nepali text content is AI-generated or written by a human. |
| * **Model**: Based on `XLMRClassifier` fine-tuned on Nepali language data. |
| * **Dataset**: Scraped dataset of **\~18,000** Nepali texts. |
| * **Location**: [`features/nepali_text_classifier`](features/nepali_text_classifier) |
| * **Docs**: [`docs/features/nepali_text_classifier.md`](docs/features/nepali_text_classifier.md) |
|
|
| ### 🌐 English Text Classifier |
|
|
| * **Purpose**: Detects if English text is AI-generated or human-written. |
| * **Pipeline**: |
|
|
| * Uses **GPT2 tokenizer** for input preprocessing. |
| * Custom binary classifier to differentiate between AI and human-written content. |
| * **Location**: [`features/text_classifier`](features/text_classifier) |
| * **Docs**: [`docs/features/text_classifier.md`](docs/features/text_classifier.md) |
|
|
| --- |
|
|
| ## 🗂️ Project Structure |
|
|
| ```bash |
| AI-Checker/ |
| │ |
| ├── app.py # Main FastAPI entry point |
| ├── config.py # Configuration settings |
| ├── Dockerfile # Docker build script |
| ├── Procfile # Deployment file for Heroku or similar |
| ├── requirements.txt # Python dependencies |
| ├── README.md # You are here 📘 |
| │ |
| ├── features/ # Core detection modules |
| │ ├── image_classifier/ |
| │ ├── image_edit_detector/ |
| │ ├── nepali_text_classifier/ |
| │ └── text_classifier/ |
| │ |
| ├── docs/ # Internal and API documentation |
| │ ├── api_endpoints.md |
| │ ├── deployment.md |
| │ ├── detector/ |
| │ │ ├── ELA.md |
| │ │ ├── fft.md |
| │ │ ├── meta.md |
| │ │ └── note-for-backend.md |
| │ ├── functions.md |
| │ ├── nestjs_integration.md |
| │ ├── security.md |
| │ ├── setup.md |
| │ └── structure.md |
| │ |
| ├── IMG_Models/ # Saved image classifier model(s) |
| │ └── latest-my_cnn_model.h5 |
| │ |
| ├── notebooks/ # Experimental and debug notebooks |
| ├── static/ # Static assets if needed |
| └── test.md # Test notes |
| ```` |
|
|
| --- |
|
|
| ## 📚 Documentation Links |
|
|
| * [API Endpoints](docs/api_endpoints.md) |
| * [Deployment Guide](docs/deployment.md) |
| * [Detector Documentation](docs/detector/) |
|
|
| * [Error Level Analysis (ELA)](docs/detector/ELA.md) |
| * [Fast Fourier Transform (FFT)](docs/detector/fft.md) |
| * [Metadata Analysis](docs/detector/meta.md) |
| * [Backend Notes](docs/detector/note-for-backend.md) |
| * [Functions Overview](docs/functions.md) |
| * [NestJS Integration Guide](docs/nestjs_integration.md) |
| * [Security Details](docs/security.md) |
| * [Setup Instructions](docs/setup.md) |
| * [Project Structure](docs/structure.md) |
|
|
| --- |
|
|
| ## 🚀 Usage |
|
|
| 1. **Install dependencies** |
|
|
| ```bash |
| pip install -r requirements.txt |
| ``` |
|
|
| 2. **Run the API** |
|
|
| ```bash |
| uvicorn app:app --reload |
| ``` |
|
|
| 3. **Build Docker (optional)** |
|
|
| ```bash |
| docker build -t ai-contain-checker . |
| docker run -p 8000:8000 ai-contain-checker |
| ``` |
|
|
| --- |
|
|
| ## 🔐 Security & Integration |
|
|
| * **Token Authentication** and **IP Whitelisting** supported. |
| * NestJS integration guide: [`docs/nestjs_integration.md`](docs/nestjs_integration.md) |
| * Rate limiting handled using `slowapi`. |
|
|
| --- |
|
|
| ## 🛡️ Future Plans |
|
|
| * Add **video classifier** module. |
| * Expand dataset for **multilingual** AI content detection. |
| * Add **fine-tuning UI** for models. |
|
|
| --- |
|
|
| ## 📄 License |
|
|
| See full license terms here: [`LICENSE.md`](license.md) |
|
|
