Update app.py

app.py

@@ -5,8 +5,6 @@ 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):
@@ -53,32 +51,12 @@ 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, Audio, and YouTube to Text Transcription")
-st.write("Upload a video, audio file, or provide a YouTube URL to convert it to transcription.")
+st.title("Video and Audio to Text Transcription")
+st.write("Upload a video or audio file to convert it to transcription.")
 
-# Create tabs to separate video, audio, and YouTube URL uploads
-tab = st.selectbox("Select the type of file to upload", ["Video", "Audio", "YouTube URL"])
+# Create tabs to separate video and audio uploads
+tab = st.selectbox("Select the type of file to upload", ["Video", "Audio"])
 
 if tab == "Video":
     # File uploader for video
@@ -195,58 +173,4 @@ 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"
             )