Update app.py

app.py

@@ -35,19 +35,22 @@ async def generate_command(
     img_count:int,
     aspect_ratio: app_commands.Choice[str],
 ):
-    genai.configure(api_key=GEMINI_API_KEY)
-    imagen = genai.ImageGenerationModel("imagen-3.0-generate-001")
-
-    result = imagen.generate_images(
-        prompt=prompt,
-        number_of_images=img_count,
-        safety_filter_level="block_only_high",
-        person_generation="allow_adult",
-        aspect_ratio=aspect_ratio,
-    )
-
-    if result:
-        await interaction.followup.send(content=f"Successful request!")
+    try:
+        genai.configure(api_key=GEMINI_API_KEY)
+        imagen = genai.ImageGenerationModel("imagen-3.0-generate-001")
+    
+        result = imagen.generate_images(
+            prompt=prompt,
+            number_of_images=img_count,
+            safety_filter_level="block_only_high",
+            person_generation="allow_adult",
+            aspect_ratio=aspect_ratio,
+        )
+    
+        if result:
+            await interaction.followup.send(content=f"Successful request!")
+    except Exception as e:
+        interaction.followup.send(content=f"exception:{e}")
     files = []
     for i, image in enumerate(result.generated_images):
         # Assuming `image` is a byte array or binary stream of the image