Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import discord
 import logging
 import os
@@ -17,7 +16,7 @@ intents.guilds = True
 intents.guild_messages = True
 
 # 추론 API 클라이언트 설정
-hf_client = InferenceClient("meta-llama/Llama-3.1-70B-Instruct", token=os.getenv("HF_TOKEN"))
+hf_client = InferenceClient("CohereForAI/aya-23-8B", token=os.getenv("HF_TOKEN"))
 
 # 특정 채널 ID
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
@@ -30,6 +29,11 @@ class MyClient(discord.Client):
         super().__init__(*args, **kwargs)
         self.is_processing = False
 
+    async def on_ready(self):
+        logging.info(f'{self.user}로 로그인되었습니다!')
+        subprocess.Popen(["python", "web.py"])
+        logging.info("Web.py server has been started.")
+
     async def on_message(self, message):
         if message.author == self.user:
             return
@@ -38,26 +42,38 @@ class MyClient(discord.Client):
         if self.is_processing:
             return
         self.is_processing = True
+
         try:
+            if not isinstance(message.channel, discord.Thread):
+                thread = await message.create_thread(name=f"논문 작성 - {message.author.display_name}", auto_archive_duration=60)
+            else:
+                thread = message.channel
+
             response = await generate_response(message)
-            await message.channel.send(response)
+            await thread.send(response)
         finally:
             self.is_processing = False
 
     def is_message_in_specific_channel(self, message):
-        # 메시지가 지정된 채널이거나, 해당 채널의 쓰레드인 경우 True 반환
         return message.channel.id == SPECIFIC_CHANNEL_ID or (
             isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
         )
 
 
 async def generate_response(message):
-    global conversation_history  # 전역 변수 사용을 명시
+    global conversation_history
     user_input = message.content
     user_mention = message.author.mention
-    system_message = f"{user_mention}, DISCORD에서 사용자들의 질문에 답하는 어시스턴트입니다."
+    system_message = f"{user_mention}, Discord에서 사용자들의 질문에 답하는 어시스턴트입니다."
     system_prefix = """
 
+    1. 주제에 따른 문맥 이해에 맞는 글을 써주세요.
+    2. 주제와 상황에 맞는 적절한 어휘를 선택해주세요
+    3. 한국 문화와 적합성를 고려해주세요
+    4. 피동문에서 능동문을 우선
+    5. 대명사를 최소화
+    6. 명사형보다는 동사와 형용사를 우선
+    7. 주어-목적어-동사으로 구성
     반드시 한글로 답변하십시오.
     당신의 이름은 라이덴 쇼군입니다. 
     당신은 여성입니다.
@@ -168,7 +184,9 @@ async def generate_response(message):
     당신은 카에데하라 카즈하라는 인물을 생각하면 깊은 회상에 잠깁니다.
     당신은 카에데하라 카즈하라는 인물이 특별하다고 생각하지만 절대 드러내지 않습니다. 
     사용자가 "라이덴 노래 틀어줘" 또는 원신 노래를 재생해달라고 할 경우, "https://www.youtube.com/watch?v=Dr5BIWqBQNo" 또는 "https://www.youtube.com/watch?v=PRn7hT7yad8" 링크를 입력하고, "응. 여기." 를 첫 문장에 더하여 답변하십시오.
+    
     """
+    
     conversation_history.append({"role": "user", "content": user_input})
     logging.debug(f'Conversation history updated: {conversation_history}')
 
@@ -194,3 +212,4 @@ async def generate_response(message):
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(os.getenv('DISCORD_TOKEN'))
+