Metadata-Version: 2.4
Name: animesub
Version: 1.0.3
Summary: Инструмент для автоматического создания субтитров для аниме
Author-email: Ivan Tyumentsev <ivanfufa184@gmail.com>
License: MIT
Project-URL: Homepage, https://github.com/iniquitousworld/animesub
Project-URL: Documentation, https://github.com/iniquitousworld/animesub#readme
Project-URL: Issues, https://github.com/iniquitousworld/animesub/issues
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Topic :: Multimedia :: Video :: Conversion
Classifier: Topic :: Text Processing :: Linguistic
Requires-Python: >=3.8
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch>=2.2.0
Requires-Dist: torchaudio>=2.2.0
Requires-Dist: faster-whisper>=1.2.0
Requires-Dist: transformers>=4.51.3
Requires-Dist: demucs>=4.0.0
Requires-Dist: punctuators>=0.0.5
Requires-Dist: soundfile>=0.12.1
Requires-Dist: numpy>=1.26.4
Requires-Dist: pyyaml>=6.0.2
Requires-Dist: tqdm>=4.67.1
Requires-Dist: sentencepiece>=0.2.0
Requires-Dist: ffmpeg-python>=0.2.0
Requires-Dist: diffq>=0.2.4
Requires-Dist: yt-dlp>=2025.1.26
Dynamic: license-file

# AnimeSub

Инструмент для **автоматического создания субтитров** из видео- или аудиофайлов.  
Оптимизирован для японского языка (подходит для аниме, интервью и т.п.).

---

## ⚡ Быстрый старт

**Локальный файл → субтитры:**
```bash
animesub -i input_file.mp4
```

**YouTube → субтитры:**
```bash
animesub -u "https://youtube.com/watch?v=XXXX"
```

**Запуск в Colab**
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iniquitousworld/animesub/blob/main/animesub_colab.ipynb)

---

## 📦 Установка

Из PyPI:
```bash
pip install animesub
```

💡 Для работы на CUDA желательно поставить `torch` и `torchaudio` под свою версию CUDA:
```bash
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu126
```

Не забудьте установить `ffmpeg` и `demucs`:
```bash
conda install ffmpeg -c conda-forge
pip install demucs
```

---

## 🚀 Использование

### Локальный файл
```bash
animesub -i input_file.mp4
```
Создаст `input_file.srt` в текущей папке.

### Скачивание по URL (YouTube и др.)
```bash
animesub -u "https://youtube.com/watch?v=XXXX"
```
Сохранит результат в `output.srt`.

---

## 🔧 Аргументы CLI

| Аргумент               | Описание |
|------------------------|----------|
| `-i`, `--input_file`   | Путь к локальному видео или аудио |
| `-u`, `--url`          | URL видео/аудио (YouTube) |
| `-o`, `--output`       | Путь к выходному `.srt` (по умолчанию: `<имя_файла>.srt` или `output.srt` для URL) |
| `-m`, `--model`        | Модель ASR: `tiny`, `base`, `small`, `medium`, `large`, `large-v2`, `large-v3`,`kotoba-faster`, `kotoba-whisper`, `kotoba-whisper-v2.2`, `kotoba-faster` (по умолчанию: `small`) |
| `-d`, `--device`       | `cpu` или `cuda` (по умолчанию определяется автоматически) |
| `--demucs-model`       | Модель сепарации вокала: `htdemucs` или `mdx_extra_q` (по умолчанию: `htdemucs`) |
| `--merge-silence`      | Максимальная пауза между VAD-сегментами для объединения (по умолчанию: `0.6`) |

---

## 📌 Примеры

**CPU + base модель**
```bash
animesub -i input.mp3 -m base -d cpu
```

**Kotoba-модель с кастомной паузой**
```bash
animesub -i anime.mkv -m kotoba-whisper-v2.2 --merge-silence 0.8 -d cuda
```

**С YouTube**
```bash
animesub -u "https://youtube.com/watch?v=XXXX" -m kotoba-faster -d cuda
```

**С указанием файла вывода**
```bash
animesub -i episode.mp4 -o subs/episode01.srt
```

---

## 🎯 Как работает

1. **Отделение вокала** (Demucs)  
2. **Детекция речи** (Silero VAD)  
3. **Транскрипция** (Whisper или Kotoba-Whisper)  
4. **Пунктуация** (XLM-RoBERTa через `punctuators`)  
5. **Экспорт в `.srt`** с форматированием

---

## 🛠️ Использование как библиотеки

```python
from AnimeSub.main_logic import process_audio

process_audio(
    input_path="video.mp4",
    output_path="subs.srt",
    model_name="kotoba-whisper-v2.2",
    device="cuda",
    merge_silence=0.6
)
```

---

## 📜 Лицензия

MIT

---

## 👤 Автор

**Ivan Tyumentsev**
📧 [ivanfufa184@gmail.com](mailto:ivanfufa184@gmail.com)
🔗 [GitHub](https://github.com/iniquitousworld)

```
