Update pages/2_Earnings_Summarization_📖_.py

pages/2_Earnings_Summarization_📖_.py

@@ -17,18 +17,34 @@ if "earnings_passages" not in st.session_state:
 
 if st.session_state['earnings_passages']:
       
-      with st.spinner("Summarizing and matching entities, this takes a few seconds..."):
-            text_to_summarize = chunk_and_preprocess_text(st.session_state['earnings_passages'])
+with st.spinner("Summarizing and matching entities, this takes a few seconds..."):
+    
+    try:
+        text_to_summarize = chunk_and_preprocess_text(st.session_state['earnings_passages'])
+        summarized_text = summarize_text(text_to_summarize,max_len=max_len,min_len=min_len)
+        
+    
+    except IndexError:
+        try:
+            
+            text_to_summarize = chunk_and_preprocess_text(st.session_state['earnings_passages'],450)
             summarized_text = summarize_text(text_to_summarize,max_len=max_len,min_len=min_len)
-            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)
+
+        except IndexError:
             
-            st.markdown("####")     
-        
-            summary_downloader(summarized_text)
+            text_to_summarize = chunk_and_preprocess_text(st.session_state['earnings_passages'],400)
+            summarized_text = summarize_text(text_to_summarize,max_len=max_len,min_len=min_len)
+                    
+    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")