Update app.py

app.py

@@ -57,43 +57,54 @@ if uploaded_file is not None:
         data = conv_txt(extype)
 
     #===check keywords===
-    @st.cache_data(ttl=3600)
-    def get_data(extype):
-        keycheck = list(data.columns)
-        keycheck = [k for k in keycheck if 'Keyword' in k]
-        return keycheck
-
-    keycheck = get_data(extype)
-
+    st.divider()    
+    keycheck = list(data.columns)
+    keycheck = [k for k in keycheck if 'Keyword' in k]
+    
     container1 = st.container(border=True)
+    container1.subheader('Keyword Stem', divider='brown')
     if not keycheck:
-        container1.write("Unfortunately, you don't have a column containing keywords in your data. Please check again. If you want to use it in another column, please rename it to 'Keywords'.")
+        container1.write("❌ Unfortunately, you don't have a column containing keywords in your data. Please check again. If you want to use it in another column, please rename it to 'Keywords'.")
     else:
-        container1.write('✅ Keywords Stem')
-
+        container1.write('✅ Congratulations! You can use Keywords Stem')
+        
     #===check any obj===
+    st.divider()
     coldf = sorted(data.select_dtypes(include=['object']).columns.tolist())
+
+    container2 = st.container(border=True)
+    container2.subheader('Topic Modeling', divider='brown')
     
     if not coldf:
-        rslt_obj = "Unfortunately, you don't have a column containing object in your data. Please check again."
+        container2.write("❌ Unfortunately, you don't have a column containing object in your data. Please check again.")
     else:
-        rslt_obj = '✅ Topic Modeling'
+        container2.write('✅ Congratulations! You can use Topic Modeling')
 
+    #===bidirected===
+    st.divider()
+    
+    container3 = st.container(border=True)
+    container3.subheader('Bidirected Network', divider='brown')
+    
     if not keycheck:
-        rslt_keycheck2 = "Unfortunately, you don't have a column containing keywords in your data. Please check again. If you want to use it in another column, please rename it to 'Keywords'."
+        container3.write("❌ Unfortunately, you don't have a column containing keywords in your data. Please check again. If you want to use it in another column, please rename it to 'Keywords'.")
     else:
-        rslt_keycheck2 = '✅ Bidirected Network'
+        container3.write('✅ Congratulations! You can use Bidirected Network')
 
     #===Visualization===
+    st.divider()
     if 'Publication Year' in data.columns:
                papers.rename(columns={'Publication Year': 'Year', 'Citing Works Count': 'Cited by',
                                      'Publication Type': 'Document Type', 'Source Title': 'Source title'}, inplace=True)
 
     col2check = ['Document Type','Source title','Cited by','Year']
     miss_col = [column for column in col2check if column not in data.columns]
+
+    container4 = st.container(border=True)
+    container4.subheader('Sunburst', divider='brown')
     
     if not miss_col:
-        rslt_viz = '✅ Vizualisation'
+        container4.write('✅ Congratulations! You can use Sunburst')
     else:
         miss_col_str = ', '.join(miss_col)
-        rslt_viz = f"Unfortunately, you don't have: {miss_col_str}. Please check again."
+        container4.write(f"❌ Unfortunately, you don't have: {miss_col_str}. Please check again.")