Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import os
 from together import Together
-from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound, VideoUnavailable
 from datetime import datetime
 
 # Get API Key
@@ -10,13 +10,20 @@ TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")
 def extract_transcript(video_id):
     """Extracts transcript from a YouTube video."""
     try:
-        transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+        ytt_api = YouTubeTranscriptApi()
+        transcript_list = ytt_api.list_transcripts(video_id)
         transcript = transcript_list.find_transcript(['en'])
-        if not transcript.is_generated:
+        if transcript.is_translatable:
             transcript = transcript.translate('en')
         transcript_segments = transcript.fetch()
         full_transcript = ' '.join(segment['text'] for segment in transcript_segments)
         return full_transcript
+    except TranscriptsDisabled:
+        return "Error: Transcripts are disabled for this video."
+    except NoTranscriptFound:
+        return "Error: No transcript found for this video."
+    except VideoUnavailable:
+        return "Error: Video is unavailable."
     except Exception as e:
         return f"Error: {e}"
 
@@ -39,7 +46,7 @@ def generate_summary(transcript_text):
         Summary:
         '''
         response = client.chat.completions.create(
-            model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free",
+            model="meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
             messages=[{"role": "user", "content": prompt}]
         )
         summary = response.choices[0].message.content
@@ -50,11 +57,11 @@ def generate_summary(transcript_text):
 def summarize_youtube(video_id):
     """Extracts transcript and generates a summary."""
     transcript = extract_transcript(video_id)
-    if "Error" in transcript:
+    if transcript.startswith("Error"):
         return transcript, None
     
     summary = generate_summary(transcript)
-    if "Error" in summary:
+    if summary.startswith("Error"):
         return summary, None
     
     filename = f"{video_id}_summary.txt"