Update app.py

app.py

@@ -10,9 +10,22 @@ 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)
-    return video_id_match.group(0) if video_id_match else None
+    # 일반적인 YouTube URL (watch?v=...)
+    match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11}).*", youtube_url)
+    if match:
+        return match.group(1)
+    
+    # YouTube Shorts URL
+    match = re.search(r"youtu\.be\/([0-9A-Za-z_-]{11})", youtube_url)
+    if match:
+        return match.group(1)
+    
+    # YouTube Shorts URL (/shorts/...)
+    match = re.search(r"\/shorts\/([0-9A-Za-z_-]{11})", youtube_url)
+    if match:
+        return match.group(1)
+    
+    return None
 
 # 자막 언어 우선순위 리스트
 LANGUAGE_PRIORITY = ['ko', 'en', 'ja', 'zh']