Update app.py

app.py

@@ -10,7 +10,7 @@ RAPIDAPI_HOST = "youtube-transcriptor.p.rapidapi.com"
 
 # 유튜브 URL에서 비디오 ID를 추출하는 함수
 def get_video_id(youtube_url):
-    # 유튜브 URL에서 video_id 추출 (일반적인 YouTube URL 또는 youtu.be 단축 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
 
@@ -22,7 +22,6 @@ def get_youtube_transcript(youtube_url):
     # 비디오 ID 추출
     video_id = get_video_id(youtube_url)
     if video_id is None:
-        print("ERROR: 잘못된 유튜브 URL입니다. 비디오 ID를 찾을 수 없습니다.")
         return {"error": "잘못된 유튜브 URL입니다. 비디오 ID를 찾을 수 없습니다."}
     
     url = "https://youtube-transcriptor.p.rapidapi.com/transcript"
@@ -35,39 +34,20 @@ def get_youtube_transcript(youtube_url):
     # 언어 우선순위에 따라 순차적으로 요청을 시도
     for lang in LANGUAGE_PRIORITY:
         querystring = {"video_id": video_id, "lang": lang}
-        print(f"요청 중: {lang} 언어에 대한 자막 요청")
-        
         response = requests.get(url, headers=headers, params=querystring)
         
-        # 상태 코드 출력
-        print(f"Response status code: {response.status_code}")
-        
-        # 원시 응답 데이터를 출력
-        print("Raw response content:", response.text)
-        
-        # 상태 코드가 200일 때만 처리
+        # 상태 코드 확인 및 전체 응답 반환
         if response.status_code == 200:
             try:
-                # JSON으로 응답을 파싱
                 data = response.json()
-                
-                # 응답 데이터 출력
-                print("Parsed JSON response:", json.dumps(data, indent=2, ensure_ascii=False))
 
-                # transcription 필드를 통해 자막 데이터가 존재하는지 확인
-                if "transcription" in data and isinstance(data["transcription"], list):
-                    # 자막 데이터를 JSON 리스트로 반환
-                    return {"language": lang, "transcription": data["transcription"]}
-                else:
-                    print(f"ERROR: 자막 데이터를 {lang} 언어로 찾을 수 없습니다.")
-                    return {"error": f"{lang} 언어로 자막 데이터를 찾을 수 없습니다."}
+                # 전체 응답 데이터를 그대로 반환
+                return {"language": lang, "data": data}
             
             except json.JSONDecodeError as e:
-                print(f"ERROR: JSON 디코딩 오류 발생: {str(e)}")
                 return {"error": f"JSON 디코딩 오류 발생: {str(e)}"}
 
-    # 자막을 찾지 못한 경우
-    print("ERROR: 우선순위 언어로 자막을 찾을 수 없습니다.")
+    # 모든 언어에서 자막을 찾지 못한 경우
     return {"error": "우선순위 언어로 자막을 찾을 수 없습니다."}
 
 # Gradio 인터페이스 정의