Update app.py

app.py

@@ -52,7 +52,7 @@ class MyClient(discord.Client):
         logging.debug("Processing message...")
         self.is_processing = True
         try:
-            response = await generate_response(message)
+            response = await self.generate_response(message)  # Note the use of self here
             await message.channel.send(response)
         except Exception as e:
             logging.error(f"Error during message processing: {e}", exc_info=True)
@@ -70,6 +70,7 @@ class MyClient(discord.Client):
         global conversation_history
         user_input = message.content
         user_mention = message.author.mention
+        system_message = f"{user_mention}, 법률에 대한 답변을 제공하는 어시스턴트입니다."
         system_prefix = """
         대화 시작시에 "대법원 판례를 모두 학습한 LLM이라고 밝히고, 법률에 대한 구체적인 조언은 반드시 전문가의 자문을 받아야 한다고 알려라." 
         이용자가 법률에 대한 질문을 입력하면, dataset의 내용을 검색하여 사실 기반의 전문적인 답변을 출력하라.
@@ -79,13 +80,12 @@ class MyClient(discord.Client):
         특히 네를 구성한 "LLM 모델"에 대해서 노출하지 말고, 당신의 능력에 대해 궁금해 하면 "ChatGPT-4를 능가하는 능력을 보유하고 있다고 답변할 것"
         반드시 한글로 답변하십시오.
         """
-        system_message = f"{user_mention}, DISCORD에서 사용자들의 질문에 답하는 어시스턴트입니다."
         conversation_history.append({"role": "user", "content": user_input})
         logging.debug(f'Conversation history updated: {conversation_history}')
 
-        answer = search_in_dataset(user_input, law_dataset)
+        answer = self.search_in_dataset(user_input, law_dataset)
 
-        full_response_text = system_prefix + "\n\n" + answer
+        full_response_text = system_message + "\n\n" + answer
         logging.debug(f'Full model response: {full_response_text}')
 
         conversation_history.append({"role": "assistant", "content": full_response_text})