tcsenpai/easy-web-summarizer
1
0
Fork 0
mirror of https://github.com/tcsenpai/easy-web-summarizer.git synced 2025-06-06 10:35:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Full Working Code

Browse Source
This commit is contained in:
Cemal Emre Albayrak 2024-04-27 18:19:01 +03:00
parent 5cf4681a5f
commit 41d8665300
1 changed files with 31 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
yt_summarizer.py
View File

@ -2,24 +2,39 @@ from langchain_community.document_loaders import YoutubeLoader
from langchain.text_splitter import TokenTextSplitter
from langchain_community.chat_models import ChatOllama
from langchain.chains.summarize import load_summarize_chain
from dotenv import load_dotenv
import re
# Get video transcript
videos = ["https://www.youtube.com/watch?v=bYjQ9fzinT8", "https://www.youtube.com/watch?v=QCg0axyXxs4"]
def check_link(link):
yt_regex = r"(https?://)?(www\.)?(youtube\.com/watch\?v=|youtu\.be/)[\w-]+"
return re.match(yt_regex, link) is not None
loader = YoutubeLoader.from_youtube_url(videos[0], language=["en", "en-US"])
transcript = loader.load()
#print(transcript)
def get_transcript(video_link):
# Get video transcript
if check_link(video_link):
loader = YoutubeLoader.from_youtube_url(video_link, language=["en", "en-US"])
transcript = loader.load()
return transcript
return "Invalid YouTube URL."
# Split the transcript into chunks
# Llama 3 model takes up to 8192 input tokens, so I set chunk size to 7500 for leaving some space to model.
splitter = TokenTextSplitter(chunk_size = 7500, chunk_overlap = 100)
chunks = splitter.split_documents(transcript)
#print(chunks)
#print("chunks: ", len(chunks))
def split_chunks(transcript):
# Split the transcript into chunks
# Llama 3 model takes up to 8192 input tokens, so I set chunk size to 7500 for leaving some space to model.
splitter = TokenTextSplitter(chunk_size = 7500, chunk_overlap = 100)
chunks = splitter.split_documents(transcript)
return chunks
llm = ChatOllama(model="llama3")
summarize_chain = load_summarize_chain(llm=llm, chain_type="refine", verbose=True)
def yt_summarization_chain():
llm = ChatOllama(model="llama3")
summarize_chain = load_summarize_chain(llm=llm, chain_type="refine", verbose=True)
return summarize_chain
summary = summarize_chain.run(chunks)
print(summary)
if __name__ == "__main__":
videos = ["https://www.youtube.com/watch?v=bYjQ9fzinT8", "https://www.youtube.com/watch?v=QCg0axyXxs4"]
transcript = get_transcript(videos[0])
chunks = split_chunks(transcript)
sum_chain = yt_summarization_chain()
result = sum_chain.run(chunks)
print(result)