Problem with displaying SmartDataframes in Streamlit

[notmanas]

I have been using pandasai version 0.8.1 by calling PandasAI API. But that will be deprecated soon, so I switched to SmartDataframe.
Earlier if I got a dataframe back from PandasAI after processing, I could just use -

st.dataframe(df, use_container_width=True)

to show that dataframe. Now, it doesn't work that way anymore. If I use st.write(result_df), I can see it as a table -

BUT if I use
st.dataframe(result_df)

I don't see the column headers anymore? -

How do I tackle this?

Thanks in advance!

[fabmeyer]

Having the same problem, please help!

[fabmeyer]

A quick fix:

smart_dataframe = SmartDataframe(dataframe, config={'llm': llm})
answer = smart_dataframe.chat(user_input)
column_names = answer.columns
answer = pd.DataFrame(data=answer, columns=column_names)

[gventuri]

@notmanas @fabmeyer this shouldn't happen. The logic is pretty much the same.
Maybe one attempt could be:

smart_dataframe = SmartDataframe(dataframe, config={'llm': llm})
df = smart_dataframe.chat(user_input)
answer = df.dataframe # this will return the original pandas dataframe instead of a smart dataframe

Let me know if it works

[fabmeyer]

Same what I did:
#506 (reply in thread)