Update app.py

app.py

@@ -10,14 +10,19 @@ AA_HOST = "youtube-transcriptor.p.rapidapi.com"
 
 # 유튜브 URL에서 비디오 ID를 추출하는 함수
 def get_video_id(youtube_url):
-    # 유튜브 URL 또는 youtu.be 단축 URL에서 video_id 추출
-    video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
+    # 정규표현식 패턴
+    patterns = [
+        r"(?:v=|/)([0-9A-Za-z_-]{11})(?:[&?]|$)",  # 일반적인 YouTube URL
+        r"(?:youtu\.be/|youtube\.com/shorts/)([0-9A-Za-z_-]{11})(?:[&?]|$)",  # 단축 URL 및 Shorts URL
+        r"^([0-9A-Za-z_-]{11})$"  # 비디오 ID만 입력된 경우
+    ]
     
-    # YouTube Shorts URL 처리
-    if not video_id_match:
-        video_id_match = re.search(r"(?<=shorts/)[^#&?]*", youtube_url)
+    for pattern in patterns:
+        match = re.search(pattern, youtube_url)
+        if match:
+            return match.group(1)
     
-    return video_id_match.group(0) if video_id_match else None
+    return None
 
 # 자막 언어 우선순위 리스트
 LANGUAGE_PRIORITY = ['ko', 'en', 'ja', 'zh']