Update app.py

app.py

@@ -95,10 +95,10 @@ if st.session_state['upload_count'] < max_attempts:
                 df = pd.DataFrame(entities)
                 st.dataframe(entities)
                 st.dataframe(df)
-                fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
+                fig1 = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
                 values='score', color='label')
-                fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
-                st.plotly_chart(fig, key = "fig")
+                fig1.update_layout(margin = dict(t=50, l=25, r=25, b=25))
+                st.plotly_chart(fig1, key = "fig1")
                 vectorizer = TfidfVectorizer()
                 tfidf_matrix = vectorizer.fit_transform(result)
                 tfidf_df = pd.DataFrame(tfidf_matrix.toarray(), columns=vectorizer.get_feature_names_out())
@@ -161,10 +161,10 @@ if st.session_state['upload_count'] < max_attempts:
                 df = pd.DataFrame(entities)
                 st.dataframe(entities)
                 st.dataframe(df)
-                fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
+                fig2 = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
                 values='score', color='label')
-                fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
-                st.plotly_chart(fig, key = "fig1")
+                fig2.update_layout(margin = dict(t=50, l=25, r=25, b=25))
+                st.plotly_chart(fig2, key = "fig2")
                 vectorizer = TfidfVectorizer()
                 tfidf_matrix = vectorizer.fit_transform(result)
                 tfidf_df = pd.DataFrame(tfidf_matrix.toarray(), columns=vectorizer.get_feature_names_out())