Update app.py

app.py

@@ -230,40 +230,37 @@ with st.form(key='columns_in_form'):
             st.write("Your digest is ready:\n")
             st.write(digestor.text)
             st.write(f"""Text approximately {len(digestor.text.split(" ") )} words.""")
-            st.write(f"""Number of articles summarized: {len(outdata['summaries'].keys())}""")
+            st.write(f"""Number of articles summarized: {outdata['article_count']}""")
 
+            
+            st.success(f"""Digest completed in {digestor.timer.timers['digest_time']} seconds.""")
+    
+            st.write("Here's some stats about the summarization:\n")
+    
+            # Summarize the findings for all models
+            labels = [i for i in range(outdata['article_count'])]
+            original_length = [i.original_length for i in outdata['summaries']]
+            summarized_length = [i.summary_length for i in outdata['summaries']]       
+            x = np.arange(len(labels))  # the label locations
+            width = 0.35  # the width of the bars
+            
+            fig, ax = plt.subplots(figsize=(14,8))
+            rects1 = ax.bar(x - width/2, original_length, width,  color='lightgreen',zorder=0)
+            rects2 = ax.bar(x + width/2, summarized_length, width, color='lightblue',zorder=0)
+            
+            rects3 = ax.bar(x - width/2, original_length, width, color='none',edgecolor='black', hatch='XX', lw=1.25,zorder=1)
+            rects4 = ax.bar(x + width/2, summarized_length, width, color='none',edgecolor='black', hatch='xx', lw=1.25,zorder=1)
+            
+            # Add some text for labels, title and custom x-axis tick labels, etc.
+            ax.set_ylabel('Original Length')
+            ax.set_xticks(x)
+            ax.set_yticks([i for i in range(0,2500,250)])
+            ax.set_xticklabels(labels)
+            ax.set_xlabel('Summarized Length')
+            
+            plt.title('Original to Summarized Lengths in Space-Separated Tokens')
+            #ax.hist(arr, bins=20)
+            st.pyplot(fig)
         
-        st.success(f"""Digest completed in {digestor.timer.timers['digest_time']} seconds.""")
-        
-"st.session_state object:", st.session_state
-
-
-
-# Summarize the findings for all models
-labels = [1,2,3,4,5,6,7,8,9,10,11]
-training_scores = [1000, 2200, 875, 234, 300, 344, 419, 705, 400, 500, 230]
-test_scores = [388, 41, 100, 400, 344, 123, 600, 433, 133, 56, 89]
-
-x = np.arange(len(labels))  # the label locations
-width = 0.35  # the width of the bars
-
-fig, ax = plt.subplots(figsize=(14,8))
-rects1 = ax.bar(x - width/2, training_scores, width,  color='lightgreen',zorder=0)
-rects2 = ax.bar(x + width/2, test_scores, width, color='lightblue',zorder=0)
-
-rects3 = ax.bar(x - width/2, training_scores, width, color='none',edgecolor='black', hatch='XX', lw=1.25,zorder=1)
-rects4 = ax.bar(x + width/2, test_scores, width, color='none',edgecolor='black', hatch='xx', lw=1.25,zorder=1)
-
-# Add some text for labels, title and custom x-axis tick labels, etc.
-ax.set_ylabel('Original Length')
-ax.set_xticks(x)
-ax.set_yticks([i for i in range(0,2500,250)])
-ax.set_xticklabels(labels)
-ax.set_xlabel('Summarized Length')
-
-plt.title('Original to Summarized Lengths in Space-Separated Tokens')
-#ax.hist(arr, bins=20)
-st.pyplot(fig)
-
-#tc = pd.DataFrame({'train':training_scores,'test':test_scores})
-#st.bar_chart(tc)
+      
+"st.session_state object:", st.session_state