Added JSON file upload

shivamgutgutia · Sep 2, 2024 · f582af7 · f582af7
1 parent 44dfc70
commit f582af7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/utils/createDf.py b/utils/createDf.py
@@ -1,4 +1,4 @@
-from pandas import read_csv, read_excel
+from pandas import read_csv, read_excel,read_json
 
 def createDf(requestFiles):
     if "file" not in requestFiles or not requestFiles["file"]:
@@ -11,12 +11,14 @@ def createDf(requestFiles):
         df = read_excel(file,dtype=str)
     elif (file.filename.endswith(".ods")):
         df = read_excel(file,engine="odf",dtype=str)
+    elif file.filename.endswith(".json"):
+        df = read_json(file, dtype=str)
     else:
         return False,"Please upload valid file format"
 
     df = df.applymap(lambda x: x.strip() if isinstance(x,str) else "")
     df = df.loc[~(df == "").all(axis=1)]
-        
+
     if not len(df):
         return False,"Empty file uploaded"
     return True,df