Update app.py

app.py

@@ -1,31 +1,20 @@
+
 import gradio as gr
 from youtube_transcript_api import YouTubeTranscriptApi
-from urllib.parse import urlparse, parse_qs
-
-def extract_video_id(url):
-    try:
-        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:
+        video_id = url.split('=')[-1]
+
+        # 자막(스크립트) 가져오기
         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: