Update app.py

app.py

@@ -4,6 +4,7 @@ import nltk
 import json
 import base64
 import pandas as pd
+import re
 
 os.environ["OPENAI_API_KEY"] = os.getenv("SECRET_KEY")
 
@@ -57,8 +58,9 @@ if uploaded_file:
     output = chain.run(input_documents=docs, question=query)
     st.subheader("Parsing result in JSON format")
     st.json(output)
-
-    json_data = json.loads(json.dumps(output))
+    
+    valid_json = re.sub( "(?<={)\'|\'(?=})|(?<=\[)\'|\'(?=\])|\'(?=:)|(?<=: )\'|\'(?=,)|(?<=, )\'", "\"", output)
+    json_data = json.loads(json.dumps(valid_json))
 
     names = [json_data.get("full_name", "N/A")]
     contacts = [json_data.get("contacts", "N/A")]