Update app.py

app.py

@@ -5,6 +5,9 @@ import re
 import asyncio
 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
@@ -25,9 +28,28 @@ intents.guild_messages = True
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
 
-# YouTube API 키 설정
+# YouTube API 키 설정 및 OAuth 2.0 인증 설정
 YOUTUBE_API_KEY = os.getenv("YOUTUBE_API_KEY")
-youtube_service = build('youtube', 'v3', developerKey=YOUTUBE_API_KEY)
+SCOPES = ["https://www.googleapis.com/auth/youtube.force-ssl"]
+creds = None
+
+# 환경변수 설정 (토큰을 저장할 파일 경로)
+TOKEN_PATH = 'token.json'
+CREDENTIALS_PATH = 'credentials.json'
+
+if os.path.exists(TOKEN_PATH):
+    creds = Credentials.from_authorized_user_file(TOKEN_PATH, SCOPES)
+if not creds or not creds.valid:
+    if creds and creds.expired and creds.refresh_token:
+        creds.refresh(Request())
+    else:
+        flow = InstalledAppFlow.from_client_secrets_file(
+            CREDENTIALS_PATH, SCOPES)
+        creds = flow.run_local_server(port=0)
+    with open(TOKEN_PATH, 'w') as token:
+        token.write(creds.to_json())
+
+youtube_service = build('youtube', 'v3', credentials=creds)
 
 # 특정 채널 ID
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
@@ -61,6 +83,7 @@ class MyClient(discord.Client):
                 if comments and transcript:
                     replies = await generate_replies(comments, transcript)
                     await create_thread_and_send_replies(message, video_id, comments, replies)
+                    await post_replies_to_youtube(video_id, comments, replies)
                 else:
                     await message.channel.send("자막이나 댓글을 가져올 수 없습니다.")
             else:
@@ -131,7 +154,8 @@ async def get_video_comments(video_id):
 
     for item in response.get('items', []):
         comment = item['snippet']['topLevelComment']['snippet']['textOriginal']
-        comments.append(comment)
+        comment_id = item['snippet']['topLevelComment']['id']
+        comments.append((comment, comment_id))  # 댓글과 댓글 ID를 함께 저장
     
     logging.debug(f'Fetched comments: {comments}')
     return comments
@@ -141,15 +165,15 @@ async def generate_replies(comments, transcript):
     댓글과 자막을 기반으로 LLM 답글을 생성합니다.
     """
     replies = []
-    for comment in comments:
+    for comment, _ in comments:
         messages = [
             {"role": "system", "content": f"비디오 자막: {transcript}"},
             {"role": "user", "content": comment}
         ]
         loop = asyncio.get_event_loop()
         response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(
-            messages, max_tokens=500, temperature=0.7, top_p=0.85))
-        
+            messages, max_tokens=400, temperature=0.7, top_p=0.85))  # max_tokens 값을 조정
+
         if response.choices and response.choices[0].message:
             reply = response.choices[0].message['content'].strip()
         else:
@@ -164,10 +188,29 @@ async def create_thread_and_send_replies(message, video_id, comments, replies):
     댓글과 답글을 새로운 쓰레드에 전송합니다.
     """
     thread = await message.channel.create_thread(name=f"{message.author.name}의 댓글 답글", message=message)
-    for comment, reply in zip(comments, replies):
+    for (comment, _), reply in zip(comments, replies):
         embed = discord.Embed(description=f"**댓글**: {comment}\n**답글**: {reply}")
         await thread.send(embed=embed)
 
+async def post_replies_to_youtube(video_id, comments, replies):
+    """
+    생성된 답글을 YouTube 댓글로 게시합니다.
+    """
+    for (comment, comment_id), reply in zip(comments, replies):
+        try:
+            youtube_service.comments().insert(
+                part='snippet',
+                body={
+                    'snippet': {
+                        'parentId': comment_id,
+                        'textOriginal': reply
+                    }
+                }
+            ).execute()
+            logging.debug(f'Posted reply to comment: {comment_id}')
+        except Exception as e:
+            logging.error(f'Error posting reply to comment {comment_id}: {e}')
+
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(os.getenv('DISCORD_TOKEN'))