save

Author: hamelsmu

hamel/writing.py

@@ -52,8 +52,9 @@ def generate_annotated_talk_post(slide_path,
                                  video_source,  # YouTube link or local MP4 path
                                  image_dir,
                                  transcript_path=None, 
-                                 example_urls=_annotated_post_urls):
-    "Assemble the prompt for the annotated post."
+                                 example_urls=_annotated_post_urls,
+                                 user_prompt=None):
+    "Assemble the prompt for the annotated post. Optional user_prompt provides additional context/examples."
 
     # Check if video_source is a local MP4 or YouTube URL
     is_local_video = Path(video_source).exists() and Path(video_source).suffix.lower() == '.mp4'
@@ -138,7 +139,9 @@ def generate_annotated_talk_post(slide_path,
 
 Please go ahead and draft the post. Please also include front matter similar to the front matter in the examples and select the best slide from the talk as the cover image (which is not the title slide, but instead another interesting slide that is punchy).
 """
+    if user_prompt:
+        prompt += f"\n\nAdditional context/instructions from the user:\n<user-context>\n{user_prompt}\n</user-context>"
     # Use the appropriate video source
     attachment = [slide_path, video_source] if is_local_video else [slide_path, video_source]
-    draft_post = gem(prompt, attachment, model='gemini-2.5-pro')
+    draft_post = gem(prompt, attachment, model='gemini-3-pro-preview')
     return draft_post

hamel/yt.py

@@ -21,7 +21,7 @@ def yt_chapters(url_or_path):
     "Generate YouTube Summary and Chapters from a video (YouTube URL or local MP4)."
 
     chapter_prompt="Generate a succinct video summary (1-2 sentences) followed by video chapter timestamps for this video. Format each line of the chapter summaries as 'MM:SS - Chapter Title' (e.g., '02:30 - Introduction'). Start with 00:00. Include all major topics and transitions and be thorough - do not miss any important topics.  For the summary, do not say 'In this video, we will cover the following topics', 'This video discusses..' or anything like that. Instead, reference the main speaker's name if you know it.  If there is a Q&A Section, enumerate individual questions as additional chapters."
-    return gem(prompt=chapter_prompt, o=url_or_path, model="gemini-2.5-pro")
+    return gem(prompt=chapter_prompt, o=url_or_path, model="gemini-3-pro-preview")
 
 # %% ../nbs/01_yt.ipynb 11
 def _extract_video_id(url: str) -> Optional[str]:

nbs/00_gem.ipynb

@@ -1366,7 +1366,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "hamel",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },

nbs/01_yt.ipynb

@@ -75,7 +75,7 @@
     "    \"Generate YouTube Summary and Chapters from a video (YouTube URL or local MP4).\"\n",
     "    \n",
     "    chapter_prompt=\"Generate a succinct video summary (1-2 sentences) followed by video chapter timestamps for this video. Format each line of the chapter summaries as 'MM:SS - Chapter Title' (e.g., '02:30 - Introduction'). Start with 00:00. Include all major topics and transitions and be thorough - do not miss any important topics.  For the summary, do not say 'In this video, we will cover the following topics', 'This video discusses..' or anything like that. Instead, reference the main speaker's name if you know it.  If there is a Q&A Section, enumerate individual questions as additional chapters.\"\n",
-    "    return gem(prompt=chapter_prompt, o=url_or_path, model=\"gemini-2.5-pro\")"
+    "    return gem(prompt=chapter_prompt, o=url_or_path, model=\"gemini-3-pro-preview\")"
    ]
   },
   {