Fix the extract summary display

Summarizer.py

@@ -27,20 +27,22 @@ class Summarizer:
             summarized_list.append(sentence._text)
         return summarized_list
 
-    def __extractive_summary(self, parser: DocumentParser, sentences_count):
+    def __extractive_summary(self, parser: DocumentParser, sentences_count) -> list:
         summarized_sentences = self.lsa_summarizer(parser.document, sentences_count)
         summarized_list = Summarizer.sentence_list(summarized_sentences)
-        all_sentences_list = Summarizer.sentence_list(parser.document.sentences)
-        return all_sentences_list, summarized_list
+        return summarized_list
 
-    def extractive_summary_from_text(self, text: str, sentences_count: int) -> (list, list):
+    def extractive_summary_from_text(self, text: str, sentences_count: int) -> list:
         parser = PlaintextParser.from_string(text, Tokenizer(Summarizer.DEFAULT_LANGUAGE))
         return self.__extractive_summary(parser, sentences_count)
 
-    def extractive_summary_from_url(self, url: str, sentences_count: int) -> (list, list):
+    def extractive_summary_from_url(self, url: str, sentences_count: int) -> list:
         parser = HtmlParser.from_url(url, Tokenizer(Summarizer.DEFAULT_LANGUAGE))
         return self.__extractive_summary(parser, sentences_count)
 
-    def abstractive_summary(self, summary: str) -> str:
-        summary_text = " ".join([result['summary_text'] for result in self.pipeline(wrap(summary, 2048))])
-        return summary_text
+    def abstractive_summary(self, extract_summary_sentences: list) -> list:
+        extract_summary = " ".join([sentence for sentence in extract_summary_sentences])
+        abstractive_summary_list = []
+        for result in self.pipeline(wrap(extract_summary, 512)):
+            abstractive_summary_list.append(result['summary_text'])
+        return abstractive_summary_list

app.py

@@ -33,19 +33,20 @@ def main() -> None:
             )
         return terms_and_conditions_pipeline
 
-    def display_abstractive_summary(summary) -> None:
+    def display_abstractive_summary(summary_sentences: list) -> None:
         st.subheader("Abstractive Summary")
         st.markdown('#####')
-        st.markdown(summary)
+        for sentence in summary_sentences:
+            st.markdown(f"- {sentence}", unsafe_allow_html=True)
 
-    def display_extractive_summary(terms_and_conditions_sentences: list, summary_sentences: list) -> None:
+    def display_extractive_summary(terms_and_conditions_text: list, summary_sentences: list) -> None:
         st.subheader("Extractive Summary")
         st.markdown('#####')
-        terms_and_conditions = " ".join(sentence for sentence in terms_and_conditions_sentences)
-        replaced_text = terms_and_conditions
+        replaced_text = terms_and_conditions_text
         for sentence in summary_sentences:
             replaced_text = replaced_text.replace(sentence,
                                                   f"<span style='background-color: #FFFF00'>{sentence}</span>")
+        replaced_text = replaced_text.replace('/\r?\n/g', '<br/>')
         st.write(replaced_text, unsafe_allow_html=True)
 
     def is_valid_url(url: str) -> bool:
@@ -82,17 +83,14 @@ def main() -> None:
     if submit_button:
 
         if is_valid_url(tc_text_input):
-            (all_sentences, extract_summary_sentences) = summarizer.extractive_summary_from_url(tc_text_input,
-                                                                                                sentences_length_input)
+            extract_summary_sentences = summarizer.extractive_summary_from_url(tc_text_input, sentences_length_input)
         else:
-            (all_sentences, extract_summary_sentences) = summarizer.extractive_summary_from_text(tc_text_input,
-                                                                                                 sentences_length_input)
+            extract_summary_sentences = summarizer.extractive_summary_from_text(tc_text_input, sentences_length_input)
 
-        extract_summary = " ".join([sentence for sentence in extract_summary_sentences])
-        abstract_summary = summarizer.abstractive_summary(extract_summary)
+        abstract_summary_list = summarizer.abstractive_summary(extract_summary_sentences)
 
-        display_extractive_summary(all_sentences, extract_summary_sentences)
-        display_abstractive_summary(abstract_summary)
+        display_abstractive_summary(abstract_summary_list)
+        display_extractive_summary(tc_text_input, extract_summary_sentences)
 
 
 if __name__ == "__main__":