Spaces:
Runtime error
Runtime error
ra2w
commited on
Commit
·
86508a8
1
Parent(s):
f8dccd1
Fixed a small bug
Browse files
app.py
CHANGED
|
@@ -427,7 +427,7 @@ def nlu_query_use_api():
|
|
| 427 |
'Which company trades at the highest revenue multiple?',
|
| 428 |
'List all companies with growth rates greater than 40?',
|
| 429 |
'What is the average gross margin for companies in this dataset?',
|
| 430 |
-
'What is the average trading multiple
|
| 431 |
|
| 432 |
]
|
| 433 |
|
|
@@ -441,16 +441,18 @@ def nlu_query_use_api():
|
|
| 441 |
if 'error' in output:
|
| 442 |
st.write(output['error'])
|
| 443 |
else:
|
| 444 |
-
|
| 445 |
-
|
| 446 |
-
|
| 447 |
-
|
| 448 |
-
|
| 449 |
-
|
| 450 |
-
|
| 451 |
-
|
| 452 |
-
|
| 453 |
-
|
|
|
|
|
|
|
| 454 |
with st.expander("Raw Output"):
|
| 455 |
st.write(output)
|
| 456 |
return
|
|
|
|
| 427 |
'Which company trades at the highest revenue multiple?',
|
| 428 |
'List all companies with growth rates greater than 40?',
|
| 429 |
'What is the average gross margin for companies in this dataset?',
|
| 430 |
+
'What is the average trading multiple?'
|
| 431 |
|
| 432 |
]
|
| 433 |
|
|
|
|
| 441 |
if 'error' in output:
|
| 442 |
st.write(output['error'])
|
| 443 |
else:
|
| 444 |
+
try:
|
| 445 |
+
df_output = pd.DataFrame.from_dict(output['cells'])
|
| 446 |
+
if output['aggregator'] == 'COUNT':
|
| 447 |
+
st.info(f"[COUNT] Answer: {df_output[0].count()}")
|
| 448 |
+
elif output['aggregator'] == 'SUM':
|
| 449 |
+
st.info(f"[SUM] Answer: {df_output[0].astype(float).sum().round(2)}")
|
| 450 |
+
elif output['aggregator'] == 'AVERAGE':
|
| 451 |
+
st.info(f"[AVERAGE] Answer: {df_output[0].astype(float).mean().round(2)}")
|
| 452 |
+
else:
|
| 453 |
+
st.info(f"Answer is {output['answer']}")
|
| 454 |
+
except ValueError:
|
| 455 |
+
st.write(output)
|
| 456 |
with st.expander("Raw Output"):
|
| 457 |
st.write(output)
|
| 458 |
return
|