mkv-episode-matcher 0.3.6__py3-none-any.whl → 0.4.1__py3-none-any.whl

mkv_episode_matcher/__init__.py

@@ -1 +1,9 @@
+"""MKV Episode Matcher package."""
+from importlib.metadata import version, PackageNotFoundError
+
+try:
+    __version__ = version("mkv-episode-matcher")
+except PackageNotFoundError:
+    # package is not installed
+    __version__ = "unknown"
 

mkv_episode_matcher/__main__.py

@@ -1,9 +1,10 @@
 # __main__.py
 import argparse
 import os
+import sys
 
 from loguru import logger
-
+from mkv_episode_matcher import __version__
 from mkv_episode_matcher.config import get_config, set_config
 
 # Log the start of the application
@@ -62,6 +63,12 @@ def main():
 
     # Parse command-line arguments
     parser = argparse.ArgumentParser(description="Process shows with TMDb API")
+    parser.add_argument(
+        "--version",
+        action="version",
+        version=f"%(prog)s {__version__}",
+        help="Show the version number and exit"
+    )
     parser.add_argument("--tmdb-api-key", help="TMDb API key")
     parser.add_argument("--show-dir", help="Main directory of the show")
     parser.add_argument(
@@ -92,7 +99,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/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.dist-info → mkv_episode_matcher-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mkv-episode-matcher
-Version: 0.3.6
+Version: 0.4.1
 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
 
@@ -56,7 +67,7 @@ Automatically match and rename your MKV TV episodes using The Movie Database (TM
 - Python 3.9 or higher
 - [FFmpeg](https://ffmpeg.org/download.html) installed and available in system PATH
 - [Tesseract OCR](https://github.com/UB-Mannheim/tesseract/wiki) installed (required for image-based subtitle processing)
-- TMDb API key
+- TMDb API key (optional, for subtitle downloads)
 - OpenSubtitles account (optional, for subtitle downloads)
 
 ## Quick Start
@@ -72,10 +83,6 @@ pip install mkv-episode-matcher
 mkv-match --show-dir "path/to/your/show"
 ```
 
-## Documentation
-
-Full documentation is available at [https://jsakkos.github.io/mkv-episode-matcher/](https://jsakkos.github.io/mkv-episode-matcher/)
-
 ## Directory Structure
 
 MKV Episode Matcher expects your TV shows to be organized as follows:
@@ -125,3 +132,6 @@ Distributed under the MIT License. See `LICENSE` for more information.
 - [OpenSubtitles](https://www.opensubtitles.com/) for subtitle integration
 - All contributors who have helped improve this project
 
+## Documentation
+
+Full documentation is available at [https://jsakkos.github.io/mkv-episode-matcher/](https://jsakkos.github.io/mkv-episode-matcher/)

{mkv_episode_matcher-0.3.6.dist-info → mkv_episode_matcher-0.4.1.dist-info}/RECORD

@@ -1,11 +1,11 @@
 mkv_episode_matcher/.gitattributes,sha256=Gh2-F2vCM7SZ01pX23UT8pQcmauXWfF3gwyRSb6ZAFs,66
-mkv_episode_matcher/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-mkv_episode_matcher/__main__.py,sha256=3ZcCUxeI7rUA-4oiCD2WXBiOFJAqLsVVWfZKN446FwQ,6792
+mkv_episode_matcher/__init__.py,sha256=aNlpgTo1kHVrBcR3SH6wRmCgKu8KjNTki1ZvFfAud6s,240
+mkv_episode_matcher/__main__.py,sha256=0ZwFXw30744lZ8HIMzUN4F2abpTRCT4glBCM9XQdD3k,7328
 mkv_episode_matcher/config.py,sha256=zDDKBcsDt5fME9BRqiTi7yWKeast1pZh36BNYMvIBYM,2419
 mkv_episode_matcher/episode_identification.py,sha256=_6M1UJkq1RGfmLI32u9dNOVvgp5Vf2MjqW2MTx0Gl8E,10329
 mkv_episode_matcher/episode_matcher.py,sha256=vunYpHQxyXo3l88BUScXa7_kMYMCV1pXpQxaLa-plZA,4325
 mkv_episode_matcher/mkv_to_srt.py,sha256=4yxBHRVhgVby0UtQ2aTXGuoQpid8pkgjMIaHU6GCdzc,10857
-mkv_episode_matcher/speech_to_text.py,sha256=-bnGvmtPCKyHFPEaXwIcEYTf_P13rNpAJA-2UFeRFrs,2806
+mkv_episode_matcher/speech_to_text.py,sha256=wVDrFFR7oASGMyq5cfOWmInEIeU9b3MPCLs9EyJrOMw,3128
 mkv_episode_matcher/subtitle_utils.py,sha256=rYSbd393pKYQW0w4sXgals02WFGqMYYYkQHDbEkWF8c,2666
 mkv_episode_matcher/tmdb_client.py,sha256=LbMCgjmp7sCbrQo_CDlpcnryKPz5S7inE24YY9Pyjk4,4172
 mkv_episode_matcher/utils.py,sha256=VASbougN3rb2iu40iZWkGjKIbahW713TOrFBo_TR9wo,14269
@@ -19,8 +19,8 @@ mkv_episode_matcher/libraries/pgs2srt/requirements.txt,sha256=sg87dqWw_qpbwciw-M
 mkv_episode_matcher/libraries/pgs2srt/Libraries/SubZero/SubZero.py,sha256=geT1LXdVd8yED9zoJ9K1XfP2JzGcM7u1SslHYrJI09o,10061
 mkv_episode_matcher/libraries/pgs2srt/Libraries/SubZero/post_processing.py,sha256=GKtVy_Lxv-z27mkRG8pJF2znKWXwZTot7jL6kN-zIxM,10503
 mkv_episode_matcher/libraries/pgs2srt/Libraries/SubZero/dictionaries/data.py,sha256=AlJHUYXl85J95OzGRik-AHVfzDd7Q8BJCvD4Nr8kRIk,938598
-mkv_episode_matcher-0.3.6.dist-info/METADATA,sha256=gjlfeFtCjCCjf2qUWTGJ9zTvSSU_1KRhp1ZtIBISecI,5113
-mkv_episode_matcher-0.3.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-mkv_episode_matcher-0.3.6.dist-info/entry_points.txt,sha256=IglJ43SuCZq2eQ3shMFILCkmQASJHnDCI3ogohW2Hn4,64
-mkv_episode_matcher-0.3.6.dist-info/top_level.txt,sha256=XRLbd93HUaedeWLtkyTvQjFcE5QcBRYa3V-CfHrq-OI,20
-mkv_episode_matcher-0.3.6.dist-info/RECORD,,
+mkv_episode_matcher-0.4.1.dist-info/METADATA,sha256=IWKnIxGcG_W1EEtPfYasyMCfyZL7-NL_ScYfNl0HVlg,5579
+mkv_episode_matcher-0.4.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+mkv_episode_matcher-0.4.1.dist-info/entry_points.txt,sha256=IglJ43SuCZq2eQ3shMFILCkmQASJHnDCI3ogohW2Hn4,64
+mkv_episode_matcher-0.4.1.dist-info/top_level.txt,sha256=XRLbd93HUaedeWLtkyTvQjFcE5QcBRYa3V-CfHrq-OI,20
+mkv_episode_matcher-0.4.1.dist-info/RECORD,,