Update app.py

app.py

@@ -44,6 +44,7 @@ class MyClient(discord.Client):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.is_processing = False
+        self.session = None
 
     async def on_ready(self):
         logging.info(f'{self.user}로 로그인되었습니다!')
@@ -52,6 +53,9 @@ class MyClient(discord.Client):
         subprocess.Popen(["python", "web.py"])
         logging.info("Web.py 서버가 시작되었습니다.")
 
+        # aiohttp 클라이언트 세션 생성
+        self.session = aiohttp.ClientSession()
+
         # 봇이 시작될 때 안내 메시지를 전송
         channel = self.get_channel(SPECIFIC_CHANNEL_ID)
         if channel:
@@ -85,6 +89,12 @@ class MyClient(discord.Client):
             isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
         )
 
+    async def close(self):
+        # aiohttp 클라이언트 세션 종료
+        if self.session:
+            await self.session.close()
+        await super().close()
+
 def extract_video_id(url):
     video_id = None
     youtube_regex = (
@@ -144,7 +154,7 @@ async def generate_replies(comments, transcript):
         ]
         loop = asyncio.get_event_loop()
         response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(
-            messages, max_tokens=400, temperature=0.7, top_p=0.85))
+            messages, max_tokens=250, temperature=0.7, top_p=0.85))
 
         if response.choices and response.choices[0].message:
             reply = response.choices[0].message['content'].strip()
@@ -155,6 +165,21 @@ async def generate_replies(comments, transcript):
     logging.debug(f'생성된 답글: {replies}')
     return replies
 
+async def send_webhook_data(session, chunk_data, chunk_number):
+    for attempt in range(MAX_RETRIES):
+        try:
+            async with session.post(WEBHOOK_URL, json=chunk_data) as resp:
+                if resp.status == 200:
+                    logging.info(f"웹훅으로 데이터 전송 성공: {chunk_number} 번째 시도")
+                    return True  # 성공 시 종료
+                else:
+                    logging.error(f"웹훅으로 데이터 전송 실패: {resp.status}, {chunk_number} 번째 시도")
+        except aiohttp.ClientError as e:
+            logging.error(f"웹훅 전송 중 오류 발생: {e}, {chunk_number} 번째 시도")
+        await asyncio.sleep(1)  # 재시도 전에 잠시 대기
+
+    return False  # 재시도 횟수 초과 시 실패로 간주
+
 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)
     webhook_data = {"video_id": video_id, "replies": []}
@@ -167,22 +192,14 @@ async def create_thread_and_send_replies(message, video_id, comments, replies):
         webhook_data["replies"].append({"comment": comment, "reply": reply, "comment_id": comment_id})
 
     # 데이터를 여러 번 나누어 전송
-    chunk_size = 5  # 전송할 데이터의 개수
+    chunk_size = 10  # 전송할 데이터의 개수
     for i in range(0, len(webhook_data["replies"]), chunk_size):
         chunk = webhook_data["replies"][i:i+chunk_size]
         chunk_data = {"video_id": video_id, "replies": chunk}
 
-        for attempt in range(MAX_RETRIES):
-            async with aiohttp.ClientSession() as session:
-                try:
-                    async with session.post(WEBHOOK_URL, json=chunk_data) as resp:
-                        if resp.status == 200:
-                            logging.info(f"웹훅으로 데이터 전송 성공: {i // chunk_size + 1} 번째 시도")
-                            break  # 성공 시 루프 탈출
-                        else:
-                            logging.error(f"웹훅으로 데이터 전송 실패: {resp.status}, {i // chunk_size + 1} 번째 시도")
-                except aiohttp.ClientError as e:
-                    logging.error(f"웹훅 전송 중 오류 발생: {e}, {i // chunk_size + 1} 번째 시도")
+        success = await send_webhook_data(self.session, chunk_data, i // chunk_size + 1)
+        if not success:
+            logging.error(f"데이터 전송 실패: {i // chunk_size + 1} 번째 청크")
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)