Update app.py

app.py

@@ -69,18 +69,23 @@ def create_pdf(report_text):
     pdf.output(output_path)
     return output_path
 
+from PIL import Image
+
 def generate_wordcloud(missing_skills):
+    if not missing_skills:
+        return None
     text = " ".join(missing_skills)
     wordcloud = WordCloud(width=600, height=400, background_color='black', colormap='Set3').generate(text)
     buffer = io.BytesIO()
-    plt.figure(figsize=(6,4))
+    plt.figure(figsize=(6, 4))
     plt.imshow(wordcloud, interpolation='bilinear')
     plt.axis('off')
     plt.tight_layout()
     plt.savefig(buffer, format='png')
     plt.close()
     buffer.seek(0)
-    return buffer
+    return Image.open(buffer)
+
 
 def process_skill_gap(resume_pdf, jd_pdfs):
     if resume_pdf is None or jd_pdfs is None: