Improve image generation error handling and update message for image count in webhook

app.py

@@ -242,8 +242,11 @@ def _fn_generate_images(
             client.send_media(message_id, chat_id, path, cap, media_type="image")
             os.remove(path)
         except Exception as e:
-            logger.warning(f"Img {i}/{count} failed: {e}")
-            _fn_send_text(message_id, chat_id, f"😢 Failed to generate image {i}/{count}.")
+            if "Timed out" in str(e):
+                logger.warning("Image generation timed out.")
+            else:
+                logger.warning(f"Img {i}/{count} failed: {e}")
+                _fn_send_text(message_id, chat_id, f"😢 Failed to generate image {i}/{count}.")
 
 def _fn_voice_reply(
     message_id: str,
@@ -489,16 +492,16 @@ async def whatsapp_webhook(request: Request):
     if low.startswith("/gen"):
         parts = body[4:].split("|")
         pr = parts[0].strip()
-        ct = int(parts[1]) if len(parts) > 1 and parts[1].isdigit() else BotConfig.DEFAULT_IMAGE_COUNT
+        count = int(parts[1]) if len(parts) > 1 and parts[1].isdigit() else BotConfig.DEFAULT_IMAGE_COUNT
         width = int(parts[2]) if len(parts) > 2 and parts[2].isdigit() else None
         height = int(parts[3]) if len(parts) > 3 and parts[3].isdigit() else None
-        _fn_send_accept(mid, chat_id, f"✨ Generating {ct} image(s)…")
+        _fn_send_accept(mid, chat_id, f"✨ Generating {count if count != 1 else 'a'} image{'s' if count != 1 else ''}...")
         task_queue.put({
             "type":       "image",
             "message_id": mid,
             "chat_id":    chat_id,
             "prompt":     pr,
-            "count":      ct,
+            "count":      count,
             "width":      width,
             "height":     height
         })