Add file path to audio

app.py

@@ -59,6 +59,7 @@ def datapipeline(url: str) -> Any:
     json_scenes = text_processor.get_json_scenes(transcribed_text)
     print("Scenes ready")
     video = video_creator.create_video(json_scenes)
+    print("Video at", video)
     return video, video
 
 css = """

youtubeaudioextractor.py

@@ -11,15 +11,13 @@ class YouTubeAudioExtractor(ABC):
 class PytubeAudioExtractor(YouTubeAudioExtractor):
 
     def __init__(self,
-                 only_audio: bool = True,
-                 extension: str = ".mp3") -> None:
+                 only_audio: bool = True) -> None:
         self.only_audio = only_audio
-        self.extension = extension
 
-    def extract(self, url: str,
-                save_path: str = "yt_audio") -> str:
+    def extract(self,
+                url: str,
+                save_path: str = "yt_audio.mp3") -> str:
         yt = pt.YouTube(url)
         stream = yt.streams.filter(only_audio=self.only_audio)[0]
-        filename = save_path + self.extension
-        stream.download(filename=filename)
-        return filename
+        stream.download(filename=save_path)
+        return "yt_audio.mp3"