Embedded Provider: improve streams filtering

libs/fese/__init__.py

@@ -4,4 +4,4 @@
 from .container import FFprobeVideoContainer
 from .stream import FFprobeSubtitleStream
 
-__version__ = "0.2.3"
+__version__ = "0.2.5"

libs/fese/disposition.py

@@ -55,6 +55,12 @@ class FFprobeSubtitleDisposition:
     def suffix(self):
         return self._content_type or ""
 
+    def language_kwargs(self):
+        return {
+            "hi": self._content_type == "hearing_impaired",
+            "forced": self._content_type == "forced",
+        }
+
     def __str__(self):
         return self.suffix.upper() or "GENERIC"
 

libs/fese/stream.py

@@ -5,6 +5,8 @@ from __future__ import annotations
 from datetime import timedelta
 import logging
 
+from babelfish import Language
+
 from .disposition import FFprobeSubtitleDisposition
 from .exceptions import UnsupportedCodec
 from .tags import FFprobeGenericSubtitleTags
@@ -85,6 +87,10 @@ class FFprobeSubtitleStream:
         # Legacy
         return self.tags.language
 
+    @language.setter
+    def language(self, value: Language):
+        self.tags.language = value
+
     @property
     def extension(self):
         return self._codec["copy_format"] or self._codec["convert_default_format"] or ""

libs/subliminal_patch/providers/embeddedsubtitles.py

@@ -297,8 +297,13 @@ def _discard_possible_incomplete_subtitles(streams):
     valid_streams = []
 
     for stream in streams:
+        # Make sure to update stream's language to reflect disposition
+        stream.language = Language.rebuild(
+            stream.language, **stream.disposition.language_kwargs()
+        )
+
         # 500 < 1200
-        if stream.tags.frames < max_frames // 2:
+        if not stream.language.forced and stream.tags.frames < max_frames // 2:
             logger.debug(
                 "Possible bad subtitle found: %s (%s frames - %s frames)",
                 stream,