mkv-episode-matcher 0.3.5__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/episode_identification.py

@@ -96,14 +96,27 @@ class EpisodeMatcher:
             # Load Whisper model
             model = whisper.load_model("base", device=self.device)
             
-            # Get season-specific reference files
+            # Get season-specific reference files using multiple patterns
             reference_dir = self.cache_dir / "data" / self.show_name
-            season_pattern = f"S{season_number:02d}E"
-            reference_files = [
-                f for f in reference_dir.glob("*.srt")
-                if season_pattern in f.name
+            
+            # Create season patterns for different formats
+            patterns = [
+                f"S{season_number:02d}E",  # S01E01
+                f"S{season_number}E",      # S1E01
+                f"{season_number:02d}x",   # 01x01
+                f"{season_number}x",       # 1x01
             ]
             
+            reference_files = []
+            for pattern in patterns:
+                files = [f for f in reference_dir.glob("*.srt") 
+                        if any(re.search(f"{p}\\d+", f.name, re.IGNORECASE) 
+                        for p in patterns)]
+                reference_files.extend(files)
+            
+            # Remove duplicates while preserving order
+            reference_files = list(dict.fromkeys(reference_files))
+            
             if not reference_files:
                 logger.error(f"No reference files found for season {season_number}")
                 return None

mkv_episode_matcher/episode_matcher.py

@@ -29,6 +29,14 @@ def process_show(season=None, dry_run=False, get_subs=False):
     show_name = clean_text(os.path.basename(show_dir))
     matcher = EpisodeMatcher(CACHE_DIR, show_name)
     
+    # Early check for reference files
+    reference_dir = Path(CACHE_DIR) / "data" / show_name
+    reference_files = list(reference_dir.glob("*.srt"))
+    if not reference_files:
+        logger.error(f"No reference subtitle files found in {reference_dir}")
+        logger.info("Please download reference subtitles first")
+        return
+        
     season_paths = get_valid_seasons(show_dir)
     if not season_paths:
         logger.warning(f"No seasons with .mkv files found")

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/utils.py

@@ -300,7 +300,7 @@ def extract_srt_text(filepath):
 
 def extract_season_episode(filename):
     """
-    Extract season and episode numbers from filename.
+    Extract season and episode numbers from filename with support for multiple formats.
     
     Args:
         filename (str): Filename to parse
@@ -308,10 +308,20 @@ def extract_season_episode(filename):
     Returns:
         tuple: (season_number, episode_number)
     """
-    match = re.search(r'S(\d+)E(\d+)', filename)
-    if match:
-        return int(match.group(1)), int(match.group(2))
+    # List of patterns to try
+    patterns = [
+        r'S(\d+)E(\d+)',          # S01E01
+        r'(\d+)x(\d+)',           # 1x01 or 01x01
+        r'Season\s*(\d+).*?(\d+)' # Season 1 - 01
+    ]
+    
+    for pattern in patterns:
+        match = re.search(pattern, filename, re.IGNORECASE)
+        if match:
+            return int(match.group(1)), int(match.group(2))
+            
     return None, None
+
 def process_srt_files(show_dir):
     """
     Process all SRT files in the given directory and its subdirectories.

{mkv_episode_matcher-0.3.5.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.5
+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
 
@@ -65,10 +76,11 @@ Automatically match and rename your MKV TV episodes using The Movie Database (TM
 ```bash
 pip install mkv-episode-matcher
 ```
+2. Download .srt subtitles files to ~/.mkv-episode-matcher/cache/data/Show Name/
 
-2. Run on your show directory:
+3. Run on your show directory:
 ```bash
-mkv-match --show-dir "path/to/your/show" --get-subs true
+mkv-match --show-dir "path/to/your/show"
 ```
 
 ## Documentation

{mkv_episode_matcher-0.3.5.dist-info → mkv_episode_matcher-0.4.0.dist-info}/RECORD

@@ -1,14 +1,14 @@
 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=xYqHq1YFbZT8L1Gfa_DhSStrLblKTWxZte__B0qikQU,9739
-mkv_episode_matcher/episode_matcher.py,sha256=BJ76DPxsmZs-KfHZZ_0WvKSBZWXsUEO6lW34YdYEaxM,3979
+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=Txnn24ou7Pg3iMq9WrT3nwBRlRP8JEuZQ2ZYW7uesp4,13972
+mkv_episode_matcher/utils.py,sha256=VASbougN3rb2iu40iZWkGjKIbahW713TOrFBo_TR9wo,14269
 mkv_episode_matcher/libraries/pgs2srt/.gitignore,sha256=mt3uxWYZaFurMw_yGE258gWhtGKPVR7e3Ll4ALJpyj4,23
 mkv_episode_matcher/libraries/pgs2srt/README.md,sha256=olb25G17tj0kxPgp_LcH5I2QWXjgP1m8JFyjYRGz4UU,1374
 mkv_episode_matcher/libraries/pgs2srt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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.5.dist-info/METADATA,sha256=mTKSbM9Ai5UDKyj2K4AKgkdjdPVEaxylfHHp95wVZv4,5048
-mkv_episode_matcher-0.3.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-mkv_episode_matcher-0.3.5.dist-info/entry_points.txt,sha256=IglJ43SuCZq2eQ3shMFILCkmQASJHnDCI3ogohW2Hn4,64
-mkv_episode_matcher-0.3.5.dist-info/top_level.txt,sha256=XRLbd93HUaedeWLtkyTvQjFcE5QcBRYa3V-CfHrq-OI,20
-mkv_episode_matcher-0.3.5.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,,