Update app.py

app.py

@@ -7,6 +7,10 @@ from extractive_summarization import summarize_with_textrank, summarize_with_lsa
 from abstractive_summarization import summarize_with_bart_cnn, summarize_with_bart_ft, summarize_with_led, summarize_with_t5
 from keyword_extraction import extract_keywords
 from keyphrase_extraction import extract_sentences_with_obligations
+from wordcloud import WordCloud
+import matplotlib.pyplot as plt
+from PIL import Image
+import io
 #from blanc import BlancHelp
 
 # Load in ToS
@@ -119,6 +123,16 @@ def main():
         if 'summary' in st.session_state:
             st.write(st.session_state.summary)
 
+            # Generate and display word cloud
+            wordcloud = WordCloud(width=800, height=400, background_color='white').generate(st.session_state.summary)
+            # Convert to PIL Image
+            image = wordcloud.to_image()
+            # Convert PIL Image to bytes
+            buf = io.BytesIO()
+            image.save(buf, format='PNG')
+            byte_im = buf.getvalue()
+            st.image(byte_im, caption='Word Cloud of Summary', use_column_width=True)
+
             # Check if no PDF or text input is provided and a ToS document is selected
             if not uploaded_file and not user_input and tos_selection_index is not None and 'summary' in dataset['train'][tos_selection_index]:
                 # Fetch the reference summary