Update app.py

app.py

@@ -67,7 +67,7 @@ class MyClient(discord.Client):
         self.is_processing = True
         try:
             response = await generate_response(message)
-            if response.strip() == "":
+            if response and response.strip() == "":
                 response = "죄송합니다, 제공할 수 있는 정보가 없습니다."
             await message.channel.send(response)
         finally:
@@ -79,8 +79,6 @@ class MyClient(discord.Client):
         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
     case_number = message.content.strip()
@@ -92,20 +90,24 @@ async def generate_response(message):
     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  # 전체 메시지를 반환
 
+    return ""  # 반환 값이 없을 경우 빈 문자열 반환
 
 def search_in_dataset(query, dataset):
     for record in dataset['train']:
-        # '사건명'이 None이 아닌지 확인하고, query가 해당 값에 포함되어 있는지 검사
         if record['사건명'] and query in record['사건명']:
             return record['사건번호']
     return "관련 법률 정보를 찾을 수 없습니다."
@@ -116,7 +118,6 @@ def search_in_dataset_by_case_number(case_number, dataset):
             return record.get('전문', "해당 사건에 대한 전문 정보가 없습니다.")
     return "관련 법률 정보를 찾을 수 없습니다."
 
-
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(os.getenv('DISCORD_TOKEN'))