Update app.py

app.py

@@ -1,8 +1,20 @@
+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():
+    subprocess.check_call(['pip', 'install', 'tensorflow==2.0'])
+
 def summarize(result):
     summarizer = pipeline('summarization')
     num_iters = int(len(result)/1000)
@@ -19,6 +31,14 @@ def summarize(result):
         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:
@@ -28,21 +48,17 @@ def get_transcript(video_link):
             result += ' ' + i['text']
         print(len(result))
         st.write(result)
-        # If you want to summarize the transcript
         summarize(result)
     else:
         st.write("Invalid YouTube video link")
 
-def extract_video_id(video_link):
-    # Regular expression to extract video ID from YouTube 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 main():
+    if not check_installation("tensorflow"):
+        st.write("TensorFlow 2.0 is not installed. Installing...")
+        install_tensorflow()
+
+    import tensorflow as tf
+
     video_link = st.text_input("Enter YouTube video link:")
     if video_link:
         get_transcript(video_link)