Update thaTube.py

thaTube.py

@@ -46,14 +46,12 @@ class thaTube:
                 self.current_audio_file = mp3_file.name
                 return self.current_audio_file
 
-    def play_audio(self):
-        """Play the extracted audio."""
+   def play_audio(self):
+        """Play the extracted audio using Streamlit's audio player."""
         if not self.current_audio_file:
             raise ValueError("No audio file to play. Please extract audio first.")
-
-        self.current_audio = pygame.mixer.Sound(self.current_audio_file)
-        self.current_audio.play()
-        self.is_playing = True
+        audio_bytes = open(self.current_audio_file, 'rb').read()
+        st.audio(audio_bytes, format='audio/mpeg', start_time=0)
 
     def stop_audio(self):
         """Stop the currently playing audio."""