Update app.py

app.py

@@ -64,11 +64,19 @@ for uploaded_file in uploaded_files:
         tfidf_df = pd.DataFrame(tfidf_matrix.toarray(), columns=vectorizer.get_feature_names_out())
         st.subheader("TF-IDF Values:")
         st.dataframe(tfidf_df)
+        
         cosine_sim_matrix = cosine_similarity(tfidf_matrix)
         cosine_sim_df = pd.DataFrame(cosine_sim_matrix)
         st.subheader("Cosine Similarity Matrix:")
         st.dataframe(cosine_sim_df)
 
+        import plotly.express as px
+        
+        fig = px.imshow(cosine_sim_df, text_auto=True, labels=dict(x="Cosine similarity", y="Text", color="Productivity"),
+                        x=['text1', 'text2'],
+                        y=['text1', 'text2'])
+        st.plotly_chart(fig)
+
         st.subheader("Cosine Similarity Scores (Job Description vs. Resumes):")
         for i, similarity_score in enumerate(cosine_sim_matrix[0][1:]):
             st.write(f"Similarity with Candidate Profile {i + 1}: {similarity_score:.4f}")