Update app.py

app.py

@@ -52,27 +52,25 @@ def get_csv_file(csv_docs):
 
     
 # json 문서로부터 텍스트를 추출하는 함수
-
-def extract_text_from_json_file(file_path):
-    try:
-        with open(file_path, 'r') as file:
-            data = json.load(file)
-
-            # 'users' 키의 값을 추출
-            users_data = data.get('users', [])
-
-            # 'users' 배열의 각 요소에서 'firstName' 키의 값을 추출하여 리스트에 저장
-            first_names = [user.get('firstName', '') for user in users_data]
-
-            # 추출한 값 리스트를 문자열로 변환하여 반환
-            text = ', '.join(first_names)
-            return text
-
-    except FileNotFoundError:
-        return f"File not found: {file_path}"
-    except json.JSONDecodeError as e:
-        return f"Invalid JSON format: {e}"
-
+class JSONLoader:
+    def __init__(self, file_path, jq_schema):
+        self.file_path = file_path
+        self.jq_schema = jq_schema
+
+    def load(self):
+        with open(self.file_path, 'r') as json_file:
+            data = json.load(json_file)
+            users_text = [user.get('text', '') for user in data.get('users', [])]
+            return users_text
+
+def get_json_file(json_docs):
+    temp_dir = tempfile.TemporaryDirectory()
+    temp_filepath = os.path.join(temp_dir.name, json_docs.name)
+    with open(temp_filepath, "wb") as f:
+        f.write(json_docs.getvalue())
+    json_loader = JSONLoader(temp_filepath, jq_schema={})
+    users_text = json_loader.load()
+    return users_text