Spaces:
Running
Running
| # 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) | |