Add support for sending multiple images at once and update image generation handling

app.py

@@ -113,6 +113,32 @@ def send_image(message_id, to_number, image_path, caption="Here you go!", retrie
             if i == retries - 1:
                 return {"error": str(e)}
 
+def send_images(message_id, to_number, image_paths, caption="Here are your images!", retries=3):
+    """
+    Uploads multiple images at once.
+    """
+    chat_id = to_number if to_number.endswith("@g.us") else to_number
+    url = f"{GREEN_API_MEDIA_URL}/waInstance{GREEN_API_ID_INSTANCE}/sendFileByUpload/{GREEN_API_TOKEN}"
+    payload = {"chatId": chat_id, "caption": caption, "quotedMessageId": message_id}
+    files = []
+    file_handles = []
+    try:
+        for idx, image_path in enumerate(image_paths):
+            f = open(image_path, "rb")
+            file_handles.append(f)
+            files.append(("file", (f"image_{idx}.jpg", f, "image/jpeg")))
+        for i in range(retries):
+            try:
+                r = requests.post(url, data=payload, files=files)
+                r.raise_for_status()
+                return r.json()
+            except requests.RequestException as e:
+                if i == retries - 1:
+                    return {"error": str(e)}
+    finally:
+        for f in file_handles:
+            f.close()
+
 def send_audio(message_id, to_number, audio_path, retries=3):
     logging.debug("send_audio")
     chat_id = to_number if to_number.endswith("@g.us") else to_number
@@ -159,22 +185,26 @@ def response_audio(message_id, chat_id, prompt):
 
 def handle_image_generation(message_id, chat_id, prompt):
     try:
-        img, path, ret_prompt, url = generate_image(prompt, message_id, message_id, image_dir)
-        if img:
-            formatted_ret_prompt = "\n\n".join(
-                f"_{paragraph.strip()}_" for paragraph in ret_prompt.split("\n\n") if paragraph.strip()
-            )
-            send_image(
-                message_id,
-                chat_id,
-                path,
-                caption=f"✨ Image ready: {url}\n>{chr(8203)} {formatted_ret_prompt}"
-            )
-        else:
-            send_message(message_id, chat_id, "Image generation failed.")
+        images_info = []
+        # Generate 4 images using the same prompt.
+        for _ in range(4):
+            img, path, ret_prompt, url = generate_image(prompt, message_id, message_id, image_dir)
+            if not img:
+                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.
+        caption = "✨ Images ready:\n" + "\n".join(url for _, _, url in images_info)
+        # Send all generated images at once.
+        image_paths = [path for path, _, _ in images_info]
+        send_images(message_id, chat_id, image_paths, caption=caption)
+        # Cleanup generated image files.
+        for path in image_paths:
+            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 image.")
+        send_message(message_id, chat_id, "Error generating images.")
 
 # --- Startup Message ---
 def send_startup_message():
@@ -187,22 +217,22 @@ def send_startup_message():
         logging.warning("BOT_STATUS_CHAT is not set; startup message not sent.")
 
 help_text = (
-            "🤖 *Hi there, I'm Eve!* Here are the commands you can use:\n\n"
-            "• */help* – _Show this help message._\n"
-            "• */summarize <text>* – _Get a quick summary of your text._\n"
-            "• */translate <language>|<text>* – _Translate text to your chosen language._\n"
-            "• */joke* – _Enjoy a random, funny joke._\n"
-            "• */weather <location>* – _Get the current weather for a location._\n"
-            "• */inspire* – _Receive a short inspirational quote._\n"
-            "• */trivia* – _Start a new trivia question._\n"
-            "• */answer [your answer]* – _Reveal the trivia answer or check your answer if provided._\n"
-            "• */meme <text>* – _Generate a fun meme image._\n"
-            "• */poll <Question>|<Option1>|<Option2>|…* – _Create a poll._\n"
-            "• */results* – _See current poll results._\n"
-            "• */endpoll* – _End the poll and show final results._\n"
-            "• */gen <prompt>* – _Generate an image from your prompt._\n\n"
-            "Send any other text and I'll reply with a voice message. I'm here to help, so don't hesitate to ask!"
-        )
+    "🤖 *Hi there, I'm Eve!* Here are the commands you can use:\n\n"
+    "• */help* – _Show this help message._\n"
+    "• */summarize <text>* – _Get a quick summary of your text._\n"
+    "• */translate <language>|<text>* – _Translate text to your chosen language._\n"
+    "• */joke* – _Enjoy a random, funny joke._\n"
+    "• */weather <location>* – _Get the current weather for a location._\n"
+    "• */inspire* – _Receive a short inspirational quote._\n"
+    "• */trivia* – _Start a new trivia question._\n"
+    "• */answer [your answer]* – _Reveal the trivia answer or check your answer if provided._\n"
+    "• */meme <text>* – _Generate a fun meme image._\n"
+    "• */poll <Question>|<Option1>|<Option2>|…* – _Create a poll._\n"
+    "• */results* – _See current poll results._\n"
+    "• */endpoll* – _End the poll and show final results._\n"
+    "• */gen <prompt>* – _Generate an image from your prompt._\n\n"
+    "Send any other text and I'll reply with a voice message. I'm here to help, so don't hesitate to ask!"
+)
 
 # --- Webhook ---
 @app.post("/whatsapp")
@@ -321,12 +351,10 @@ async def whatsapp_webhook(request: Request):
 
     # ANSWER: Accept any message starting with /answer. If additional text is provided, check it.
     if low.startswith("/answer"):
-        # Remove command and any extra spaces
         user_response = body[len("/answer"):].strip()
         if chat in trivia_store:
             correct_answer = trivia_store[chat]["answer"]
             question = trivia_store[chat]["question"]
-            # If user provided an answer, evaluate it via LLM; otherwise, just reveal the answer.
             if user_response:
                 eval_prompt = (
                     f"Question: {question}\n"
@@ -431,7 +459,6 @@ def index():
     return "Server is running!"
 
 if __name__ == "__main__":
-    # Send startup message on launch
     def send_startup_message():
         if BOT_STATUS_CHAT:
             startup_msg = "🌟 Hi! I'm Eve, your friendly AI assistant. I'm now live and ready to help with images, voice replies, and more!"