Update app.py

app.py

@@ -31,30 +31,24 @@ def bullete(text,wikipedia_language="en"):
   result = list(filter(lambda x: x != '', final_out.split('\n\n')))
 
   answer = []
-  for i in range(4):
-      if len(result[i]) > 500:
-          summary_point=result[i].split(".")[0:3]
-          answer.append(summary_point)
-  l = []
-  for i in range(len(answer)):
-      l.append("".join(answer[i]))
-  gen_output = []
-  for i in range(len(l)):
-      gen_output.append(l[i] + ".")
-
-  listToStr = ' '.join([str(elem) for elem in gen_output])
-  listToStr = listToStr.replace("\n", "")
-  print(listToStr)
-  nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
-  QA_input = {
-      'question': text,
-      'context': listToStr
-  }
-  print(QA_input)
-  res = nlp(QA_input)
-  values = list(res.values())[3]
-
-  return values
+  try:
+      for i in range(len(result[0].split('.'))):
+          nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
+          QA_input = {
+              'question': text,
+              'context': result[0].split('.')[i]
+          }
+          res = nlp(QA_input)
+          print(QA_input)
+          values = list(res.values())[3]
+          answer.append(values)
+  except:
+      gen_output = []
+      for i in range(len(answer)):
+          gen_output.append("* " + answer[i] + ".")
+      paraphrase = "\n".join(gen_output)
+      final_answer = paraphrase.replace(" ", " ")
+      return final_answer
 
 
 interface = gr.Interface(fn=bullete,
@@ -62,4 +56,4 @@ interface = gr.Interface(fn=bullete,
                          outputs="text",
                          title='Bullet Point')
 
-interface.launch(inline=False)
+interface.launch(inline=False)