Commit
·
6fd2572
1
Parent(s):
e765d2d
handled err better
Browse files
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)
|