Update app.py

app.py

@@ -16,6 +16,12 @@ import torch
 from urllib.parse import urlparse, parse_qs
 
 
+YOUTUBE_API_KEY = os.getenv("YOUTUBE_API_KEY")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+if not YOUTUBE_API_KEY:
+    return "YouTube API key is missing! Please set the `YOUTUBE_API_KEY` environment variable."
+if not OPENAI_API_KEY:
+    return "OpenAI API key is missing! Please set the `OPENAI_API_KEY` environment variable."
 
 def extract_video_id(url):
     """Extracts the video ID from a YouTube URL."""
@@ -62,8 +68,6 @@ def download_and_transcribe_with_whisper(youtube_url):
             ydl_opts = {
                 'format': 'bestaudio/best',
                 'outtmpl': temp_audio_file,
-                'extractaudio': True,
-                'audioquality': 1,
             }
 
             # Download audio using yt-dlp
@@ -76,7 +80,7 @@ def download_and_transcribe_with_whisper(youtube_url):
             audio.export(wav_file, format="wav")
 
             # Run Whisper transcription
-            model = whisper.load_model("turbo")
+            model = whisper.load_model("turbo",weights_only=True)
             result = model.transcribe(wav_file)
             transcript = result['text']
             return transcript
@@ -150,8 +154,11 @@ def get_transcript(youtube_url, api_key):
         return None
 
 def summarize_text_huggingface(text):
+    
     """Summarizes text using a Hugging Face summarization model."""
-    summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=0 if torch.cuda.is_available() else -1)
+    device = 0 if torch.cuda.is_available() else -1
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=device)
+
     max_input_length = 1024
     chunk_overlap = 100
     text_chunks = [
@@ -203,8 +210,7 @@ def generate_optimized_content(api_key, summarized_transcript):
     except Exception as e:
         print(f"Error generating content: {e}")
         return None
-YOUTUBE_API_KEY = os.getenv("YOUTUBE_API_KEY")
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
 
 # Add all your functions like `extract_video_id()`, `get_transcript()`, etc.