Update app.py

app.py

@@ -12,6 +12,7 @@ from huggingface_hub import InferenceClient
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api.formatters import TextFormatter
 from dotenv import load_dotenv
+import time
 
 # 환경 변수 로드
 load_dotenv()
@@ -44,12 +45,13 @@ def authorize():
     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}' 파일이 존재하지 않습니다.")
+    print(f'Enter the authorization code in the file {auth_code_path}')
     
+    # 파일에 인증 코드가 저장될 때까지 기다림
+    while not os.path.exists(auth_code_path):
+        print("Waiting for authorization code...")
+        time.sleep(5)
+
     with open(auth_code_path, 'r') as file:
         code = file.read().strip()
     
@@ -59,6 +61,9 @@ def authorize():
     with open(token_path, 'w') as token:
         token.write(creds.to_json())
     
+    # 인증 코드 파일 삭제 (보안)
+    os.remove(auth_code_path)
+
     return creds
 
 # 기존 토큰을 로드하거나 새로 인증