Update app.py

app.py

@@ -1,20 +1,31 @@
-
 import gradio as gr
 from youtube_transcript_api import YouTubeTranscriptApi
+from urllib.parse import urlparse, parse_qs
 
-def extract_script(url):
-    # '=' 기호 이후 텍스트 추출
+def extract_video_id(url):
     try:
-        video_id = url.split('=')[-1]
+        parsed_url = urlparse(url)
+        if parsed_url.netloc == 'www.youtube.com':
+            query_params = parse_qs(parsed_url.query)
+            video_id = query_params.get('v')
+            if video_id:
+                return video_id[0]
+        elif parsed_url.netloc == 'youtu.be':
+            video_id = parsed_url.path[1:]
+            return video_id
+    except Exception:
+        return None
 
-        # 자막(스크립트) 가져오기
+def extract_script(url):
+    video_id = extract_video_id(url)
+    if not video_id:
+        return "잘못된 유튜브 URL입니다."
+    try:
         srt = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko'])
-
-        # 자막 텍스트 합치기
         script = "\n".join([i["text"] for i in srt])
         return script
     except Exception as e:
-        return f"오류 발생: {e}"
+        return f"스크립트를 가져오는 중 오류가 발생했습니다: {e}"
 
 # Gradio 인터페이스 정의
 with gr.Blocks() as app: