import os
from google_auth_oauthlib.flow import InstalledAppFlow
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError

## OAuth 2.0 클라이언트 ID 및 스코프 설정
CLIENT_SECRET_FILE = 'client_secret.json'
SCOPES = ['https://www.googleapis.com/auth/youtube.force-ssl']

def get_authenticated_service():
    # OAuth 2.0 인증 흐름 초기화
    flow = InstalledAppFlow.from_client_secrets_file(CLIENT_SECRET_FILE, SCOPES)
    credentials = flow.run_local_server()

    # 인증된 서비스 반환
    return build('youtube', 'v3', credentials=credentials)

def set_video_captions(youtube, video_id, english_caption, korean_caption):
    try:
        # 영어 자막 설정
        youtube.captions().insert(
            part='snippet',
            body={
                'snippet': {
                    'videoId': video_id,
                    'language': 'en',
                    'name': 'English',
                    'isDraft': False
                }
            },
            media_body=english_caption
        ).execute()

        # 한국어 자막 설정
        youtube.captions().insert(
            part='snippet',
            body={
                'snippet': {
                    'videoId': video_id,
                    'language': 'ko',
                    'name': 'Korean',
                    'isDraft': False
                }
            },
            media_body=korean_caption
        ).execute()

        print('영어와 한국어 자막 설정이 완료되었습니다.')

    except HttpError as e:
        print('자막 설정 중 오류 발생:', e)

if __name__ == '__main__':
    # OAuth 2.0 인증 및 YouTube 서비스 생성
    youtube = get_authenticated_service()

    # 동영상 ID 설정 (업로드한 동영상의 ID를 넣어주세요.)
    video_id = 'YOUR_VIDEO_ID_HERE'

    # 영어 자막 파일 경로와 한국어 자막 파일 경로 설정
    english_caption_path = 'YOUR_ENGLISH_CAPTION_FILE_PATH_HERE'  # 영어 자막 파일 경로로 바꿔주세요.
    korean_caption_path = 'YOUR_KOREAN_CAPTION_FILE_PATH_HERE'  # 한국어 자막 파일 경로로 바꿔주세요.

    # 영어 자막 파일과 한국어 자막 파일 열기
    with open(english_caption_path, 'r', encoding='utf-8') as english_file, \
            open(korean_caption_path, 'r', encoding='utf-8') as korean_file:
        english_caption = english_file.read()
        korean_caption = korean_file.read()

    set_video_captions(youtube, video_id, english_caption, korean_caption)