Update app.py

app.py

@@ -9,6 +9,7 @@ import numpy as np
 import re
 import urllib.request
 import random
+import plotly.express as px
 
 st.set_page_config(
     page_title="FATA4 Science",
@@ -51,9 +52,9 @@ if opt == "Neuroblastoma corpus":
     num_abstracts = 29032
     database_name = "Neuroblastoma"
 
-st.header(":red[Fast Acting Text Analysis (FATA) 4 Science]")
+st.header(":red[*F*]ast :red[*A*]cting :red[*T*]ext :red[*A*]nalysis (:red[*FATA*]) 4 Science")
 
-st.subheader(":red[Uncovering knowledge through Natural Language Processing (NLP)]")
+st.subheader("Uncovering knowledge through Natural Language Processing (NLP)")
 st.markdown("---")
 
 st.header(f":blue[{database_name} Pubmed corpus.]")
@@ -110,9 +111,11 @@ if query:
 
     short_table.set_index('Word', inplace=True)
     squarify.plot(sizes=sizes, label=short_table.index.tolist(), color=color, edgecolor="#EBF5FB",
-                  text_kwargs={'fontsize': 10})
+                  text_kwargs={'fontsize': 10},)
     # # plot the treemap using matplotlib
     plt.axis('off')
+    # Add legend to top right, outside plot region
+    # plt.legend("upper right", bbox_to_anchor=(-.2, 0))
     fig = plt.gcf()
     fig.patch.set_facecolor('#CCFFFF')
     # # display the treemap in Streamlit
@@ -158,7 +161,6 @@ if query:
     plt.axis('off')
     fig2 = plt.gcf()
     fig2.patch.set_facecolor('#CCFFFF')
-    # plt.show()
     #
     # # display the treemap in Streamlit
     with treemap2:
@@ -205,3 +207,12 @@ if query:
         with c3:
            st.video("https://www.youtube.com/watch?v=" + video_ids[2])
     st.markdown("---")
+
+
+
+
+
+
+
+
+