Update app.py

app.py

@@ -124,7 +124,7 @@ def main():
             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)
+            wordcloud = WordCloud(width=800, height=400, background_color='white', max_words=20).generate(st.session_state.summary)
             # Convert to PIL Image
             image = wordcloud.to_image()
             # Convert PIL Image to bytes
@@ -142,12 +142,19 @@ def main():
                 rouge = Rouge()
                 scores = rouge.get_scores(st.session_state.summary, reference_summary)
     
-                # Display ROUGE scores in a box
-                with st.container():
-                    st.write("ROUGE Scores:")
-                    st.write(f"ROUGE-1: {scores[0]['rouge-1']['f']:.4f}")
-                    st.write(f"ROUGE-2: {scores[0]['rouge-2']['f']:.4f}")
-                    st.write(f"ROUGE-L: {scores[0]['rouge-l']['f']:.4f}")
+
+                # Using st.markdown with HTML and inline CSS to create a box around the content
+                st.markdown(
+                    """
+                    <div style="border: 2px solid #f0f2f6; border-radius: 5px; padding: 10px;">
+                        <h4 style="color: #2678bf;">ROUGE Scores:</h4>
+                        <p>ROUGE-1: {rouge_1:.4f}</p>
+                        <p>ROUGE-2: {rouge_2:.4f}</p>
+                        <p>ROUGE-L: {rouge_l:.4f}</p>
+                    </div>
+                    """.format(rouge_1=scores[0]['rouge-1']['f'], rouge_2=scores[0]['rouge-2']['f'], rouge_l=scores[0]['rouge-l']['f']),
+                    unsafe_allow_html=True
+                )
 
 if __name__ == "__main__":
     main()