content-core 1.0.3__tar.gz → 1.0.4__tar.gz

{content_core-1.0.3 → content_core-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: content-core
-Version: 1.0.3
+Version: 1.0.4
 Summary: Extract what matters from any media source
 Author-email: LUIS NOVO <lfnovo@gmail.com>
 License-File: LICENSE

{content_core-1.0.3 → content_core-1.0.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "content-core"
-version = "1.0.3"
+version = "1.0.4"
 description = "Extract what matters from any media source"
 readme = "README.md"
 homepage = "https://github.com/lfnovo/content-core"

{content_core-1.0.3 → content_core-1.0.4}/src/content_core/content/summary/core.py

@@ -8,7 +8,7 @@ async def summarize(content: str, context: str) -> str:
     templated_message_fn = partial(templated_message, model=ModelFactory.get_model('summary_model'))
     response = await templated_message_fn(
         TemplatedMessageInput(
-            user_prompt_template="content/summarize",
+            user_prompt_template="prompts/content/summarize",
             data={"content": content, "context": context},
         )
     )

{content_core-1.0.3 → content_core-1.0.4}/src/content_core/processors/youtube.py

@@ -3,13 +3,12 @@ import ssl
 
 import aiohttp
 from bs4 import BeautifulSoup
-from youtube_transcript_api import YouTubeTranscriptApi  # type: ignore
-from youtube_transcript_api.formatters import TextFormatter  # type: ignore
-
 from content_core.common import ProcessSourceState
 from content_core.common.exceptions import NoTranscriptFound
 from content_core.config import CONFIG
 from content_core.logging import logger
+from youtube_transcript_api import YouTubeTranscriptApi  # type: ignore
+from youtube_transcript_api.formatters import TextFormatter  # type: ignore
 
 ssl._create_default_https_context = ssl._create_unverified_context
 
@@ -137,7 +136,7 @@ def extract_transcript_pytubefix(url, languages=["en", "es", "pt"]):
     from pytubefix import YouTube
 
     yt = YouTube(url)
-    print(yt.captions)
+    logger.debug(f"Captions: {yt.captions}")
 
     # Try to get captions in the preferred languages
     if yt.captions:
@@ -149,12 +148,20 @@ def extract_transcript_pytubefix(url, languages=["en", "es", "pt"]):
                 caption = yt.captions[f"a.{lang}"]
                 break
         else:  # No preferred language found, use the first available
-            caption_key = next(iter(yt.captions))
-            caption = yt.captions[caption_key]
-
-        srt_captions = caption.generate_srt_captions()
-        txt_captions = caption.generate_txt_captions()
-        return txt_captions, srt_captions
+            caption_key = list(yt.captions.keys())[0]
+            caption = yt.captions[caption_key.code]
+        try:
+            srt_captions = caption.generate_srt_captions()
+            txt_captions = caption.generate_txt_captions()
+            return txt_captions, srt_captions
+        except KeyError as e:
+            logger.error(f"KeyError while generating captions for {caption}: {e}")
+            return None, None
+        except Exception as e:
+            logger.error(
+                f"Unexpected error while generating captions for {caption}: {e}"
+            )
+            return None, None
 
     return None, None
 

{content_core-1.0.3 → content_core-1.0.4}/uv.lock

@@ -410,7 +410,7 @@ wheels = [
 
 [[package]]
 name = "content-core"
-version = "1.0.3"
+version = "1.0.4"
 source = { editable = "." }
 dependencies = [
     { name = "ai-prompter" },