seawolf2357 commited on
Commit
7ac41e0
ยท
verified ยท
1 Parent(s): 65f94cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -12
app.py CHANGED
@@ -95,12 +95,13 @@ async def generate_response(message):
95
  conversation_history.append({"role": "assistant", "content": full_response_text})
96
 
97
 
98
-
99
  def search_in_dataset(query, dataset):
 
100
  response = []
101
  for record in dataset['train']:
102
- # '์‚ฌ๊ฑด๋ช…' ํ•„๋“œ๊ฐ€ None์ด ์•„๋‹ˆ๋ฉด ๊ฒ€์ƒ‰ ์‹œ๋„
103
- if record['์‚ฌ๊ฑด๋ช…'] and query.lower() in record['์‚ฌ๊ฑด๋ช…'].lower():
 
104
  detail = (
105
  f"ํŒ๋ก€์ •๋ณด์ผ๋ จ๋ฒˆํ˜ธ: {record['ํŒ๋ก€์ •๋ณด์ผ๋ จ๋ฒˆํ˜ธ']}\n"
106
  f"์‚ฌ๊ฑด๋ช…: {record['์‚ฌ๊ฑด๋ช…']}\n"
@@ -117,16 +118,9 @@ def search_in_dataset(query, dataset):
117
  f"์ „๋ฌธ: {record['์ „๋ฌธ']}\n"
118
  )
119
  response.append(detail)
120
- break # ์ฒซ ๋ฒˆ์งธ ์ผ์น˜ํ•˜๋Š” ๊ฒฐ๊ณผ๋งŒ ๋ฐ˜ํ™˜
121
 
122
- # ๊ฒฐ๊ณผ๊ฐ€ ๋น„์–ด ์žˆ์ง€ ์•Š์€์ง€ ํ™•์ธํ•˜๊ณ  ๋กœ๊ทธ์— ์ถœ๋ ฅ
123
- if response:
124
- logging.debug("Search successful, response generated.")
125
- return "\n".join(response)
126
- else:
127
- logging.debug("No relevant legal information found.")
128
- return "๊ด€๋ จ ๋ฒ•๋ฅ  ์ •๋ณด๋ฅผ ์ฐพ์„ ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค."
129
-
130
 
131
 
132
  if __name__ == "__main__":
 
95
  conversation_history.append({"role": "assistant", "content": full_response_text})
96
 
97
 
 
98
  def search_in_dataset(query, dataset):
99
+ # ์‚ฌ์šฉ์ž์˜ ์ฟผ๋ฆฌ์™€ ๊ด€๋ จ๋œ ๋ชจ๋“  ํ•„๋“œ๋ฅผ ๊ฒ€์ƒ‰ํ•˜๊ณ  ์ƒ์„ธ ์ •๋ณด๋ฅผ ๋ฐ˜ํ™˜ํ•ฉ๋‹ˆ๋‹ค.
100
  response = []
101
  for record in dataset['train']:
102
+ # ์‚ฌ๊ฑด๋ช… ํ•„๋“œ์—์„œ ์‚ฌ์šฉ์ž์˜ ์ฟผ๋ฆฌ์™€ ๊ด€๋ จ๋œ ์ •๋ณด๋ฅผ ์ฐพ์Šต๋‹ˆ๋‹ค.
103
+ if query in record['์‚ฌ๊ฑด๋ช…']:
104
+ # ์ •๋ณด๊ฐ€ ๋ฐœ๊ฒฌ๋˜๋ฉด, ๋ชจ๋“  ํ•„๋“œ์˜ ์ƒ์„ธํ•œ ์ •๋ณด๋ฅผ ํฌ๋งทํŒ…ํ•˜์—ฌ response ๋ฆฌ์ŠคํŠธ์— ์ถ”๊ฐ€ํ•ฉ๋‹ˆ๋‹ค.
105
  detail = (
106
  f"ํŒ๋ก€์ •๋ณด์ผ๋ จ๋ฒˆํ˜ธ: {record['ํŒ๋ก€์ •๋ณด์ผ๋ จ๋ฒˆํ˜ธ']}\n"
107
  f"์‚ฌ๊ฑด๋ช…: {record['์‚ฌ๊ฑด๋ช…']}\n"
 
118
  f"์ „๋ฌธ: {record['์ „๋ฌธ']}\n"
119
  )
120
  response.append(detail)
 
121
 
122
+ # response ๋ฆฌ์ŠคํŠธ์— ๋‹ด๊ธด ์ •๋ณด๋“ค์„ ๋ฐ˜ํ™˜ํ•ฉ๋‹ˆ๋‹ค.
123
+ return "\n".join(response) if response else "๊ด€๋ จ ๋ฒ•๋ฅ  ์ •๋ณด๋ฅผ ์ฐพ์„ ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค."
 
 
 
 
 
 
124
 
125
 
126
  if __name__ == "__main__":