File size: 2,289 Bytes
265ec34 5ee2b3a 484c597 5ee2b3a 8a6427f 5ee2b3a 1152ce2 f6451f5 1152ce2 5ee2b3a 484c597 f6451f5 2a8225f f6451f5 484c597 5ee2b3a 484c597 265ec34 5ee2b3a b871d30 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import os
import streamlit as st
import pandas as pd
from pandasai import Agent
from langchain_groq import ChatGroq
llm = ChatGroq(
model="llama3-70b-8192",
temperature=0,
max_retries=2,
)
# Initialize session state for URL and DataFrame
if 'pre_url' not in st.session_state:
st.session_state['pre_url'] = ''
if 'df' not in st.session_state:
st.session_state['df'] = None
# Streamlit app
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.
> :bulb: For example, try asking for: "**Show me a pie chart of the most common sentiment expressed in these comments.**"
""")
# Step 1: Choose conversation
opendata_options = [
'15-per-hour-seattle', 'american-assembly.bowling-green', 'brexit-consensus',
'canadian-electoral-reform', 'football-concussions', 'march-on.operation-marchin-orders',
'scoop-hivemind.affordable-housing', 'scoop-hivemind.biodiversity',
'scoop-hivemind.freshwater', 'scoop-hivemind.taxes', 'scoop-hivemind.ubi',
'ssis.land-bank-farmland.2rumnecbeh.2021-08-01', 'vtaiwan.uberx'
]
selected_option = st.selectbox("Choose conversation", opendata_options)
url = (
f"https://raw.githubusercontent.com/compdemocracy/openData/master/{selected_option}/comments.csv"
)
# Load data only if URL changes
if st.session_state['pre_url'] != url:
agent = Agent(
pd.read_csv(url, index_col=0),
config={"llm": llm}
)
st.session_state['agent'] = agent
st.session_state['pre_url'] = url
# Step 2: Request for analysis or chart
request = st.text_input(
"Enter your analysis request:",
"Plot a histogram about the distribution of agree, disagree and neutral comments against the topic")
# Execute chat and display results
if st.button("Analyze"):
if st.session_state['agent'] is not None:
file = st.session_state['agent'].chat(request)
if os.path.exists(file):
st.image(file)
else:
st.info(file)
else:
st.warning("Please select a conversation and load data first.") |