| | import subprocess |
| | import importlib.util |
| | from transformers import pipeline |
| | from youtube_transcript_api import YouTubeTranscriptApi |
| | import streamlit as st |
| | import re |
| |
|
| | def check_installation(package): |
| | spec = importlib.util.find_spec(package) |
| | if spec is None: |
| | return False |
| | else: |
| | return True |
| |
|
| | def install_tf_keras(): |
| | try: |
| | subprocess.check_call(['pip', 'install', 'tf-keras']) |
| | return True |
| | except subprocess.CalledProcessError as e: |
| | st.error(f"Failed to install tf-keras package: {e}") |
| | return False |
| |
|
| | def summarize(result): |
| | summarizer = pipeline('summarization') |
| | num_iters = int(len(result)/1000) |
| | summarized_text = [] |
| | for i in range(0, num_iters + 1): |
| | start = 0 |
| | start = i * 1000 |
| | end = (i + 1) * 1000 |
| | print("input text \n" + result[start:end]) |
| | out = summarizer(result[start:end]) |
| | out = out[0] |
| | out = out['summary_text'] |
| | print("Summarized text\n"+out) |
| | summarized_text.append(out) |
| | st.write(summarized_text) |
| |
|
| | def extract_video_id(video_link): |
| | regex = r"(?:https:\/\/)?(?:www\.)?(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})" |
| | matches = re.findall(regex, video_link) |
| | if matches: |
| | return matches[0] |
| | else: |
| | return None |
| |
|
| | def get_transcript(video_link): |
| | video_id = extract_video_id(video_link) |
| | if video_id: |
| | transcript = YouTubeTranscriptApi.get_transcript(video_id) |
| | result = "" |
| | for i in transcript: |
| | result += ' ' + i['text'] |
| | print(len(result)) |
| | st.write(result) |
| | summarize(result) |
| | else: |
| | st.write("Invalid YouTube video link") |
| |
|
| | def main(): |
| | st.title('YouTube Transcript Generator') |
| | st.write("Get the full text of any YouTube video in seconds.") |
| |
|
| | if not check_installation("tf_keras"): |
| | st.write("tf-keras is not installed. Installing...") |
| | if install_tf_keras(): |
| | st.success("tf-keras has been successfully installed.") |
| | else: |
| | st.error("Failed to install tf-keras. Please install it manually and retry.") |
| | return |
| |
|
| | import tensorflow as tf |
| |
|
| | video_link = st.text_input("Enter YouTube video link:") |
| | if video_link: |
| | if st.button("Generate"): |
| | get_transcript(video_link) |
| |
|
| | main() |
| |
|
