mkv-episode-matcher 0.3.6__tar.gz → 0.4.0__tar.gz

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mkv-episode-matcher
-Version: 0.3.6
+Version: 0.4.0
 Summary: The MKV Episode Matcher is a tool for identifying TV series episodes from MKV files and renaming the files accordingly.
 Home-page: https://github.com/Jsakkos/mkv-episode-matcher
 Author: Jonathan Sakkos
@@ -25,7 +25,18 @@ Requires-Dist: pytesseract>=0.3.13
 Requires-Dist: rapidfuzz>=3.10.1
 Requires-Dist: requests>=2.32.3
 Requires-Dist: tmdb-client>=0.0.1
+Requires-Dist: torch>=2.5.1
+Requires-Dist: torchaudio>=2.5.1
+Requires-Dist: torchvision>=0.20.1
 Requires-Dist: wave>=0.0.2
+Provides-Extra: cpu
+Requires-Dist: torch>=2.5.1; extra == "cpu"
+Requires-Dist: torchvision>=0.20.1; extra == "cpu"
+Requires-Dist: torchaudio>=2.5.1; extra == "cpu"
+Provides-Extra: cu124
+Requires-Dist: torch>=2.5.1; extra == "cu124"
+Requires-Dist: torchvision>=0.20.1; extra == "cu124"
+Requires-Dist: torchaudio>=2.5.1; extra == "cu124"
 
 # MKV Episode Matcher
 

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/docs/configuration.md

@@ -102,4 +102,8 @@ Adjust based on your system's capabilities:
 tesseract_path = /path/to/tesseract
 ```
 
-Required only if processing image-based subtitles.
+Required for processing image-based subtitles. Common paths:
+- Windows: `C:\Program Files\Tesseract-OCR\tesseract.exe`
+- Linux/macOS: `/usr/bin/tesseract`
+
+Make sure Tesseract OCR is properly installed before using this feature.

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/docs/installation.md

@@ -1,5 +1,23 @@
 # Installation Guide
 
+## Prerequisites
+
+### Required System Dependencies
+
+1. **FFmpeg**
+   - Windows: Download from [FFmpeg website](https://ffmpeg.org/download.html)
+   - Linux: `sudo apt install ffmpeg` or equivalent
+   - macOS: `brew install ffmpeg`
+   - Verify installation: `ffmpeg -version`
+
+2. **Tesseract OCR**
+   - Windows: Install from [UB-Mannheim](https://github.com/UB-Mannheim/tesseract/wiki)
+   - Linux: `sudo apt install tesseract-ocr` or equivalent
+   - macOS: `brew install tesseract`
+   - Verify installation: `tesseract --version`
+
+Make sure both FFmpeg and Tesseract are added to your system PATH.
+
 ## Basic Installation
 
 Install MKV Episode Matcher using pip:

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/mkv_episode_matcher/__main__.py

@@ -92,7 +92,18 @@ def main():
         nargs="?",
         help="Path to the tesseract executable (default: None)",
     )
+    parser.add_argument(
+        "--check-gpu",
+        type=bool,
+        default=False,
+        nargs="?",
+        help="Check if GPU is available (default: False)",
+    )
     args = parser.parse_args()
+    if args.check_gpu:
+        from mkv_episode_matcher.speech_to_text import check_gpu_support
+        check_gpu_support()
+        return
     logger.debug(f"Command-line arguments: {args}")
     open_subtitles_api_key = ""
     open_subtitles_user_agent = ""

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/mkv_episode_matcher/speech_to_text.py

@@ -87,4 +87,10 @@ def extract_audio(mkv_file, output_dir):
     else:
         logger.info(f"Audio file {wav_file} already exists, skipping extraction")
     
-    return wav_file
+    return wav_file
+def check_gpu_support():
+    logger.info('Checking GPU support...')
+    if torch.cuda.is_available():
+        logger.info(f"CUDA is available. Using GPU: {torch.cuda.get_device_name(0)}")
+    else:
+        logger.warning("CUDA not available. Using CPU. Refer to https://pytorch.org/get-started/locally/ for GPU support.")

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/mkv_episode_matcher.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mkv-episode-matcher
-Version: 0.3.6
+Version: 0.4.0
 Summary: The MKV Episode Matcher is a tool for identifying TV series episodes from MKV files and renaming the files accordingly.
 Home-page: https://github.com/Jsakkos/mkv-episode-matcher
 Author: Jonathan Sakkos
@@ -25,7 +25,18 @@ Requires-Dist: pytesseract>=0.3.13
 Requires-Dist: rapidfuzz>=3.10.1
 Requires-Dist: requests>=2.32.3
 Requires-Dist: tmdb-client>=0.0.1
+Requires-Dist: torch>=2.5.1
+Requires-Dist: torchaudio>=2.5.1
+Requires-Dist: torchvision>=0.20.1
 Requires-Dist: wave>=0.0.2
+Provides-Extra: cpu
+Requires-Dist: torch>=2.5.1; extra == "cpu"
+Requires-Dist: torchvision>=0.20.1; extra == "cpu"
+Requires-Dist: torchaudio>=2.5.1; extra == "cpu"
+Provides-Extra: cu124
+Requires-Dist: torch>=2.5.1; extra == "cu124"
+Requires-Dist: torchvision>=0.20.1; extra == "cu124"
+Requires-Dist: torchaudio>=2.5.1; extra == "cu124"
 
 # MKV Episode Matcher
 

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/mkv_episode_matcher.egg-info/requires.txt

@@ -7,4 +7,17 @@ pytesseract>=0.3.13
 rapidfuzz>=3.10.1
 requests>=2.32.3
 tmdb-client>=0.0.1
+torch>=2.5.1
+torchaudio>=2.5.1
+torchvision>=0.20.1
 wave>=0.0.2
+
+[cpu]
+torch>=2.5.1
+torchvision>=0.20.1
+torchaudio>=2.5.1
+
+[cu124]
+torch>=2.5.1
+torchvision>=0.20.1
+torchaudio>=2.5.1

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/pyproject.toml

@@ -32,6 +32,9 @@ dependencies = [
     "rapidfuzz>=3.10.1",
     "requests>=2.32.3",
     "tmdb-client>=0.0.1",
+    "torch>=2.5.1",
+    "torchaudio>=2.5.1",
+    "torchvision>=0.20.1",
     "wave>=0.0.2",
 ]
 [project.urls]
@@ -50,6 +53,49 @@ dev = [
     "chardet>=5.2.0",
 ]
 
+[project.optional-dependencies]
+cpu = [
+  "torch>=2.5.1",
+  "torchvision>=0.20.1",
+  "torchaudio>=2.5.1",
+]
+cu124 = [
+  "torch>=2.5.1",
+  "torchvision>=0.20.1",
+    "torchaudio>=2.5.1",
+]
+
+[tool.uv]
+conflicts = [
+  [
+    { extra = "cpu" },
+    { extra = "cu124" },
+  ],
+]
+
+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cpu", extra = "cpu" },
+  { index = "pytorch-cu124", extra = "cu124" },
+]
+torchvision = [
+  { index = "pytorch-cpu", extra = "cpu" },
+  { index = "pytorch-cu124", extra = "cu124" },
+]
+torchaudio = [
+  { index = "pytorch-cpu", extra = "cpu" },
+  { index = "pytorch-cu124", extra = "cu124" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+[[tool.uv.index]]
+name = "pytorch-cu124"
+url = "https://download.pytorch.org/whl/cu124"
+explicit = true
 
 [tool.ruff.format]
 preview = true

{mkv_episode_matcher-0.3.6 → mkv_episode_matcher-0.4.0}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = mkv_episode_matcher
-version = 0.3.6
+version = 0.4.0
 author = Jonathan Sakkos
 author_email = jonathansakkos@gmail.com
 description = The MKV Episode Matcher is a tool for identifying TV series episodes from MKV files and renaming the files accordingly.