Update app.py

app.py

@@ -43,12 +43,12 @@ def get_csv_file(csv_docs):
     csv_doc = csv_loader.load()
     return csv_doc
 
-def get_json_file(json_docs):
+def get_json_file(json_docs, jq_schema=None):
     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)
+    json_loader = JSONLoader(temp_filepath, jq_schema)
     json_doc = json_loader.load()
     return json_doc