Update app.py

app.py

@@ -61,6 +61,7 @@ class MyClient(discord.Client):
         if not self.is_message_in_specific_channel(message):
             return
         if self.is_processing:
+            logging.debug("Currently processing another message, skipping this one.")
             return
 
         self.is_processing = True
@@ -71,41 +72,35 @@ class MyClient(discord.Client):
             await message.channel.send(response)
         finally:
             self.is_processing = False
+            logging.debug("Message processing completed, ready for the next one.")
 
     def is_message_in_specific_channel(self, message):
-        return message.channel.id == SPECIFIC_CHANNEL_ID or (
-            isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
-        )
+        channel_condition = message.channel.id == SPECIFIC_CHANNEL_ID
+        thread_condition = isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
+        return channel_condition or thread_condition
 
 
 
 async def generate_response(message):
     global conversation_history
-    user_input = message.content
+    case_number = message.content.strip()
     user_mention = message.author.mention
-    system_message = f"{user_mention}, DISCORD에서 사용자들의 질문에 답하는 어시스턴트입니다."
-    
+    system_message = f"{user_mention}, 다음은 요청하신 사건의 전문입니다:"
+
     # 데이터 검색 및 응답 준비
-    answer = search_in_dataset(user_input, law_dataset)
-    if not answer:
-        answer = "관련 법률 정보를 찾을 수 없습니다."
-    
-    full_response_text = system_message + "\n\n" + answer
+    full_text = search_in_dataset_by_case_number(case_number, law_dataset)
+    full_response_text = f"{system_message}\n\n{full_text}"
     max_length = 2000
     if len(full_response_text) > max_length:
-        response_parts = []
         for i in range(0, len(full_response_text), max_length):
             part_response = full_response_text[i:i+max_length]
             await message.channel.send(part_response)
-            response_parts.append(part_response)
         logging.debug(f'Full model response sent: {full_response_text}')
         conversation_history.append({"role": "assistant", "content": full_response_text})
-        return "".join(response_parts)  # 조각난 메시지들을 연결하여 반환
     else:
         await message.channel.send(full_response_text)
         logging.debug(f'Full model response sent: {full_response_text}')
         conversation_history.append({"role": "assistant", "content": full_response_text})
-        return full_response_text  # 전체 메시지를 반환
 
 
 def search_in_dataset(query, dataset):
@@ -115,6 +110,12 @@ def search_in_dataset(query, dataset):
             return record['사건번호']
     return "관련 법률 정보를 찾을 수 없습니다."
 
+def search_in_dataset_by_case_number(case_number, dataset):
+    for record in dataset['train']:
+        if record['사건번호'] == case_number:
+            return record.get('전문', "해당 사건에 대한 전문 정보가 없습니다.")
+    return "관련 법률 정보를 찾을 수 없습니다."
+
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)