Update app.py

app.py

@@ -151,6 +151,31 @@ def ner_results(ner_object, groups=True, NER_THRESHOLD=0.5) -> List[str]:
     ner_list = [i for i in set(people) if len(i) > 2] + [i for i in set(places) if len(i) > 2] + [i for i in set(orgs) if len(i) > 2] + [i for i in set(misc) if len(i) > 2]
 
     return ner_list
+    
+def show_length_graph():
+    labels = [i for i in range(outdata['article_count'])]
+    original_length = [outdata['summaries'][i]['original_length'] for i in outdata['summaries']]
+    summarized_length = [outdata['summaries'][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('Text Length')
+    ax.set_xticks(x)
+    ax.set_yticks([i for i in range(0,max(original_length),max(summarized_length))])
+    ax.set_xticklabels(labels)
+    ax.set_xlabel('Article')
+    
+    plt.title('Original to Summarized Lengths in Space-Separated Tokens')
+    #ax.hist(arr, bins=20)
+    st.pyplot(fig)
 
 # These could be passed through the command line
 # or read from a config file.
@@ -249,29 +274,7 @@ with st.form(key='columns_in_form'):
             st.write("Here are some stats about the summarization:\n")
     
             # Summarize the findings for all models
-            labels = [i for i in range(outdata['article_count'])]
-            original_length = [outdata['summaries'][i]['original_length'] for i in outdata['summaries']]
-            summarized_length = [outdata['summaries'][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('Text Length')
-            ax.set_xticks(x)
-            ax.set_yticks([i for i in range(0,max(original_length),max(summarized_length))])
-            ax.set_xticklabels(labels)
-            ax.set_xlabel('Article')
-            
-            plt.title('Original to Summarized Lengths in Space-Separated Tokens')
-            #ax.hist(arr, bins=20)
-            st.pyplot(fig)
+            show_length_graph()
         
       
 "st.session_state object:", st.session_state