Update app.py

app.py

@@ -24,12 +24,19 @@ def generate_image(prompt):
         },
     }
 
-    response = requests.post(api_url, headers=headers, json=payload)
-    response.raise_for_status()  # Verifica se a requisição foi bem-sucedida
-    image_data = response.json().get("data", [])[0]  # Obtém os dados da imagem
-
-    # Converte os dados da imagem para um objeto Image do PIL e retorna
-    return Image.open(BytesIO(image_data))
+    try:
+        response = requests.post(api_url, headers=headers, json=payload)
+        response.raise_for_status()  # Verifica se a requisição foi bem-sucedida
+        image_data = response.json().get("data", [])[0]  # Obtém os dados da imagem
+        return Image.open(BytesIO(image_data))
+    except requests.exceptions.HTTPError as http_err:
+        print(f"HTTP error occurred: {http_err}")
+    except requests.exceptions.RequestException as err:
+        print(f"An error occurred: {err}")
+    except requests.exceptions.JSONDecodeError as json_err:
+        print(f"JSON decode error: {json_err}")
+    except Exception as e:
+        print(f"An unexpected error occurred: {e}")
 
 # Criando a interface do Gradio
 iface = gr.Interface(