Update functions.py

functions.py

@@ -1,33 +1,48 @@
 from pytube import YouTube
 from youtube_transcript_api import YouTubeTranscriptApi
 import re
+import os
 
 def download_video_transcript(video_url):
     try:
-        video_id = re.search(r"(?<=v=)[\w-]+", video_url).group()
-        yt = YouTube(video_url)
-        stream = yt.streams.get_highest_resolution()
+        video_id = re.search(r"(?<=v=)[\w-]+", video_url)
+        if video_id:
+            video_id = video_id.group()
+            yt = YouTube(video_url)
+            stream = yt.streams.get_highest_resolution()
 
-        print(f'Downloading {yt.title}...')
-        modified_title = yt.title.replace(" ", "_")
-        download_path = 'Milestone1/videos'
-        stream.download(output_path=download_path, filename=modified_title + '.mp4')
-        print('Download completed!')
+            print(f'Downloading {yt.title}...')
+            modified_title = yt.title.replace(" ", "_")
+            download_path = 'CS370_Milestone5/videos'  # Updated path to CS370_Milestone5 repository
+            captions_path = 'CS370_Milestone5/captions'  # Updated path to CS370_Milestone5 repository
 
-        transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
-        transcript = transcript_list.find_generated_transcript(['en']).fetch()
+            # Create directories if they don't exist
+            os.makedirs(download_path, exist_ok=True)
+            os.makedirs(captions_path, exist_ok=True)
 
-        captions = ""
-        for i, line in enumerate(transcript):
-            start_time = line['start']
-            formatted_time = f"{int(start_time // 60):02d}:{int(start_time % 60):02d}"
-            captions += f"{formatted_time} {line['text']}\n"
+            video_file = f'{download_path}/{modified_title}.mp4'
+            stream.download(output_path=download_path, filename=modified_title + '.mp4')
+            print('Download completed!')
 
-        with open(f'Milestone1/captions/{modified_title}.txt', 'w', encoding='utf-8') as file:
-            file.write(captions)
-        
-        return captions
+            transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+            transcript = transcript_list.find_generated_transcript(['en']).fetch()
+
+            captions = ""
+            for i, line in enumerate(transcript):
+                start_time = line['start']
+                formatted_time = f"{int(start_time // 60):02d}:{int(start_time % 60):02d}"
+                captions += f"{formatted_time} {line['text']}\n"
+
+            with open(f'{captions_path}/{modified_title}.txt', 'w', encoding='utf-8') as file:
+                file.write(captions)
+            
+            # Delete video file after captions are saved
+            os.remove(video_file)
+            
+            return captions
+        else:
+            print("Video ID not found in URL.")
+            return None
     except Exception as e:
         print("Error:", e)
         return None
-