edit codes

app.py

@@ -30,24 +30,10 @@ def get_pdf_text(pdf_docs):
 # 과제
 # 아래 텍스트 추출 함수를 작성
 
-# def get_text_file(docs):
-#     temp_dir2 = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-#     temp_filepath2 = os.path.join(temp_dir2.name, docs.name)  # 임시 파일 경로를 생성합니다.
-#     with open(temp_filepath2, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-#         f.write(docs.getvalue())  # TXT 문서의 내용을 임시 파일에 씁니다.
-#     text_loader = TextLoader(
-#         file_path=temp_filepath2,
-#         txt_schema='',
-#         text_content=False
-#     )  # Use your specific text loader here.
-#     text_data = text_loader.load()  # Extract text using the loader.
-#     return text_data  # 추출한 텍스트를 반환합니다.
 def get_text_file(docs):
     temp_dir2 = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
     temp_filepath2 = os.path.join(temp_dir2.name, docs.name)  # 임시 파일 경로를 생성합니다.
-    txt_loader = TextLoader(
-        file_path=temp_filepath2
-    )
+    txt_loader = TextLoader(temp_filepath2)
     txt_data = txt_loader.load()
     return txt_data
 
@@ -56,18 +42,15 @@ def get_csv_file(docs):
     temp_filepath3 = os.path.join(temp_dir3.name, docs.name)
     with open(temp_filepath3, "wb") as f:
         f.write(docs.getvalue())
-
-    # csv_data = []
-    # with open(temp_filepath3, "r") as csv_file:
-    #     csv_reader = csv.reader(csv_file)
-    #     for row in csv_reader:
-    #         csv_data.append(row)
-    # csv_loader = CSVLoader(
-    #     file_path=temp_filepath3,
-    #     cs_schema='.data[].address',
-    #     text_content=False
-    # )
-    # csv_data = csv_loader.load()
+    csv_loader = CSVLoader(
+        file_path=temp_filepath3,
+        csv_args={
+            "delimiter": ",",
+            "quotechar": '"',
+            "fieldnames": ["name", "school", "address", "phone"],
+        },
+    )
+    csv_data = csv_loader.load()
     return csv_data
 
 def get_json_file(docs):