Update app.py

app.py

@@ -37,6 +37,8 @@ for message in st.session_state.messages:
         if "sql_query" in message:
             with st.expander("SQL Query", expanded=False):
                 st.code(message["sql_query"], language="sql")
+        if "dataframe" in message:
+            st.dataframe(message["dataframe"])
 
 input_text = st.chat_input("Chat with your bot here...")
 
@@ -68,4 +70,5 @@ if input_text:
         "role": "assistant",
         "reasoning": reasoning,
         "sql_query": sql_query,
+        "dataframe": df,
     })