Question answering over YouTube videos with embeddings and LLMs.
Project description
YouTube Question Answer
Simple experiment for question answering on YouTube videos using embeddings and the top n YouTube search result transcripts.
The function will take a question and optionally a YouTube search query (otherwise an LLM will auto-generate one), will compile transcripts for each video result, generate an embedding index using the transcripts and then answer the question using the relevant embeddings.
The function will return both a string response and a list of sources that were used for the answer.
Installation
The package can be installed from PyPI with pip install youtube-qa
. Make sure to set your OPENAI_API_KEY
environment variable before using.
Example
from youtube_qa.youtube_video_index import VideoIndexQueryResponse, YouTubeVideoIndex
video_index = YouTubeVideoIndex()
video_index.build_index(
search_term="huberman motivation",
video_results=3,
)
response: VideoIndexQueryResponse = video_index.answer_question(
question="what are the best researched supplements to help with exercise motivation",
)
print(response.answer) # The answer to the question.
print(response.sources) # Video links and other metadata.
You can also generate the search query given the question:
from youtube_qa.youtube_video_index import VideoIndexQueryResponse, YouTubeVideoIndex
question = "what are the best researched supplements to help with exercise motivation"
video_index = YouTubeVideoIndex()
search_term = video_index.generate_search_query(question)
video_index.build_index(
search_term=search_term,
video_results=3,
)
response: VideoIndexQueryResponse = video_index.answer_question(
question=question,
)
print(response.answer) # The answer to the question.
print(response.sources) # Video links and other metadata.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for youtube_qa-1.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a0355a87c34543e1c2844c65aaac08b18400438dff854f4a10ed9aa8e29b574e |
|
MD5 | 78738eafcad893529c34ba33b3af4df7 |
|
BLAKE2b-256 | cd37d43267bf5fac9669b03bdec6116e2d2226efefa448c06745e1af1b7a6a0d |