adding agent

.python-version

@@ -0,0 +1 @@
+3.13

functions.py

@@ -0,0 +1,63 @@
+from youtube_transcript_api import YouTubeTranscriptApi
+import re
+from agents import function_tool
+
+@function_tool
+def fetch_youtube_transcript(url: str) -> str:
+    """
+    Extract transcript with timestamps from a YouTube video URL and format it for LLM consumption
+    
+    Args:
+        url (str): YouTube video URL
+        
+    Returns:
+        str: Formatted transcript with timestamps, where each entry is on a new line
+             in the format: "[MM:SS] Text"
+    """
+    # Extract video ID from URL
+    video_id_pattern = r'(?:v=|\/)([0-9A-Za-z_-]{11}).*'
+    video_id_match = re.search(video_id_pattern, url)
+    
+    if not video_id_match:
+        raise ValueError("Invalid YouTube URL")
+    
+    video_id = video_id_match.group(1)
+    
+    try:
+        ytt_api = YouTubeTranscriptApi()
+        transcript = ytt_api.fetch(video_id)
+        
+        # Format each entry with timestamp and text
+        formatted_entries = []
+        for entry in transcript:
+            # Convert seconds to MM:SS format
+            minutes = int(entry.start // 60)
+            seconds = int(entry.start % 60)
+            timestamp = f"[{minutes:02d}:{seconds:02d}]"
+            
+            formatted_entry = f"{timestamp} {entry.text}"
+            formatted_entries.append(formatted_entry)
+        
+        # Join all entries with newlines
+        return "\n".join(formatted_entries)
+    
+    except Exception as e:
+        raise Exception(f"Error fetching transcript: {str(e)}")
+
+@function_tool
+def fetch_intstructions(prompt_name: str) -> str:
+    """
+    Fetch instructions for a given prompt name from the prompts/ directory
+
+    Args:
+        prompt_name (str): Name of the prompt to fetch instructions for
+        Available prompts: 
+            - write_blog_post
+            - write_social_post
+            - write_video_chapters
+
+    Returns:
+        str: Instructions for the given prompt
+    """
+    with open(f"prompts/{prompt_name}.md", "r") as f:
+        return f.read()

main.py

@@ -0,0 +1,32 @@
+from agents import Agent, function_tool, Runner
+from openai.types.responses import ResponseTextDeltaEvent
+from dotenv import load_dotenv
+import asyncio
+import os
+from functions import fetch_video_transcript, fetch_intstructions
+
+# try to import OPENAI_API_KEY from .env file, if not found, take user input
+if not os.getenv("OPENAI_API_KEY"):
+    OPENAI_API_KEY = input("Enter your OpenAI API key: ")
+    with open(".env", "w") as f:
+        f.write(f"OPENAI_API_KEY={OPENAI_API_KEY}")
+else:
+    OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+# load system instructions
+with open("prompts/system_instructions.md", "r") as f:
+    system_instructions = f.read()
+    
+# create agent
+agent = Agent(
+    name="YouTube Agent",
+    model="gpt-5",
+    system_instructions=system_instructions,
+    tools=[fetch_video_transcript, fetch_intstructions],
+)
+
+async def main():
+    pass
+
+if __name__ == "__main__":
+    asyncio.run(main())

pyproject.toml

@@ -0,0 +1,13 @@
+[project]
+name = "yt-agent-streamlit"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = [
+    "openai>=1.107.2",
+    "openai-agents>=0.3.0",
+    "python-dotenv>=1.1.1",
+    "streamlit>=1.49.1",
+    "youtube-transcript-api>=1.2.2",
+]