Update app.py

app.py

@@ -2,15 +2,27 @@ import os
 import requests
 import json
 import gradio as gr
+import re
 
 # Hugging Face 환경 변수로부터 RapidAPI 키와 호스트 가져오기
 RAPIDAPI_KEY = os.getenv("RAPIDAPI_KEY")
 RAPIDAPI_HOST = "youtube-transcriptor.p.rapidapi.com"
 
+# 유튜브 URL에서 비디오 ID를 추출하는 함수
+def get_video_id(youtube_url):
+    # 일반적인 YouTube URL 또는 youtu.be 단축 URL에서 video_id 추출
+    video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
+    return video_id_match.group(0) if video_id_match else None
+
 # 유튜브 자막을 요청하는 함수
-def get_youtube_transcript(video_id, lang):
+def get_youtube_transcript(youtube_url, lang_code):
+    # 비디오 ID 추출
+    video_id = get_video_id(youtube_url)
+    if video_id is None:
+        return {"error": "잘못된 유튜브 URL입니다. 비디오 ID를 찾을 수 없습니다."}
+    
     url = f"https://{RAPIDAPI_HOST}/transcript"
-    querystring = {"video_id": video_id, "lang": lang}
+    querystring = {"video_id": video_id, "lang": lang_code}
     
     headers = {
         "x-rapidapi-key": RAPIDAPI_KEY,
@@ -27,9 +39,9 @@ def get_youtube_transcript(video_id, lang):
         return {"error": str(e)}
 
 # Gradio 인터페이스 정의
-def youtube_transcript_interface(video_id, lang):
+def youtube_transcript_interface(youtube_url, lang_code):
     # 자막 데이터 가져오기
-    transcript_data = get_youtube_transcript(video_id, lang)
+    transcript_data = get_youtube_transcript(youtube_url, lang_code)
     
     # JSON 포맷으로 출력
     return json.dumps(transcript_data, ensure_ascii=False, indent=2)
@@ -39,8 +51,8 @@ interface = gr.Interface(
     fn=youtube_transcript_interface, 
     inputs=["text", "text"], 
     outputs="text",
-    title="YouTube Transcript Fetcher",
-    description="Enter YouTube Video ID and Language Code (e.g., 'ko' for Korean)."
+    title="YouTube 자막 추출기",
+    description="유튜브 URL과 언어 코드를 입력하세요 (예: 'ko' for Korean)."
 )
 
 # Gradio 인터페이스 실행