Update app.py

app.py

@@ -56,14 +56,14 @@ if uploaded_file is not None:
     elif extype.endswith('.txt'):
         data = conv_txt(extype)
 
-    #edited_df = st.data_editor(data)
-    
-    columns_to_check = ['Authors', 'Title', 'Source title', 'Author Keywords', 'Keywords Plus']
+    #===check keywords===
+    @st.cache_data(ttl=3600)
+    def get_data(extype):
+        list_of_column_key = list(keywords.columns)
+        list_of_column_key = [k for k in list_of_column_key if 'Keyword' in k]
+        return list_of_column_key
 
-    for column in columns_to_check:
-        if column in data.columns:
-            st.write(f"The column '{column}' exists in the DataFrame.")
-        else:
-            st.write(f"The column '{column}' does not exist in the DataFrame.")
-
-    st.write(data)
+    if not list_of_column_key:
+        st.error('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:
+        st.write('✅')