Update app.py

app.py

@@ -46,13 +46,14 @@ 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):
     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,
-                            text_content=False) 
+                             jq_schema,
+                             text_content=False) 
     json_doc = json_loader.load() 
     return json_doc 
     
@@ -151,7 +152,7 @@ def main():
                         doc_list.extend(get_csv_file(file))
                     elif file.type == 'application/json':
                         # file is .json
-                        doc_list.extend(get_json_file(file))
+                        doc_list.extend(get_json_file(file, jq_schema))
 
                 # get the text chunks
                 text_chunks = get_text_chunks(doc_list)