Improve error handling in parsing

utils/parse.py

@@ -5,8 +5,11 @@ import matplotlib.pyplot as plt
 import numpy as np
 import warnings
 import inflect
+import gradio as gr
 
 p = inflect.engine()
+# user_error = ValueError
+user_error = gr.Error
 
 img_dir = "imgs"
 objects_text = "Objects: "
@@ -27,7 +30,7 @@ def parse_input(text=None, no_input=False):
     
     if not text:
         if no_input:
-            return
+            raise user_error(f"No input parsed in \"{text}\".")
         
         text = input("Enter the response: ")
     if objects_text in text:
@@ -38,7 +41,7 @@ def parse_input(text=None, no_input=False):
         gen_boxes, bg_prompt = text_split
     elif len(text_split) == 1:
         if no_input:
-            return
+            raise user_error(f"Invalid input (no background prompt): {text}")
         gen_boxes = text
         bg_prompt = ""
         while not bg_prompt:
@@ -47,7 +50,7 @@ def parse_input(text=None, no_input=False):
         if bg_prompt_text_no_trailing_space in bg_prompt:
             bg_prompt = bg_prompt.split(bg_prompt_text_no_trailing_space)[1]
     else:
-        raise ValueError(f"text: {text}")
+        raise user_error(f"Invalid input (possibly multiple background prompts): {text}")
     try:
         gen_boxes = ast.literal_eval(gen_boxes)    
     except SyntaxError as e:
@@ -65,7 +68,7 @@ def parse_input_with_negative(text=None, no_input=False):
     
     if not text:
         if no_input:
-            return
+            raise user_error(f"No input parsed in \"{text}\".")
         
         text = input("Enter the response: ")
     if objects_text in text:
@@ -76,7 +79,7 @@ def parse_input_with_negative(text=None, no_input=False):
         gen_boxes, text_rem = text_split
     elif len(text_split) == 1:
         if no_input:
-            return
+            raise user_error(f"Invalid input (no background prompt): {text}")
         gen_boxes = text
         text_rem = ""
         while not text_rem:
@@ -85,7 +88,7 @@ def parse_input_with_negative(text=None, no_input=False):
         if bg_prompt_text_no_trailing_space in text_rem:
             text_rem = text_rem.split(bg_prompt_text_no_trailing_space)[1]
     else:
-        raise ValueError(f"text: {text}")
+        raise user_error(f"Invalid input (possibly multiple background prompts): {text}")
     
     text_split = text_rem.split(neg_prompt_text_no_trailing_space)
     
@@ -101,7 +104,7 @@ def parse_input_with_negative(text=None, no_input=False):
             if neg_prompt_text_no_trailing_space in neg_prompt:
                 neg_prompt = neg_prompt.split(neg_prompt_text_no_trailing_space)[1]
     else:
-        raise ValueError(f"text: {text}")
+        raise user_error(f"Invalid input (possibly multiple negative prompts): {text}")
     
     try:
         gen_boxes = ast.literal_eval(gen_boxes)