Update app.py

app.py

@@ -94,14 +94,11 @@ async def generate_response(message):
     logging.debug(f'Full model response sent: {full_response_text}')
     conversation_history.append({"role": "assistant", "content": full_response_text})
 
-
 def search_in_dataset(query, dataset):
-    # 사용자의 쿼리와 관련된 모든 필드를 검색하고 상세 정보를 반환합니다.
     response = []
     for record in dataset['train']:
-        # 사건명 필드에서 사용자의 쿼리와 관련된 정보를 찾습니다.
-        if query in record['사건명']:
-            # 정보가 발견되면, 모든 필드의 상세한 정보를 포맷팅하여 response 리스트에 추가합니다.
+        # '사건명' 필드가 None이 아니고, 쿼리가 '사건명' 필드 내에 포함되어 있는 경우만 처리
+        if record['사건명'] and query in record['사건명']:
             detail = (
                 f"판례정보일련번호: {record['판례정보일련번호']}\n"
                 f"사건명: {record['사건명']}\n"
@@ -118,8 +115,10 @@ def search_in_dataset(query, dataset):
                 f"전문: {record['전문']}\n"
             )
             response.append(detail)
-    
-    # response 리스트에 담긴 정보들을 반환합니다.
+        elif not record['사건명']:
+            # '사건명' 필드가 None인 경우 로깅
+            logging.debug(f"Skipped a record with None in '사건명': {record}")
+
     return "\n".join(response) if response else "관련 법률 정보를 찾을 수 없습니다."