hyonee commited on
Commit
03d0f67
โ€ข
1 Parent(s): 5747adb
Files changed (1) hide show
  1. app.py +7 -17
app.py CHANGED
@@ -41,24 +41,14 @@ def get_text_file(text_docs):
41
 
42
 
43
  def get_csv_file(csv_docs):
44
- temp_dir = tempfile.TemporaryDirectory()
45
- temp_filepath = os.path.join(temp_dir.name, csv_docs.name)
46
-
47
- try:
48
- with open(temp_filepath, "wb") as f:
49
- f.write(csv_docs.getvalue())
50
-
51
- csv_loader = CSVLoader(temp_filepath)
52
- csv_content = csv_loader.load()
53
-
54
- except Exception as e:
55
- print(f"Error loading CSV file: {e}")
56
- csv_content = None
57
-
58
- finally:
59
- temp_dir.cleanup()
60
 
61
- return csv_content
62
 
63
  def get_json_file(json_docs):
64
  with tempfile.TemporaryDirectory() as temp_dir:
 
41
 
42
 
43
  def get_csv_file(csv_docs):
44
+ temp_dir = tempfile.TemporaryDirectory() # ์ž„์‹œ ๋””๋ ‰ํ† ๋ฆฌ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
45
+ temp_filepath = os.path.join(temp_dir.name, csv_docs.name) # ์ž„์‹œ ํŒŒ์ผ ๊ฒฝ๋กœ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
46
+ with open(temp_filepath, "wb") as f: # ์ž„์‹œ ํŒŒ์ผ์„ ๋ฐ”์ด๋„ˆ๋ฆฌ ์“ฐ๊ธฐ ๋ชจ๋“œ๋กœ ์—ฝ๋‹ˆ๋‹ค.
47
+ f.write(csv_docs.getvalue()) # PDF ๋ฌธ์„œ์˜ ๋‚ด์šฉ์„ ์ž„์‹œ ํŒŒ์ผ์— ์”๋‹ˆ๋‹ค.
48
+ csv_loader = CSVLoader(temp_filepath) # PyPDFLoader๋ฅผ ์‚ฌ์šฉํ•ด PDF๋ฅผ ๋กœ๋“œํ•ฉ๋‹ˆ๋‹ค.
49
+ csv_doc = csv_loader.load() # ํ…์ŠคํŠธ๋ฅผ ์ถ”์ถœํ•ฉ๋‹ˆ๋‹ค.
50
+ return csv_doc # ์ถ”์ถœํ•œ ํ…์ŠคํŠธ๋ฅผ ๋ฐ˜ํ™˜ํ•ฉ๋‹ˆ๋‹ค.
 
 
 
 
 
 
 
 
 
51
 
 
52
 
53
  def get_json_file(json_docs):
54
  with tempfile.TemporaryDirectory() as temp_dir: