Update app.py

app.py

@@ -1,65 +1,126 @@
 import gradio as gr
 from youtube_transcript_api import YouTubeTranscriptApi
-from youtube_transcript_api._errors import NoTranscriptFound, TranscriptsDisabled
-import re
+from youtube_transcript_api._errors import NoTranscriptFound, TranscriptsDisabled, NoTranscriptAvailable
+import requests
+import json
+import os
+
+def get_video_info(video_id):
+    """YouTube 영상 정보를 가져오는 함수"""
+    try:
+        response = requests.get(f'https://www.youtube.com/oembed?url=http://www.youtube.com/watch?v={video_id}&format=json')
+        return response.json()
+    except:
+        return None
 
 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
+    try:
+        if 'youtube.com/watch?v=' in url:
+            video_id = url.split('youtube.com/watch?v=')[1].split('&')[0][:11]
+        elif 'youtu.be/' in url:
+            video_id = url.split('youtu.be/')[1].split('?')[0][:11]
+        return video_id
+    except:
+        return None
+
+def get_available_transcripts(video_id):
+    """사용 가능한 모든 자막 목록을 가져오는 함수"""
+    try:
+        transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+        return transcript_list
+    except:
+        return None
 
 def get_transcript(url):
     """YouTube 영상의 스크립트를 추출하는 함수"""
     try:
+        # 비디오 ID 추출
         video_id = extract_video_id(url)
         if not video_id:
             return "올바른 YouTube URL을 입력해주세요."
         
+        # 영상 정보 확인
+        video_info = get_video_info(video_id)
+        if not video_info:
+            return "영상 정보를 가져올 수 없습니다. URL을 확인해주세요."
+
+        # 자막 추출 시도
+        transcript_list = None
+        transcript_data = None
+        
         try:
-            # 먼저 한국어 자막 시도
-            transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko'])
-        except NoTranscriptFound:
-            try:
-                # 한국어 자막이 없으면 영어 자막 시도
-                transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['en'])
-            except NoTranscriptFound:
+            # 모든 가능한 자막 목록 가져오기
+            transcript_list = get_available_transcripts(video_id)
+            
+            if transcript_list:
+                # 1. 수동 한국어 자막 시도
                 try:
-                    # 일반 자막이 없으면 자동 생성된 자막 시도
-                    transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
-                    transcript = transcript_list.find_generated_transcript(['ko', 'en'])
-                    transcript_list = transcript.fetch()
+                    transcript = transcript_list.find_manually_created_transcript(['ko'])
+                    transcript_data = transcript.fetch()
                 except:
-                    return "이 영상에는 자막이 없거나 자막을 가져올 수 없습니다."
-        
-        # 전체 스크립트 텍스트 구성
-        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 TranscriptsDisabled:
-        return "이 영상은 자막이 비활성화되어 있습니다."
+                    # 2. 자동 생성된 한국어 자막 시도
+                    try:
+                        transcript = transcript_list.find_generated_transcript(['ko'])
+                        transcript_data = transcript.fetch()
+                    except:
+                        # 3. 영어 자막 시도 (수동 또는 자동)
+                        try:
+                            transcript = transcript_list.find_transcript(['en'])
+                            transcript_data = transcript.fetch()
+                        except:
+                            return "지원되는 자막을 찾을 수 없습니다."
+            
+            if not transcript_data:
+                # 직접 API로 시도
+                transcript_data = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko', 'en'])
+            
+            # 전체 스크립트 텍스트 구성
+            full_transcript = f"제목: {video_info.get('title', '제목 없음')}\n\n"
+            for transcript in transcript_data:
+                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 NoTranscriptAvailable:
+            return "이 영상에는 자막이 없습니다."
+        except TranscriptsDisabled:
+            return "이 영상은 자막이 비활성화되어 있습니다."
+        except Exception as e:
+            return f"자막 추출 중 오류가 발생했습니다: {str(e)}"
+
     except Exception as e:
-        return f"오류가 발생했습니다: {str(e)}"
+        return f"처리 중 오류가 발생했습니다: {str(e)}"
 
 # Gradio 인터페이스 생성
 iface = gr.Interface(
     fn=get_transcript,
-    inputs=gr.Textbox(label="YouTube URL을 입력하세요"),
-    outputs=gr.Textbox(label="추출된 스크립트", lines=10),
+    inputs=[
+        gr.Textbox(label="YouTube URL을 입력하세요", placeholder="https://www.youtube.com/watch?v=...")
+    ],
+    outputs=gr.Textbox(label="추출된 스크립트", lines=20),
     title="YouTube 스크립트 추출기",
-    description="YouTube 영상의 URL을 입력하면 자동으로 스크립트를 추출합니다. (일반 자막 또는 자동 생성된 자막)"
+    description="""
+    YouTube 영상의 URL을 입력하면 자동으로 스크립트를 추출합니다.
+    - 한국어 자막 (수동/자동)
+    - 영어 자막 (수동/자동)
+    을 순차적으로 시도합니다.
+    """,
+    examples=[
+        ["https://www.youtube.com/watch?v=example1"],
+        ["https://youtu.be/example2"]
+    ]
 )
 
+# 환경변수 설정
+os.environ['GRADIO_SERVER_NAME'] = "0.0.0.0"
+os.environ['GRADIO_SERVER_PORT'] = "7860"
+
 # 애플리케이션 실행
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(server_name="0.0.0.0", server_port=7860)