Update app.py

app.py

@@ -45,25 +45,35 @@ def get_text_file(docs):
 
 
 def get_csv_file(docs):
-    import pandas as pd
-    text = ''
+    # import pandas as pd
+    # text = ''
+
+    # data = pd.read_csv(docs)
+
+    # for index, row in data.iterrows():
+    #     item_name = row[0]
+    #     row_text = item_name
+    #     for col_name in data.columns[1:]:
+    #         row_text += '{} is {} '.format(col_name, row[col_name])
+    #     text += row_text + '\n'
+
+    # return text
 
-    data = pd.read_csv(docs)
+    with NamedTemporaryFile() as temp_file:
+        temp_file.write(docs.getvalue())
+        temp_file.seek(0)
+        text_loader = CSVLoader(temp_file.name)
+        text_doc = text_loader.load()
 
-    for index, row in data.iterrows():
-        item_name = row[0]
-        row_text = item_name
-        for col_name in data.columns[1:]:
-            row_text += '{} is {} '.format(col_name, row[col_name])
-        text += row_text + '\n'
+        return text_doc
 
-    return text
 
 def get_json_file(docs):
     with NamedTemporaryFile() as temp_file:
         temp_file.write(docs.getvalue())
         temp_file.seek(0)
-        json_loader = JSONLoader(temp_file.name)
+        json_loader = JSONLoader(temp_file.name,
+                                jq_schema = '.messages[].content')
         json_doc = json_loader.load()
 
         return json_doc