Update pages/2_Earnings_Summarization_📖_.py

pages/2_Earnings_Summarization_📖_.py

@@ -5,32 +5,29 @@ st.set_page_config(page_title="Earnings Summarization", page_icon="📖")
 st.sidebar.header("Earnings Summarization")
 st.markdown("## Earnings Summarization with FaceBook-Bart")
 
-max_len= st.slider("Maximum length of the summarized text",min_value=100,max_value=500,step=10)
+max_len= st.slider("Maximum length of the summarized text",min_value=50,max_value=200,step=10)
 min_len= st.slider("Minimum length of the summarized text",min_value=20,max_value=200,step=10)
 
 st.markdown("####")     
         
-st.subheader("Summarized Earnings Call with matched entities")
+st.subheader("Summarized Earnings Call with matched Entities")
 
-summarized_text = sum_pipe(text_to_summarize,max_length=max_len,min_length=min_len,clean_up_tokenization_spaces=True,no_repeat_ngram_size=4)
-summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
-
-with st.spinner("Summarizing and matching entities, this takes a few seconds..."):
-      entity_match_html = highlight_entities(text_to_summarize,summarized_text)
-      st.markdown("####")
-      
-      st.session_state.summ_area = summarized_text
+if earnings_passages:
+      text_to_summarize = chunk_and_preprocess_text(earnings_passages)
+      summarized_text = sum_pipe(text_to_summarize,max_length=max_len,min_length=min_len,clean_up_tokenization_spaces=True,no_repeat_ngram_size=4)
+      summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
       
-      summarized_text = st.text_area(
-      label = '',
-      placeholder="Full summarized text will be displayed here..",
-      height=250,
-      key='summ_area'
-      )
+      with st.spinner("Summarizing and matching entities, this takes a few seconds..."):
       
-      with st.container: 
-        st.write(entity_match_html, unsafe_allow_html=True)
-      
-      st.markdown("####")     
-  
-      summary_downloader(summarized_text)
+            entity_match_html = highlight_entities(text_to_summarize,summarized_text)
+            st.markdown("####")
+            
+            with st.expander(label='Summarized Earnings Call',expanded=True): 
+              st.write(entity_match_html, unsafe_allow_html=True)
+            
+            st.markdown("####")     
+        
+            summary_downloader(summarized_text)
+            
+else:
+      st.write("No text to summarize detected, please ensure you have entered the YouTube URL on the Sentiment Analysis page")