Update app.py

app.py

@@ -10,22 +10,14 @@ AA_HOST = "youtube-transcriptor.p.rapidapi.com"
 
 # 유튜브 URL에서 비디오 ID를 추출하는 함수
 def get_video_id(youtube_url):
-    # 일반적인 YouTube URL (watch?v=...)
-    match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11}).*", youtube_url)
-    if match:
-        return match.group(1)
+    # 유튜브 URL 또는 youtu.be 단축 URL에서 video_id 추출
+    video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
     
-    # 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 처리
+    if not video_id_match:
+        video_id_match = re.search(r"(?<=shorts/)[^#&?]*", youtube_url)
     
-    # YouTube Shorts URL (/shorts/...)
-    match = re.search(r"\/shorts\/([0-9A-Za-z_-]{11})", youtube_url)
-    if match:
-        return match.group(1)
-    
-    return None
+    return video_id_match.group(0) if video_id_match else None
 
 # 자막 언어 우선순위 리스트
 LANGUAGE_PRIORITY = ['ko', 'en', 'ja', 'zh']
@@ -49,20 +41,16 @@ def get_youtube_transcript(youtube_url):
         querystring = {"video_id": video_id, "lang": lang}
         response = requests.get(url, headers=headers, params=querystring)
         
+        # 상태 코드 확인 및 전체 응답 반환
         if response.status_code == 200:
             try:
                 data = response.json()
-                
-                # API 응답 구조 확인 및 올바른 처리
-                if isinstance(data, dict) and "transcript" in data:
-                    return {"language": lang, "data": data}
-                elif isinstance(data, list) and len(data) > 0 and "text" in data[0]:
-                    return {"language": lang, "data": {"transcript": data}}
-                else:
-                    continue  # 현재 언어에 자막이 없으면 다음 언어로 시도
+
+                # 전체 응답 데이터를 그대로 반환
+                return {"language": lang, "data": data}
             
             except json.JSONDecodeError as e:
-                continue  # JSON 디코딩 오류 시 다음 언어로 시도
+                return {"error": f"JSON 디코딩 오류 발생: {str(e)}"}
 
     # 모든 언어에서 자막을 찾지 못한 경우
     return {"error": "우선순위 언어로 자막을 찾을 수 없습니다."}