Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 from keyphrasetransformer import KeyPhraseTransformer
+from wordcloud import WordCloud
+import matplotlib.pyplot as plt
 
 kp = KeyPhraseTransformer()
 
@@ -15,6 +17,14 @@ def calculate_similarity(model, text1, text2):
     embedding2 = model.encode([text2])
     return cosine_similarity(embedding1, embedding2)[0][0]
 
+def generate_wordcloud(text, title):
+    wordcloud = WordCloud(width=800, height=400, background_color='white').generate(text)
+    plt.figure(figsize=(10, 5))
+    plt.imshow(wordcloud, interpolation='bilinear')
+    plt.axis('off')
+    plt.title(title)
+    st.pyplot(plt)
+
 st.title("Resume Match Calculator")
 
 model = load_model()