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

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/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.0.dist-info}/METADATA

@@ -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.dist-info → mkv_episode_matcher-0.4.0.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/__main__.py,sha256=Tpppeh59E2PKCrALweBdfzrm4GUeK8FjAECz2ECvmXs,7105
 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.0.dist-info/METADATA,sha256=o6xssGRpkRcr9elsAt0lf2ktwr2-7aSVweBZUDtlr7I,5545
+mkv_episode_matcher-0.4.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+mkv_episode_matcher-0.4.0.dist-info/entry_points.txt,sha256=IglJ43SuCZq2eQ3shMFILCkmQASJHnDCI3ogohW2Hn4,64
+mkv_episode_matcher-0.4.0.dist-info/top_level.txt,sha256=XRLbd93HUaedeWLtkyTvQjFcE5QcBRYa3V-CfHrq-OI,20
+mkv_episode_matcher-0.4.0.dist-info/RECORD,,