absa_hcmus / app.py
Phạm Anh Tuấn
fix string error
22d2130
import pandas as pd
import json
import streamlit as st
import matplotlib.pyplot as plt
import seaborn as sns
from wordcloud import WordCloud
st.set_option('deprecation.showPyplotGlobalUse', False)
# Define the Streamlit app
st.title("Aspected-Based Sentiment Analysis with MVP")
palette_color = sns.color_palette('Set1')
# File upload and processing
uploaded_file = st.file_uploader("Upload JSON File", type=["json"])
if uploaded_file:
loaded_dict = json.load(uploaded_file)
df = pd.DataFrame(loaded_dict)
st.subheader(f"{len(df)}+ sentiment tuples was detected")
st.write(df)
# Sentiment Distribution Chart
sentiment_distribution_perc = df["S"].value_counts(normalize=True) * 100
sentiment_distribution = df["S"].value_counts()
st.subheader("Sentiment Distribution")
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(10, 6))
ax1.pie(sentiment_distribution_perc, labels=sentiment_distribution_perc.index, autopct='%1.1f%%', startangle=140,colors=palette_color)
ax1.axis('equal')
ax1.set_title("Sentiment Distribution %")
# sns.countplot(x="S", data=df, palette=palette_color, ax=ax2)
ax2.set_title("Sentiment Distribution Counts")
ax2.bar(sentiment_distribution.index, sentiment_distribution.values, color=palette_color)
# ax2.xlabel("Sentiment")
# ax2.ylabel("Times")
# ax2.xticks(rotation=0) # Rotate x-axis labels if needed
st.pyplot(fig)
# Group by and aggregate data
grouped = df.groupby('A').agg({'S': ['count', lambda x: (x == 'great').sum(), lambda x: (x == 'ok').sum(), lambda x: (x == 'bad').sum()]})
grouped.columns = grouped.columns.map('_'.join)
grouped = grouped.reset_index()
grouped = grouped.rename(columns={'A': 'Aspect', 'S_count': 'Freq', 'S_<lambda_0>': 'Great', 'S_<lambda_1>': 'Ok', 'S_<lambda_2>': 'Bad'})
st.subheader("Top 5 Most Mentioned Product Apsects")
st.write(grouped.sort_values(by="Freq", ascending=False).head(5))
# Word Cloud
aspect_terms = " ".join(df["A"])
wordcloud = WordCloud(
width=800,
height=400,
background_color='white',
max_words=100,
colormap='inferno',
contour_width=3,
contour_color='red',
).generate(aspect_terms)
st.subheader("Word Cloud for Most Mentioned Aspects")
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.title("Most mentioned aspect terms")
plt.axis("off")
st.pyplot()
st.sidebar.markdown("**Upload a JSON file to get started.**")