init

app.py

@@ -0,0 +1,78 @@
+import streamlit as st
+from pytube import YouTube
+import yt_dlp
+
+
+# from faster_whisper import WhisperModel
+# model = WhisperModel("large-v2", device="cpu", device_index=0,  compute_type="float16")
+
+DESCRIPTION = """
+Welcome to the **YouTube Video summary** powered by Llama-2 models.
+"""
+st.title("YouTube Video summary")
+st.markdown(DESCRIPTION)
+
+def get_video_title(youtube_url: str) -> str:
+    yt = YouTube(youtube_url)
+    embed_url = f"https://www.youtube.com/embed/{yt.video_id}"
+    embed_html = f'<iframe src="{embed_url}" frameborder="0" allowfullscreen></iframe>'
+    return yt.title, embed_html
+
+
+def initialize_session_state():
+    if "youtube_url" not in st.session_state:
+        st.session_state.youtube_url = ""
+    if "model_choice" not in st.session_state:
+        st.session_state.model_choice = "Llama2-70b"
+    if "setup_done" not in st.session_state:
+        st.session_state.setup_done = False
+    if "doneYoutubeurl" not in st.session_state:
+        st.session_state.doneYoutubeurl = ""
+
+def sidebar():
+    with st.sidebar:
+        st.markdown("Enter the YouTube Video URL below🔗")
+        st.session_state.youtube_url = st.text_input("YouTube Video URL:")
+
+        if st.session_state.youtube_url:
+            # Get the video title
+            video_title, embed_html = get_video_title(st.session_state.youtube_url)
+            st.markdown(f"### {video_title}")
+
+            # Embed the video
+            st.markdown(embed_html, unsafe_allow_html=True)
+
+sidebar()
+initialize_session_state()
+
+ydl_opts = {
+    'outtmpl': 'demo.m4a',
+    'format': 'm4a/bestaudio/best',
+    # ℹ️ See help(yt_dlp.postprocessor) for a list of available Postprocessors and their arguments
+    'postprocessors': [{  # Extract audio using ffmpeg
+        'key': 'FFmpegExtractAudio',
+        'preferredcodec': 'm4a',
+    }],
+}
+
+if st.session_state.youtube_url:
+    with st.status("Get video Audio..."):
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            error_code = ydl.download([st.session_state.youtube_url])
+
+    audio_file = open('demo.m4a', 'rb')
+    audio_bytes = audio_file.read()
+    st.audio(audio_bytes, format='audio/ogg')
+
+    
+    # segments, info = model.transcribe("demo.m4a", beam_size=5)
+    # st.markdown("Detected language '%s' with probability %f" % (info.language, info.language_probability))
+
+    # full_response = ""
+    # message_placeholder = st.empty()
+    # for segment in segments:
+    #     # print("[%.2fs -> %.2fs] %s" % (segment.start, segment.end, segment.text))
+    #     full_response += segment.text + " "
+    #     st.write(segment.text)
+        # message_placeholder.markdown(full_response + "▌")
+    # message_placeholder.markdown(full_response)

requirements.txt

@@ -0,0 +1,2 @@
+pytube
+yt_dlp