Update app.py

app.py

@@ -8,23 +8,16 @@ import re
 AA_KEY = os.getenv("AA_KEY")
 AA_HOST = "youtube-transcriptor.p.rapidapi.com"
 
+# 유튜브 URL에서 비디오 ID를 추출하는 함수
 def get_video_id(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만 입력된 경우
-    ]
+    # 유튜브 URL 또는 youtu.be 단축 URL에서 video_id 추출
+    video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
     
-    # URL에서 쿼리 파라미터 제거
-    url_without_params = youtube_url.split('?')[0]
+    # 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, url_without_params)
-        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']
@@ -46,27 +39,21 @@ def get_youtube_transcript(youtube_url):
     # 언어 우선순위에 따라 순차적으로 요청을 시도
     for lang in LANGUAGE_PRIORITY:
         querystring = {"video_id": video_id, "lang": lang}
-        try:
-            response = requests.get(url, headers=headers, params=querystring)
-            response.raise_for_status()  # 오류 상태 코드에 대해 예외 발생
-            
-            data = response.json()
-            
-            # 자막이 없는 경우 처리
-            if "transcript" not in data or not data["transcript"]:
-                continue
-            
-            return {"language": lang, "data": data}
+        response = requests.get(url, headers=headers, params=querystring)
+        
+        # 상태 코드 확인 및 전체 응답 반환
+        if response.status_code == 200:
+            try:
+                data = response.json()
+
+                # 전체 응답 데이터를 그대로 반환
+                return {"language": lang, "data": data}
             
-        except requests.exceptions.RequestException as e:
-            print(f"Error for language {lang}: {str(e)}")
-            continue
-        except json.JSONDecodeError as e:
-            print(f"JSON Decode Error for language {lang}: {str(e)}")
-            continue
+            except json.JSONDecodeError as e:
+                return {"error": f"JSON 디코딩 오류 발생: {str(e)}"}
 
     # 모든 언어에서 자막을 찾지 못한 경우
-    return {"error": "우선순위 언어로 자막을 찾을 수 없습니다. 해당 동영상에 자막이 없거나 접근할 수 없습니다."}
+    return {"error": "우선순위 언어로 자막을 찾을 수 없습니다."}
 
 # Gradio 인터페이스 정의
 def youtube_transcript_interface(youtube_url):