Add error message.

controlled_summarization.py

@@ -3,6 +3,7 @@ import torch
 from SciAssist import Summarization
 import os
 import requests
+import gradio as gr
 from datasets import load_dataset
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
@@ -107,7 +108,7 @@ def ctrlsum_for_str(input, length=None, keywords=None) -> List[Tuple[str, str]]:
 def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> List[Tuple[str, str, str]]:
     if input == None and url == "":
         if text == "":
-            return None, "Input cannot be left blank.", None
+            raise gr.Error("Input cannot be left blank!")
         else:
             return ctrlsum_for_str(text, length, keywords), text, None
     else:
@@ -132,7 +133,7 @@ def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> L
 
                 filename = download_pdf(url, './cache/')
             else:
-                "Invalid url(Not PDF)!", None, None
+                raise gr.Error("Invalid URL (Not PDF)!")
         else:
             filename = input.name
         if keywords != "":
@@ -150,11 +151,13 @@ def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> L
             results = ctrlsum_pipeline.predict(filename,
                                                save_results=False, length=length, keywords=keywords, num_beams=1)
         else:
-            return "File Format Error !", None, filename
+            raise gr.Error("File Format Error! Please upload .txt or .pdf files.")
 
         output = []
         for res in results["summary"]:
             output.append(f"{res}\n\n")
+        if results["raw_text"] == "":
+            raise gr.Error("Unable to parse File! Please try a different file.")
         return "".join(output), results["raw_text"], filename
 
 

description.py

@@ -44,8 +44,6 @@ To **test on strings**, simply input a string.
 '''
 
 ctrlsum_file_md = '''
-This is the demo for **CocoSciSum**.
-
 ## Controlled Summarization uses FLAN-T5 to generate user-customised summaries from your input file or URL link.
 
 To **test on a file**, the input can be: