seawolf2357 commited on
Commit
5d883c6
·
verified ·
1 Parent(s): e1a5542

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -90,11 +90,8 @@ async def generate_response(message):
90
  def search_in_dataset(query, dataset):
91
  response = []
92
  for record in dataset['train']:
93
- if record['사건명'] and query in record['사건명']:
94
- # 사건명으로 검색 사건번호만 반환
95
- response.append(f"사건번호: {record['사건번호']}")
96
- elif record['사건번호'] == query:
97
- # 사건번호로 검색 시 전체 내용 반환
98
  detail = (
99
  f"판례정보일련번호: {record['판례정보일련번호']}\n"
100
  f"사건명: {record['사건명']}\n"
@@ -111,7 +108,16 @@ def search_in_dataset(query, dataset):
111
  f"전문: {record['전문']}\n"
112
  )
113
  response.append(detail)
114
- return "\n".join(response) if response else "관련 법률 정보를 찾을 수 없습니다."
 
 
 
 
 
 
 
 
 
115
 
116
 
117
  if __name__ == "__main__":
 
90
  def search_in_dataset(query, dataset):
91
  response = []
92
  for record in dataset['train']:
93
+ # '사건명' 필드가 None이 아니면 검색 시도
94
+ if record['사건명'] and query.lower() in record['사건명'].lower():
 
 
 
95
  detail = (
96
  f"판례정보일련번호: {record['판례정보일련번호']}\n"
97
  f"사건명: {record['사건명']}\n"
 
108
  f"전문: {record['전문']}\n"
109
  )
110
  response.append(detail)
111
+ break # 번째 일치하는 결과만 반환
112
+
113
+ # 결과가 비어 있지 않은지 확인하고 로그에 출력
114
+ if response:
115
+ logging.debug("Search successful, response generated.")
116
+ return "\n".join(response)
117
+ else:
118
+ logging.debug("No relevant legal information found.")
119
+ return "관련 법률 정보를 찾을 수 없습니다."
120
+
121
 
122
 
123
  if __name__ == "__main__":