Update app.py

app.py

@@ -1,14 +1,14 @@
+import os
+import google_auth_oauthlib.flow
+from google.oauth2.credentials import Credentials
+from googleapiclient.discovery import build
+from google.auth.transport.requests import Request
 import discord
 import logging
-import os
 import re
 import asyncio
 import subprocess
 from huggingface_hub import InferenceClient
-from googleapiclient.discovery import build
-from google.oauth2.credentials import Credentials
-from google_auth_oauthlib.flow import InstalledAppFlow
-from google.auth.transport.requests import Request
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api.formatters import TextFormatter
 from dotenv import load_dotenv
@@ -19,6 +19,7 @@ load_dotenv()
 # JSON_TOKEN.json 파일의 경로
 credentials_path = 'JSON_TOKEN.json'
 token_path = 'token.json'
+auth_code_path = 'auth_code.txt'  # 인증 코드를 저장할 파일 경로
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s:%(message)s', handlers=[logging.StreamHandler()])
@@ -38,33 +39,45 @@ SCOPES = ["https://www.googleapis.com/auth/youtube.force-ssl"]
 creds = None
 
 def authorize():
-    flow = InstalledAppFlow.from_client_secrets_file(credentials_path, SCOPES)
+    flow = google_auth_oauthlib.flow.InstalledAppFlow.from_client_secrets_file(
+        credentials_path, SCOPES)
     auth_url, _ = flow.authorization_url(prompt='consent')
     print('Please go to this URL and finish the authentication: {}'.format(auth_url))
+
+    print(f'Enter the authorization code in the file {auth_code_path} and press Enter')
+    input('Press Enter after saving the authorization code...')
+
+    if not os.path.exists(auth_code_path):
+        raise FileNotFoundError(f"'{auth_code_path}' 파일이 존재하지 않습니다.")
     
-    # 인증 코드를 입력하도록 요청
-    code = input('Enter the authorization code: ')
+    with open(auth_code_path, 'r') as file:
+        code = file.read().strip()
     
     flow.fetch_token(code=code)
     creds = flow.credentials
+    
     with open(token_path, 'w') as token:
         token.write(creds.to_json())
+    
     return creds
 
+# 기존 토큰을 로드하거나 새로 인증
 if os.path.exists(token_path):
     creds = Credentials.from_authorized_user_file(token_path, SCOPES)
 else:
     creds = authorize()
 
+# 토큰 갱신 또는 재인증
 if not creds or not creds.valid:
     if creds and creds.expired and creds.refresh_token:
         creds.refresh(Request())
     else:
         creds = authorize()
 
+# YouTube API 클라이언트 생성
 youtube_service = build('youtube', 'v3', credentials=creds)
 
-# 특정 채널 ID
+# 디스코드 봇 설정
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
 
 class MyClient(discord.Client):