show error on UI if it happens

app.py

@@ -5,13 +5,15 @@ from llm_api.exceptions import RefusalError
 # from llm_api.langchain_api import extract_info, follow_structure
 from llm_api.native_api import extract_info, follow_structure
 from openai import BadRequestError
+# from utils import log_input, log_output
 
 # load_dotenv('.env', override=True)
 
 def extract_free_fn(gallery, provider, schema):
-    img_paths = [item[0] for item in gallery]
     # write schema to file before calling extract_free_properties
     try:
+        img_paths = [item[0] for item in gallery]
+        # request_id = log_input('extract', img_paths, provider, schema)
         if schema is not None and isinstance(schema, str) and 'import' not in schema.replace("'import'", '').replace("'important'", ''):
             with open('schema_free.py', 'r') as f:
                 default_schema = f.read()
@@ -26,21 +28,23 @@ def extract_free_fn(gallery, provider, schema):
             print('SCHEMA ELSE')
             raise Exception('Invalid schema')
     except Exception as e:
-        print(e)
-        return 'Invalid schema! Please check the schema or Reload this page to restore default settings', {}, []
+        return f"ERROR: {e}", {}, []
 
     try:
         free_attributes = extract_info(img_paths, provider, Garment)
         free_attributes = free_attributes.model_dump_json()
     except RefusalError as e:
-        return e.message, {}, []
+        return f"ERROR: {e}", {}, []
     except BadRequestError as e:
-        return e.message, {}, []
-
+        return f"ERROR: {e}", {}, []
+    except Exception as e:
+        return f"ERROR: {e}", {}, []
+    # log_output(request_id, str(free_attributes))
     return f'OK - provider: {provider}', free_attributes, [1] * 100
 
 def follow_schema_fn(free_attributes, provider, probs, schema):
     try:
+        # request_id = log_input('follow', None, provider, schema)
         if schema is not None and isinstance(schema, str) and 'import' not in schema.replace("'import'", '').replace("'important'", ''):
             with open('schema_structured.py', 'r') as f:
                 default_schema = f.read()
@@ -54,17 +58,20 @@ def follow_schema_fn(free_attributes, provider, probs, schema):
         else:
             print('SCHEMA ELSE')
             raise Exception('Invalid schema')
-    except:
-        return 'Invalid schema! Please check the schema or Reload this page to restore default settings', {}
+    except Exception as e:
+        return f"ERROR: {e}", {}
 
     try:
         schema_attributes = follow_structure(free_attributes, provider, Garment)
         schema_attributes = schema_attributes.model_dump_json()
     except RefusalError as e:
-        return e.message, {}
+        return f"ERROR: {e}", {}
     except BadRequestError as e:
-        return e.message, {}
+        return f"ERROR: {e}", {}
+    except Exception as e:
+        return f"ERROR: {e}", {}
     
+    # log_output(request_id, str(schema_attributes))
     return f'OK - provider: {provider}', schema_attributes
 
 with gr.Blocks(title='Internal Demo for Attribution') as demo: