Refactor image sending logic to send first three images without caption and the last image with a caption listing all URLs.

app.py

@@ -191,20 +191,31 @@ def handle_image_generation(message_id, chat_id, prompt):
                 send_message(message_id, chat_id, "Image generation failed.")
                 return
             images_info.append((path, ret_prompt, url))
-        # Create a caption listing the URLs of all generated images.
+        
+        # Build caption with URLs of all images.
         caption = "✨ Images ready:\n" + "\n".join(url for _, _, url in images_info)
-        # Extract the file paths from the generated images.
-        image_paths = [path for path, _, _ in images_info]
-        # Send all generated images concurrently.
-        send_images(message_id, chat_id, image_paths, caption=caption)
+        
+        # Send first 3 images without caption.
+        for path, _, _ in images_info[:3]:
+            result = send_image(message_id, chat_id, path, caption="")
+            if "error" in result:
+                send_message(message_id, chat_id, "Failed to send one of the images.")
+        
+        # Send the last image with the caption.
+        last_image_path = images_info[3][0]
+        result = send_image(message_id, chat_id, last_image_path, caption=caption)
+        if "error" in result:
+            send_message(message_id, chat_id, "Failed to send the final image with caption.")
+        
         # Cleanup generated image files.
-        for path in image_paths:
+        for path, _, _ in images_info:
             if os.path.exists(path):
                 os.remove(path)
     except Exception as e:
         logging.error("Error in handle_image_generation: %s", e)
         send_message(message_id, chat_id, "Error generating images.")
 
+
 # --- Startup Message ---
 def send_startup_message():
     if BOT_STATUS_CHAT: