Update app.py

app.py

@@ -29,14 +29,40 @@ def get_pdf_text(pdf_docs):
 # 아래 텍스트 추출 함수를 작성
 
 def get_text_file(docs):
-    pass
-
+    # 텍스트 파일의 경우, 내용을 바로 읽어옵니다.
+    text_content = docs.getvalue().decode("utf-8")  # UTF-8 인코딩을 기준으로 디코딩합니다.
+    return [text_content]
 
 def get_csv_file(docs):
-    pass
+    # CSV 파일의 경우, 각 행 또는 열에서 텍스트를 추출합니다.
+    import pandas as pd
+    csv_content = docs.getvalue().decode("utf-8")  # 바이트를 문자열로 디코딩합니다.
+    csv_data = pd.read_csv(pd.compat.StringIO(csv_content))  # Pandas를 사용하여 CSV를 읽어옵니다.
+    text_list = []
+
+    # 필요한 대로 각 열 또는 행에서 텍스트를 추출합니다.
+    for column in csv_data.columns:
+        text_list.extend(csv_data[column].astype(str).tolist())
+
+    return text_list
 
 def get_json_file(docs):
-    pass
+    # JSON 파일의 경우, 특정 키 또는 값에서 텍스트를 추출합니다.
+    import json
+    json_content = docs.getvalue().decode("utf-8")  # 바이트를 문자열로 디코딩합니다.
+    json_data = json.loads(json_content)
+
+    # 필요한 대로 JSON 키 또는 값에서 텍스트를 추출합니다.
+    text_list = []
+    for key, value in json_data.items():
+        if isinstance(value, str):
+            text_list.append(value)
+        elif isinstance(value, list):
+            text_list.extend(value)
+        elif isinstance(value, dict):
+            text_list.extend(value.values())
+
+    return text_list
 
     
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.