Update app.py

app.py

@@ -10,14 +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)
+    # 다양한 YouTube URL 형식에 대한 정규표현식 패턴
+    patterns = [
+        r"(?:v=|\/)([0-9A-Za-z_-]{11}).*",  # 일반적인 YouTube URL 및 Shorts 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)
+    # URL에서 쿼리 파라미터 제거
+    url_without_params = youtube_url.split('?')[0]
     
-    return video_id_match.group(0) if video_id_match else None
+    for pattern in patterns:
+        match = re.search(pattern, url_without_params)
+        if match:
+            return match.group(1)
+    
+    return None
 
 # 자막 언어 우선순위 리스트
 LANGUAGE_PRIORITY = ['ko', 'en', 'ja', 'zh']