Update app.py

app.py

@@ -71,7 +71,11 @@ async def generate_response(message):
 def search_in_dataset(query, dataset):
     response = []
     for record in dataset['train']:
-        if query in record['사건명']:  # 부분 일치 검사
+        if record['사건명'] and query in record['사건명']:
+            # 사건명으로 검색 시 사건번호만 반환
+            response.append(f"사건번호: {record['사건번호']}")
+        elif record['사건번호'] == query:
+            # 사건번호로 검색 시 전체 내용 반환
             detail = (
                 f"판례정보일련번호: {record['판례정보일련번호']}\n"
                 f"사건명: {record['사건명']}\n"
@@ -90,6 +94,7 @@ def search_in_dataset(query, dataset):
             response.append(detail)
     return "\n".join(response) if response else "관련 법률 정보를 찾을 수 없습니다."
 
+
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(os.getenv('DISCORD_TOKEN'))