Update app.py

app.py

@@ -5,6 +5,8 @@ from pydub import AudioSegment
 import tempfile
 import os
 import io
+from pytube import YouTube
+import requests
 
 # Function to convert video to audio
 def video_to_audio(video_file):
@@ -51,12 +53,32 @@ def transcribe_audio(audio_file):
     except sr.RequestError:
         return "Could not request results from Google Speech Recognition service."
 
+# Function to download audio from YouTube and convert it to WAV
+def download_youtube_audio(url):
+    # Get the YouTube video
+    yt = YouTube(url)
+    
+    # Get the highest quality stream available (audio only)
+    audio_stream = yt.streams.filter(only_audio=True).first()
+    
+    # Download the audio as an MP4 file (audio-only)
+    temp_audio_path = tempfile.mktemp(suffix=".mp4")
+    audio_stream.download(output_path=temp_audio_path)
+    
+    # Convert the downloaded MP4 to WAV format
+    wav_audio_file = convert_mp3_to_wav(temp_audio_path)
+    
+    # Cleanup the temporary MP4 file
+    os.remove(temp_audio_path)
+    
+    return wav_audio_file
+
 # Streamlit app layout
-st.title("Video and Audio to Text Transcription")
-st.write("Upload a video or audio file to convert it to transcription.")
+st.title("Video, Audio, and YouTube to Text Transcription")
+st.write("Upload a video, audio file, or provide a YouTube URL to convert it to transcription.")
 
-# Create tabs to separate video and audio uploads
-tab = st.selectbox("Select the type of file to upload", ["Video", "Audio"])
+# Create tabs to separate video, audio, and YouTube URL uploads
+tab = st.selectbox("Select the type of file to upload", ["Video", "Audio", "YouTube URL"])
 
 if tab == "Video":
     # File uploader for video
@@ -173,4 +195,58 @@ elif tab == "Audio":
                 data=st.session_state.wav_audio_file_audio,
                 file_name="converted_audio_audio.wav",
                 mime="audio/wav"
+            )
+
+elif tab == "YouTube URL":
+    # Input for YouTube URL
+    youtube_url = st.text_input("Enter YouTube URL")
+
+    if youtube_url:
+        # Add an "Analyze YouTube URL" button
+        if st.button("Analyze YouTube URL"):
+            with st.spinner("Processing YouTube video... Please wait."):
+                try:
+                    # Download audio from the YouTube video
+                    wav_audio_file = download_youtube_audio(youtube_url)
+                    
+                    # Transcribe audio to text
+                    transcription = transcribe_audio(wav_audio_file)
+
+                    # Show the transcription
+                    st.text_area("Transcription", transcription, height=300)
+
+                    # Store transcription and audio file in session state
+                    st.session_state.transcription_youtube = transcription
+                    
+                    # Store the audio file as a BytesIO object in memory
+                    with open(wav_audio_file, "rb") as f:
+                        audio_data = f.read()
+                        st.session_state.wav_audio_file_youtube = io.BytesIO(audio_data)
+
+                    # Cleanup the temporary audio file
+                    os.remove(wav_audio_file)
+
+                except Exception as e:
+                    st.error(f"Error processing the YouTube URL: {e}")
+
+        # Check if transcription and audio file are stored in session state
+        if 'transcription_youtube' in st.session_state and 'wav_audio_file_youtube' in st.session_state:
+            # Provide the audio file to the user for download
+            st.audio(st.session_state.wav_audio_file_youtube, format='audio/wav')
+            
+            # Add download buttons for the transcription and audio
+            # Downloadable transcription file
+            st.download_button(
+                label="Download Transcription",
+                data=st.session_state.transcription_youtube,
+                file_name="transcription_youtube.txt",
+                mime="text/plain"
+            )
+            
+            # Downloadable audio file
+            st.download_button(
+                label="Download Audio",
+                data=st.session_state.wav_audio_file_youtube,
+                file_name="converted_audio_youtube.wav",
+                mime="audio/wav"
             )