format text

graphs.py

@@ -9,9 +9,15 @@ from wordcloud import WordCloud
 from underthesea import word_tokenize
 import matplotlib.pyplot as plt
 
+
 def sentence_topic_plot(result):
     labels = ["Quality",	"Serve",	"Pack",	"Shipping", "Price", "Other"][::-1]
     values = result.detach().numpy()[0][::-1]
+    combined = {labels[i] : values[i] for i in range(len(labels))}
+    sorted_data = dict(sorted(combined.items(), key=lambda item: item[1]))
+    labels = list(sorted_data.keys())
+    values = list(sorted_data.values())
+    filtered_data = {key: value for key, value in combined.items() if value >= 0.5}
     fig = go.Figure()
     fig.add_trace(go.Bar(
         y=labels,
@@ -19,6 +25,8 @@ def sentence_topic_plot(result):
         orientation='h'
     ))
     fig.update_layout(xaxis_title="Probability", yaxis_title="Topics")
+    message = ", ".join(i for i in filtered_data.keys())
+    st.header(f"Your review is related to :blue[{message}]" )
     st.plotly_chart(fig, use_container_width=True)
 
 def KPI_card(name = "Total Reviews", value = 1000, box_color = (123,167,212), font_color = (0, 0, 0), icon = "fa-list"):