lakxs commited on
Commit
b30ada3
β€’
1 Parent(s): 5b1ff51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -7
app.py CHANGED
@@ -259,14 +259,20 @@ def get_json_file(json_docs):
259
  temp_dir = tempfile.TemporaryDirectory() # μž„μ‹œ 디렉토리λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
260
  temp_filepath = os.path.join(temp_dir.name, json_docs.name) # μž„μ‹œ 파일 경둜λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
261
 
262
- with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ ν…μŠ€νŠΈ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
263
  f.write(json_docs.getvalue()) # JSON λ¬Έμ„œμ˜ λ‚΄μš©μ„ μž„μ‹œ νŒŒμΌμ— μ”λ‹ˆλ‹€.
264
- # json_loader = JSONLoader(temp_filepath)
265
- json_loader = JSONLoader(file_path=temp_filepath,
266
- jq_schema='.messages[].content',
267
- text_content=False)
268
- json_doc = json_loader.load() # ν…μŠ€νŠΈλ₯Ό μΆ”μΆœν•©λ‹ˆλ‹€.
269
- return json_doc # μΆ”μΆœλœ ν…μŠ€νŠΈλ₯Ό λ°˜ν™˜ν•©λ‹ˆλ‹€.
 
 
 
 
 
 
270
 
271
 
272
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.
 
259
  temp_dir = tempfile.TemporaryDirectory() # μž„μ‹œ 디렉토리λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
260
  temp_filepath = os.path.join(temp_dir.name, json_docs.name) # μž„μ‹œ 파일 경둜λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
261
 
262
+ with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
263
  f.write(json_docs.getvalue()) # JSON λ¬Έμ„œμ˜ λ‚΄μš©μ„ μž„μ‹œ νŒŒμΌμ— μ”λ‹ˆλ‹€.
264
+
265
+ json_loader = JSONLoader(file_path=temp_filepath, jq_schema='.messages[].content', text_content=False)
266
+
267
+ try:
268
+ json_doc = json_loader.load() # JSON λ¬Έμ„œλ‘œλΆ€ν„° ν…μŠ€νŠΈλ₯Ό μΆ”μΆœν•©λ‹ˆλ‹€.
269
+ except Exception as e:
270
+ # μ˜ˆμ™Έλ‚˜ null 데이터λ₯Ό gracefulν•˜κ²Œ μ²˜λ¦¬ν•©λ‹ˆλ‹€.
271
+ print(f"JSON λ‘œλ“œ 쀑 였λ₯˜ λ°œμƒ: {e}")
272
+ json_doc = [] # 였λ₯˜ λ°œμƒ μ‹œ κΈ°λ³Έκ°’μœΌλ‘œ 빈 리슀트λ₯Ό μ„€μ •ν•©λ‹ˆλ‹€.
273
+
274
+ return json_doc # μΆ”μΆœλœ ν…μŠ€νŠΈλ₯Ό λ°˜ν™˜ν•©λ‹ˆλ‹€. (였λ₯˜ λ°œμƒ μ‹œ 빈 리슀트λ₯Ό λ°˜ν™˜ν•©λ‹ˆλ‹€.)
275
+
276
 
277
 
278
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.