Update app.py

app.py

@@ -88,7 +88,8 @@ if uploaded_file is not None:
                papers.rename(columns={'Publication Year': 'Year', 'Citing Works Count': 'Cited by',
                                      'Publication Type': 'Document Type', 'Source Title': 'Source title'}, inplace=True)
 
-    miss_col = [column for column in columns_to_check if column not in data.columns]
+    col2check = ['Document Type','Source title','Cited by','Year']
+    miss_col = [column for column in col2check if column not in data.columns]
     
     if not missing_columns:
         st.write('✅ Vizualisation')