rahuln2002 commited on
Commit
bee1cf0
·
verified ·
1 Parent(s): bb44a9e

Update knowledgeassistant/components/summarization.py

Browse files
knowledgeassistant/components/summarization.py CHANGED
@@ -21,7 +21,13 @@ class DataSummarization:
21
  logging.info("Summarization Pipeline Successfully Setup")
22
 
23
  text = read_txt_file(input_text_path)
 
 
 
 
 
24
  summary = pipe(text, min_length = min_length, do_sample = False)
 
25
  logging.info("Text successfully summarized")
26
 
27
  write_txt_file(self.data_summarization_config.summarized_text_file_path, summary[0].get("summary_text"))
 
21
  logging.info("Summarization Pipeline Successfully Setup")
22
 
23
  text = read_txt_file(input_text_path)
24
+ logging.info(f"Input Text: {text[:500]}")
25
+
26
+ if not text.strip():
27
+ raise KnowledgeAssistantException("Input text is empty")
28
+
29
  summary = pipe(text, min_length = min_length, do_sample = False)
30
+ logging.info(f"Summary Output: {summary}")
31
  logging.info("Text successfully summarized")
32
 
33
  write_txt_file(self.data_summarization_config.summarized_text_file_path, summary[0].get("summary_text"))