Update app.py

app.py

@@ -1,16 +1,15 @@
 import discord
 import logging
 import os
+import re  # re 모듈 import 추가
 import asyncio
+import subprocess
 import aiohttp
 from huggingface_hub import InferenceClient
 from googleapiclient.discovery import build
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api.formatters import TextFormatter
 from dotenv import load_dotenv
-import subprocess
-import re  # re 모듈 import 추가
-
 
 # 환경 변수 로드
 load_dotenv()
@@ -26,7 +25,7 @@ intents.guilds = True
 intents.guild_messages = True
 
 # 추론 API 클라이언트 설정
-hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
+hf_client = InferenceClient(token=os.getenv("HF_TOKEN"))
 
 # YouTube API 설정
 API_KEY = os.getenv("YOUTUBE_API_KEY")
@@ -35,14 +34,15 @@ youtube_service = build('youtube', 'v3', developerKey=API_KEY)
 # 특정 채널 ID
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
 
+# 웹훅 URL 설정
+WEBHOOK_URL = "https://connect.pabbly.com/workflow/sendwebhookdata/IjU3NjUwNTY1MDYzMjA0MzA1MjY4NTUzMDUxMzUi_pc"
+
 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}로 로그인되었습니다!')
 
@@ -61,6 +61,9 @@ class MyClient(discord.Client):
     async def on_message(self, message):
         if message.author == self.user or not self.is_message_in_specific_channel(message):
             return
+        if self.is_processing:
+            await message.channel.send("현재 다른 요청을 처리 중입니다. 잠시 후 다시 시도해 주세요.")
+            return
         self.is_processing = True
         try:
             video_id = self.extract_video_id(message.content)
@@ -110,7 +113,11 @@ class MyClient(discord.Client):
                 {"role": "user", "content": comment},
                 {"role": "system", "content": f"비디오 자막: {transcript}"}
             ]
-            response = await hf_client.chat_completion(messages, max_tokens=300, temperature=0.7, top_p=0.85)
+            
+            # 동기식 메서드로 사용
+            loop = asyncio.get_event_loop()
+            response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(messages, max_tokens=300, temperature=0.7, top_p=0.85))
+            
             if response.choices and response.choices[0].message:
                 return response.choices[0].message['content'].strip()
             else:
@@ -127,8 +134,22 @@ class MyClient(discord.Client):
             return None
 
     async def send_webhook_data(self, comment, reply, comment_id):
-        # 웹훅 데이터 준비 및 전송 로직 구현
-        pass
+        webhook_data = {
+            "video_id": video_id,
+            "replies": [{"comment": comment, "reply": reply, "comment_id": comment_id}]
+        }
+        for attempt in range(3):
+            try:
+                async with self.session.post(WEBHOOK_URL, json=webhook_data) as resp:
+                    if resp.status == 200:
+                        logging.info("웹훅 데이터 전송 성공")
+                        return True
+                    else:
+                        logging.error(f"웹훅 데이터 전송 실패: {resp.status}")
+            except aiohttp.ClientError as e:
+                logging.error(f"웹훅 전송 중 오류 발생: {e}")
+            await asyncio.sleep(1)
+        return False
 
     async def close(self):
         if self.session:
@@ -138,3 +159,5 @@ class MyClient(discord.Client):
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(os.getenv('DISCORD_TOKEN'))
+
+