Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 from pytube import YouTube
 from crewai import Agent, Task, Crew
@@ -6,6 +5,7 @@ from transformers import pipeline
 import torch
 import whisper
 import yt_dlp
+import os
 
 st.title("AI Blog Writer from YouTube")
 
@@ -13,11 +13,16 @@ st.title("AI Blog Writer from YouTube")
 whisper_model = whisper.load_model("base")
 
 def download_audio(video_url):
-    """Downloads audio from the YouTube video using yt-dlp."""
+    """Downloads audio from the YouTube video using yt-dlp with cookies."""
+    # Get the path to the cookies file from the current directory
+    cookies_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'cookies.json')
+    
     ydl_opts = {
         'format': 'bestaudio/best',
         'outtmpl': 'audio.mp4',
+        'cookies': cookies_file,  # Path to the cookies file
     }
+    
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info_dict = ydl.extract_info(video_url, download=True)
     return 'audio.mp4'
@@ -47,7 +52,7 @@ if st.button("Generate Blog"):
     if video_url:
         try:
             st.info("Downloading audio...")
-            audio_path = download_audio(video_url)
+            audio_path = download_audio(video_url)  # No need to ask for cookies file path anymore
             st.success("Audio downloaded successfully!")
             
             st.info("Transcribing audio using Whisper...")