Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from dotenv import load_dotenv | |
| load_dotenv() ##load all the nevironment variables | |
| import os | |
| import google.generativeai as genai | |
| from youtube_transcript_api import YouTubeTranscriptApi | |
| genai.configure(api_key="AIzaSyBjEeTXoxrkk3eFc_o_CAfHMIfgWglPrlU") | |
| prompt="""You are Yotube video summarizer. You will be taking the transcript text | |
| and summarizing the entire video and providing the important summary | |
| within 200 words. Please provide the summary of the text given here: """ | |
| ## getting the summary based on Prompt from Google Gemini Pro | |
| 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 Video Summarizer") | |
| youtube_link = st.text_input("Enter YouTube Video Link:") | |
| if youtube_link: | |
| try: | |
| video_id = youtube_link.split("=")[1] | |
| # print(video_id) | |
| st.image(f"http://img.youtube.com/vi/{video_id}/0.jpg", use_container_width=True) | |
| # Fetch subtitles | |
| transcript_list = YouTubeTranscriptApi.list_transcripts(video_id) | |
| data = "" | |
| for transcript in transcript_list: | |
| if transcript.is_translatable: | |
| subtitles = transcript.translate('en').fetch() | |
| data += " ".join([entry['text'] for entry in subtitles]) | |
| if data: | |
| st.subheader("Extracted Subtitles") | |
| st.write(data) | |
| if st.button("Summarize"): | |
| with st.spinner("Summarizing..."): | |
| summary = generate_gemini_content(data, prompt) | |
| st.subheader("Summary") | |
| st.write(summary) | |
| # transcript_text=YouTubeTranscriptApi.get_transcript(video_id) | |
| # transcript = "" | |
| # for i in transcript_text: | |
| # transcript += " " + i["text"] | |
| # return transcript | |
| except Exception as e: | |
| raise e | |
| # if st.button("Summarize"): | |
| # if transcript_text: | |
| # summary=generate_gemini_content(transcript_text,prompt) | |
| # st.markdown("## Summary:") | |
| # st.write(summary) | |
| # # App title | |
| # st.title("YouTube Video Summarizer") | |
| # # Input: YouTube video URL | |
| # video_url = st.text_input("Enter YouTube Video URL:") | |
| # if video_url: | |
| # try: | |
| # # Extract video ID from the URL | |
| # if "youtube.com" in video_url: | |
| # video_id = video_url.split("v=")[-1] | |
| # elif "youtu.be" in video_url: | |
| # video_id = video_url.split("/")[-1] | |
| # else: | |
| # st.error("Invalid YouTube URL!") | |
| # video_id = None | |
| # if video_id: | |
| # # Fetch subtitles | |
| # transcript_list = YouTubeTranscriptApi.list_transcripts(video_id) | |
| # data = "" | |
| # for transcript in transcript_list: | |
| # if transcript.is_translatable: | |
| # subtitles = transcript.translate('en').fetch() | |
| # data += " ".join([entry['text'] for entry in subtitles]) | |
| # break | |
| # # Display subtitles | |
| # if data: | |
| # st.subheader("Extracted Subtitles") | |
| # st.write(data) | |
| # # Summarize subtitles | |
| # if st.button("Summarize"): | |
| # with st.spinner("Summarizing..."): | |
| # summary = generate_gemini_content(data, prompt) | |
| # st.subheader("Summary") | |
| # st.write(summary) | |
| # else: | |
| # st.warning("No subtitles found for this video.") | |
| # except Exception as e: | |
| # st.error(f"An error occurred: {e}") | |