Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api._errors import NoTranscriptFound, TranscriptsDisabled
 import re
 
 def extract_video_id(url):
@@ -18,7 +19,21 @@ def get_transcript(url):
         if not video_id:
             return "올바른 YouTube URL을 입력해주세요."
         
-        transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko', 'en'])
+        try:
+            # 먼저 한국어 자막 시도
+            transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko'])
+        except NoTranscriptFound:
+            try:
+                # 한국어 자막이 없으면 영어 자막 시도
+                transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['en'])
+            except NoTranscriptFound:
+                try:
+                    # 일반 자막이 없으면 자동 생성된 자막 시도
+                    transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+                    transcript = transcript_list.find_generated_transcript(['ko', 'en'])
+                    transcript_list = transcript.fetch()
+                except:
+                    return "이 영상에는 자막이 없거나 자막을 가져올 수 없습니다."
         
         # 전체 스크립트 텍스트 구성
         full_transcript = ""
@@ -31,6 +46,8 @@ def get_transcript(url):
         
         return full_transcript
     
+    except TranscriptsDisabled:
+        return "이 영상은 자막이 비활성화되어 있습니다."
     except Exception as e:
         return f"오류가 발생했습니다: {str(e)}"
 
@@ -40,7 +57,7 @@ iface = gr.Interface(
     inputs=gr.Textbox(label="YouTube URL을 입력하세요"),
     outputs=gr.Textbox(label="추출된 스크립트", lines=10),
     title="YouTube 스크립트 추출기",
-    description="YouTube 영상의 URL을 입력하면 자동으로 스크립트를 추출합니다."
+    description="YouTube 영상의 URL을 입력하면 자동으로 스크립트를 추출합니다. (일반 자막 또는 자동 생성된 자막)"
 )
 
 # 애플리케이션 실행