Update thaTube.py

thaTube.py

@@ -25,22 +25,35 @@ class thaTube:
     def extract_audio(self):
         """Extract audio from the selected video and save it as an MP3 file."""
         if not self.video_url:
-            raise ValueError("No video selected. Please select a video first.")
-
-        video = YouTube(self.video_url)
-        audio_stream = video.streams.filter(only_audio=True).first()
-
-        # Create a temporary MP3 file
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_audio_file:
-            audio_stream.download(output_path=temp_audio_file.name)
-            temp_audio_file.close()
-
-            audio = AudioSegment.from_file(temp_audio_file.name)
-            with tempfile.NamedTemporaryFile(delete=False, suffix='.mp3') as mp3_file:
-                audio.export(mp3_file.name, format="mp3")
-                self.current_audio_file = mp3_file.name
-                return self.current_audio_file
-
+            st.error("No video selected. Please select a video first.")
+            return None
+    
+        try:
+            video = YouTube(self.video_url)
+            audio_stream = video.streams.filter(only_audio=True).first()
+    
+            if not audio_stream:
+                st.error("No audio stream available for this video.")
+                return None
+    
+            # Create a temporary MP3 file
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_audio_file:
+                audio_stream.download(output_path=temp_audio_file.name)
+                temp_audio_file.close()
+    
+                audio = AudioSegment.from_file(temp_audio_file.name)
+                with tempfile.NamedTemporaryFile(delete=False, suffix='.mp3') as mp3_file:
+                    audio.export(mp3_file.name, format="mp3")
+                    self.current_audio_file = mp3_file.name
+                    return self.current_audio_file
+
+        except HTTPError as e:
+            st.error(f"HTTP error occurred: {e}")
+            return None
+        except Exception as e:
+            st.error(f"An error occurred while extracting audio: {e}")
+            return None
+            
     def play_audio(self):
         """Play the extracted audio using Streamlit's audio player."""
         if not self.current_audio_file: