untilize streamlit support of pandasai

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 from pandasai import SmartDataframe
+from pandasai.responses.streamlit_response import StreamlitResponse
 from langchain_groq import ChatGroq
 
 llm = ChatGroq(
@@ -20,9 +21,10 @@ st.title("Conversation Analysis")
 
 # Introduction and Explanation
 st.markdown("""
-**Go beyond static visualizations!** This app lets you directly ask for the insights you need from Polis conversation data. 
-Instead of being limited to a standard PCA chart, you can request custom plots and analyses using natural language. 
-For example, try asking for: "**Show me a pie chart of the most common sentiment expressed in these comments.**"
+## Go beyond static visualizations!
+This app lets you directly ask for the insights you need from Polis conversation data.  
+Instead of being limited to a standard PCA chart, you can request custom plots and analyses using natural language.  
+> :bulb: For example, try asking for: "**Show me a pie chart of the most common sentiment expressed in these comments.**"
 """)
 
 # Step 1: Choose conversation
@@ -41,7 +43,13 @@ url =  (
 
 # Load data only if URL changes
 if st.session_state['pre_url'] != url:
-    df = SmartDataframe(pd.read_csv(url, index_col=0), config={"llm": llm})
+    df = SmartDataframe(
+        pd.read_csv(url, index_col=0),
+        config={
+            "llm": llm,
+            "response_parser": StreamlitResponse
+        }
+    )
     st.session_state['df'] = df
     st.session_state['pre_url'] = url