Upload app.py

app.py

@@ -59,14 +59,23 @@ with tab2:
         st.divider()
         st.header('Output')
         resumeClf = pd.read_excel(uploadedResumeClf)
-        resumeClf = classifyResumes(resumeClf)
-        with st.expander('View Bar Chart'):
-            barChart = createBarChart(resumeClf)
-            st.altair_chart(barChart, use_container_width = True)
-        currentClf = filterDataframeClf(resumeClf)
-        st.dataframe(currentClf, use_container_width = True, hide_index = True)
-        xlsxClf = convertDfToXlsx(currentClf)
-        st.download_button(label='Save Current Output as XLSX', data = xlsxClf, file_name = 'Resumes_categorized.xlsx')
+        if 'Resume' in resumeClf.columns:
+            resumeClf = classifyResumes(resumeClf)
+            with st.expander('View Bar Chart'):
+                barChart = createBarChart(resumeClf)
+                st.altair_chart(barChart, use_container_width = True)
+            currentClf = filterDataframeClf(resumeClf)
+            st.dataframe(currentClf, use_container_width = True, hide_index = True)
+            xlsxClf = convertDfToXlsx(currentClf)
+            st.download_button(label='Save Current Output as XLSX', data = xlsxClf, file_name = 'Resumes_categorized.xlsx')
+        else:
+            st.error("""
+            #### Oops! Something went wrong.
+
+            The **"Resume"** column is not found in the uploaded excel file.
+            
+            Kindly make sure the column is present :)
+            """)
 
 with tab3:
     st.header('Input')
@@ -89,11 +98,20 @@ with tab3:
         st.header('Output')
         jobDescriptionRnk = uploadedJobDescriptionRnk.read().decode('utf-8')
         resumeRnk = pd.read_excel(uploadedResumeRnk)
-        resumeRnk = rankResumes(jobDescriptionRnk, resumeRnk)
-        with st.expander('View Job Description'):
-            st.write(jobDescriptionRnk)
-        currentRnk = filterDataframeRnk(resumeRnk)
-        st.dataframe(currentRnk, use_container_width = True, hide_index = True)
-        xlsxRnk = convertDfToXlsx(currentRnk)
-        st.download_button(label='Save Current Output as XLSX', data = xlsxRnk, file_name = 'Resumes_ranked.xlsx')
+        if 'Resume' in resumeRnk.columns:
+            resumeRnk = rankResumes(jobDescriptionRnk, resumeRnk)
+            with st.expander('View Job Description'):
+                st.write(jobDescriptionRnk)
+            currentRnk = filterDataframeRnk(resumeRnk)
+            st.dataframe(currentRnk, use_container_width = True, hide_index = True)
+            xlsxRnk = convertDfToXlsx(currentRnk)
+            st.download_button(label='Save Current Output as XLSX', data = xlsxRnk, file_name = 'Resumes_ranked.xlsx')
+        else:
+            st.error("""
+            #### Oops! Something went wrong.
+
+            The **"Resume"** column is not found in the uploaded excel file.
+            
+            Kindly make sure the column is present :)
+            """)