venkataseetharam's picture
Upload 3 files
e2085a1 verified
import streamlit as st
from dotenv import load_dotenv
load_dotenv() #load all the environment variables
import os
import google.generativeai as genai
from youtube_transcript_api import YouTubeTranscriptApi
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
prompt="""You are a youtube video summarizer.You will be taking the transcript text and summarizing the entire video
and providing the important summary in points.Please provide the summary of the text given here"""
#getting the transcript data from the video
def extract_transcript_details(youtube_video_url):
try:
video_id=youtube_video_url.split("=")[1]
transcript_text=YouTubeTranscriptApi.get_transcript(video_id)
transcript=""
for i in transcript_text:
transcript+=" "+i["text"]
return transcript
except Exception as e:
raise e
#getting the summary from google gemini
def generate_gemini_content(transcript_text,prompt):
model=genai.GenerativeModel("gemini-pro")
response=model.generate_content(prompt+transcript_text)
return response.text
st.title("YouTube Transcript to Detailed Notes Converted")
youtube_link=st.text_input("Enter you YouTube Video URL:")
if youtube_link:
video_id=youtube_link.split("=")[1]
st.image(f"http://img.youtube.com/vi/{video_id}/0.jpg",use_column_width=True)
if st.button("Get detailed notes"):
transcript_text=extract_transcript_details(youtube_link)
if transcript_text:
summary=generate_gemini_content(transcript_text,prompt)
st.markdown("## Detailed Notes:")
st.write(summary)