Update app.py

app.py

@@ -1,31 +1,48 @@
 import gradio as gr
-from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound
 
+def get_youtube_transcript(video_id):
+    """
+    Retrieves the transcript of a YouTube video.
 
-def transkribiere_youtube_video(video_url):
-#3.Get transcripts
-    transcript = YouTubeTranscriptApi.get_transcript("7ySqrQVLIF8")
-    return transcript
+    Args:
+        video_id: The ID of the YouTube video (e.g., 'dQw4w9WgXcQ').
 
-
-    
+    Returns:
+        A string containing the full transcript, or None if no transcript is found or an error occurs.
+    """
     try:
-        video_id = video_url.split("v=")[1]
-        if "&" in video_id:
-             video_id = video_id.split("&")[0] # falls zusätzliche Parameter in URL
-        transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=['de', 'en']) # Transkripte in DE & EN (falls vorhanden)
-        
-        text = " ".join([entry['text'] for entry in transcript])
-        return text
+        transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+        transcript = transcript_list.find_generated_transcript(['en'])  # Try English first
+        if not transcript:
+            transcript = transcript_list.find_manually_created_transcript(['en'])  # Fallback to manual if no generated
+        if not transcript:
+            return None  # No English transcript found
+        transcript_data = transcript.fetch()
+        full_transcript = " ".join([item['text'] for item in transcript_data])
+        return full_transcript
+    except TranscriptsDisabled:
+        return f"Transcripts are disabled for video ID: {video_id}"
+    except NoTranscriptFound:
+        return f"No transcript found for video ID: {video_id}"
     except Exception as e:
-       return f"Fehler: {str(e)}. Stelle sicher, dass ein Transkript für das Video verfügbar ist und der Link korrekt ist."
+        return f"An error occurred: {e}"
+
+def fetch_transcript(video_id):
+    transcript = get_youtube_transcript(video_id)
+    if transcript:
+        return transcript
+    else:
+        return "Could not retrieve the transcript."
 
+# Create the Gradio interface
 iface = gr.Interface(
-    fn=transkribiere_youtube_video,
-    inputs=gr.Textbox(lines=1, placeholder="Gib hier den YouTube Video-Link ein"),
-    outputs=gr.Textbox(lines=20, label="Transkript"),
-    title="YouTube Transkription",
-    description="Gib einen YouTube-Videolink ein und erhalte den transkribierten Text."
+    fn=fetch_transcript,
+    inputs=gr.inputs.Textbox(lines=1, placeholder="Enter YouTube Video ID..."),
+    outputs=gr.outputs.Textbox(),
+    title="YouTube Transcript Fetcher",
+    description="Enter a YouTube video ID to fetch its transcript."
 )
 
+# Launch the Gradio app
 iface.launch()