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_tensorflow():
    try:
        subprocess.check_call(['pip', 'install', 'tensorflow==2.0'])
        return True
    except subprocess.CalledProcessError as e:
        st.error(f"TensorFlow installation failed with error: {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():
    if not check_installation("tensorflow"):
        st.write("TensorFlow 2.0 is not installed. Installing...")
        if install_tensorflow():
            st.success("TensorFlow 2.0 has been successfully installed.")
        else:
            st.error("Failed to install TensorFlow 2.0. Please install it manually and retry.")
            return

    import tensorflow as tf

    video_link = st.text_input("Enter YouTube video link:")
    if video_link:
        get_transcript(video_link)

main()