TheOneHong commited on
Commit
6fd2572
·
1 Parent(s): e765d2d

handled err better

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  import fal_client
 
3
  import requests
4
  from PIL import Image
5
  from io import BytesIO
@@ -44,6 +45,15 @@ def generate_image(api_key, prompt, image_size, seed, sync_mode, num_images, ena
44
  img = Image.open(BytesIO(response.content))
45
  images.append(img)
46
  return [gr.update(value=images, visible=True), gr.update(value=str(result), visible=True)]
 
 
 
 
 
 
 
 
 
47
  except Exception as e:
48
  error_msg = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
49
  print(error_msg)
 
1
  import gradio as gr
2
  import fal_client
3
+ from fal_client.client import FalClientError
4
  import requests
5
  from PIL import Image
6
  from io import BytesIO
 
45
  img = Image.open(BytesIO(response.content))
46
  images.append(img)
47
  return [gr.update(value=images, visible=True), gr.update(value=str(result), visible=True)]
48
+
49
+ except FalClientError as e:
50
+ error_messages = []
51
+ for error_obj in e.args[0]:
52
+ error_messages.append(error_obj['msg'])
53
+
54
+ error_msg = "Errors:\n" + "\n".join(error_messages)
55
+ print(error_msg)
56
+ return [gr.update(value=[]), gr.update(value=error_msg)]
57
  except Exception as e:
58
  error_msg = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
59
  print(error_msg)