generates timestamp of a youtube video

cookbook/agents/46_generate_yt_timestamps.py

@@ -0,0 +1,18 @@
+from phi.agent import Agent
+from phi.model.openai import OpenAIChat
+from phi.tools.youtube_tools import YouTubeTools
+
+agent = Agent(
+    name="YouTube Timestamps Agent",
+    model=OpenAIChat(id="gpt-4o"),
+    tools=[YouTubeTools()],
+    show_tool_calls=True,
+    instructions=[
+        "You are a YouTube agent. First check the length of the video. Then get the detailed timestamps for a YouTube video corresponding to correct timestamps.",
+        "Don't hallucinate timestamps.",
+        "Make sure to return the timestamps in the format of `[start_time, end_time, summary]`.",
+    ],
+)
+agent.print_response(
+    "Get the detailed timestamps for this video https://www.youtube.com/watch?v=M5tx7VI-LFA", markdown=True
+)

phi/tools/youtube_tools.py

@@ -124,3 +124,31 @@ def get_youtube_video_captions(self, url: str) -> str:
             return "No captions found for video"
         except Exception as e:
             return f"Error getting captions for video: {e}"
+
+    def get_video_timestamps(self, url: str) -> str:
+        """Generate timestamps for a YouTube video based on captions.
+
+        Args:
+            url: The URL of the YouTube video.
+
+        Returns:
+            str: Timestamps and summaries for the video.
+        """
+        if not url:
+            return "No URL provided"
+
+        try:
+            video_id = self.get_youtube_video_id(url)
+        except Exception:
+            return "Error getting video ID from URL, please provide a valid YouTube url"
+
+        try:
+            captions = YouTubeTranscriptApi.get_transcript(video_id, languages=self.languages or ["en"])
+            timestamps = []
+            for line in captions:
+                start = int(line["start"])
+                minutes, seconds = divmod(start, 60)
+                timestamps.append(f"{minutes}:{seconds:02d} - {line['text']}")
+            return "\n".join(timestamps)
+        except Exception as e:
+            return f"Error generating timestamps: {e}"