Update app.py

app.py

@@ -1,29 +1,48 @@
-
 import gradio as gr
 from youtube_transcript_api import YouTubeTranscriptApi
+import re
 
-def extract_script(url):
-    # '=' 기호 이후 텍스트 추출
-    try:
-        video_id = url.split('=')[-1]
-
-        # 자막(스크립트) 가져오기
-        srt = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko'])
+def extract_video_id(url):
+    """YouTube URL에서 비디오 ID를 추출하는 함수"""
+    video_id = None
+    if 'youtube.com/watch?v=' in url:
+        video_id = url.split('youtube.com/watch?v=')[1][:11]
+    elif 'youtu.be/' in url:
+        video_id = url.split('youtu.be/')[1][:11]
+    return video_id
 
-        # 자막 텍스트 합치기
-        script = "\n".join([i["text"] for i in srt])
-        return script
+def get_transcript(url):
+    """YouTube 영상의 스크립트를 추출하는 함수"""
+    try:
+        video_id = extract_video_id(url)
+        if not video_id:
+            return "올바른 YouTube URL을 입력해주세요."
+        
+        transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko', 'en'])
+        
+        # 전체 스크립트 텍스트 구성
+        full_transcript = ""
+        for transcript in transcript_list:
+            text = transcript['text']
+            timestamp = transcript['start']
+            minutes = int(timestamp // 60)
+            seconds = int(timestamp % 60)
+            full_transcript += f"[{minutes:02d}:{seconds:02d}] {text}\n"
+        
+        return full_transcript
+    
     except Exception as e:
-        return f"오류 발생: {e}"
-
-# Gradio 인터페이스 정의
-with gr.Blocks() as app:
-    gr.Markdown("## 유튜브 스크립트 추출기")
-    url_input = gr.Textbox(label="유튜브 URL", placeholder="예제 https://www.youtube.com/watch?v=HtMVLneHZR0")
-    output = gr.Textbox(label="스크립트 출력")
-    extract_button = gr.Button("스크립트 추출")
+        return f"오류가 발생했습니다: {str(e)}"
 
-    extract_button.click(extract_script, inputs=url_input, outputs=output)
+# Gradio 인터페이스 생성
+iface = gr.Interface(
+    fn=get_transcript,
+    inputs=gr.Textbox(label="YouTube URL을 입력하세요"),
+    outputs=gr.Textbox(label="추출된 스크립트", lines=10),
+    title="YouTube 스크립트 추출기",
+    description="YouTube 영상의 URL을 입력하면 자동으로 스크립트를 추출합니다."
+)
 
-# 웹 앱 실행
-app.launch()
+# 애플리케이션 실행
+if __name__ == "__main__":
+    iface.launch()