Update app.py

app.py

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