namkwonwoo commited on
Commit
d633ef3
ยท
1 Parent(s): 03abd5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -49,11 +49,17 @@ def get_json_file(json_docs):
49
  temp_filepath = os.path.join(temp_dir.name, json_docs.name)
50
  with open(temp_filepath, "wb") as f:
51
  f.write(json_docs.getvalue())
52
- json_loader = JSONLoader(file_path = temp_filepath,
53
- jq_schema = '.messages[].content',
54
- text_content=False)
55
- json_doc = json_loader.load()
56
- return json_doc
 
 
 
 
 
 
57
 
58
 
59
  # ๋ฌธ์„œ๋“ค์„ ์ฒ˜๋ฆฌํ•˜์—ฌ ํ…์ŠคํŠธ ์ฒญํฌ๋กœ ๋‚˜๋ˆ„๋Š” ํ•จ์ˆ˜์ž…๋‹ˆ๋‹ค.
 
49
  temp_filepath = os.path.join(temp_dir.name, json_docs.name)
50
  with open(temp_filepath, "wb") as f:
51
  f.write(json_docs.getvalue())
52
+ jq_schema = '.messages[].content' # ์ˆ˜์ •๋œ JQ Query
53
+
54
+ # JQ Query๊ฐ€ None์ด ์•„๋‹Œ์ง€ ํ™•์ธ
55
+ if jq_schema is not None:
56
+ json_loader = JSONLoader(file_path=temp_filepath, jq_schema=jq_schema, text_content=False)
57
+ json_doc = json_loader.load()
58
+ return json_doc
59
+ else:
60
+ # jq_schema๊ฐ€ None์ธ ๊ฒฝ์šฐ ์ฒ˜๋ฆฌ
61
+ print("jq_schema is None")
62
+ return None # ๋˜๋Š” ์ ์ ˆํ•œ ๊ฐ’์„ ๋ฐ˜ํ™˜ํ•˜์—ฌ ํ•จ์ˆ˜๋ฅผ ์ข…๋ฃŒํ•ฉ๋‹ˆ๋‹ค.
63
 
64
 
65
  # ๋ฌธ์„œ๋“ค์„ ์ฒ˜๋ฆฌํ•˜์—ฌ ํ…์ŠคํŠธ ์ฒญํฌ๋กœ ๋‚˜๋ˆ„๋Š” ํ•จ์ˆ˜์ž…๋‹ˆ๋‹ค.