Spaces:
Runtime error
Runtime error
ddovidovich
commited on
Commit
•
de0702c
1
Parent(s):
6f756fc
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import nltk
|
|
4 |
import json
|
5 |
import base64
|
6 |
import pandas as pd
|
7 |
-
import
|
8 |
|
9 |
os.environ["OPENAI_API_KEY"] = os.getenv("SECRET_KEY")
|
10 |
|
@@ -57,9 +57,9 @@ if uploaded_file:
|
|
57 |
chain = load_qa_chain(ChatOpenAI(temperature=0), chain_type="stuff")
|
58 |
output = chain.run(input_documents=docs, question=query)
|
59 |
st.subheader("Parsing result in JSON format")
|
60 |
-
|
|
|
61 |
|
62 |
-
valid_json = re.sub( "(?<={)\'|\'(?=})|(?<=\[)\'|\'(?=\])|\'(?=:)|(?<=: )\'|\'(?=,)|(?<=, )\'", "\"", output)
|
63 |
json_data = json.loads(json.dumps(valid_json))
|
64 |
|
65 |
names = [json_data.get("full_name", "N/A")]
|
|
|
4 |
import json
|
5 |
import base64
|
6 |
import pandas as pd
|
7 |
+
import ast
|
8 |
|
9 |
os.environ["OPENAI_API_KEY"] = os.getenv("SECRET_KEY")
|
10 |
|
|
|
57 |
chain = load_qa_chain(ChatOpenAI(temperature=0), chain_type="stuff")
|
58 |
output = chain.run(input_documents=docs, question=query)
|
59 |
st.subheader("Parsing result in JSON format")
|
60 |
+
valid_json = ast.literal_eval(output)
|
61 |
+
st.json(valid_json)
|
62 |
|
|
|
63 |
json_data = json.loads(json.dumps(valid_json))
|
64 |
|
65 |
names = [json_data.get("full_name", "N/A")]
|