Enhance image sending functionality to support concurrent uploads of multiple images

app.py

@@ -112,32 +112,30 @@ def send_image(message_id, to_number, image_path, caption="Here you go!", retrie
         except requests.RequestException as e:
             if i == retries - 1:
                 return {"error": str(e)}
+        finally:
+            for _, filetuple in files:
+                filetuple[1].close()
 
 def send_images(message_id, to_number, image_paths, caption="Here are your images!", retries=3):
     """
-    Uploads multiple images at once.
+    Sends multiple images concurrently by calling send_image for each image in parallel.
     """
-    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()
+    responses = []
+    threads = []
+    lock = threading.Lock()
+
+    def send_single(image_path):
+        resp = send_image(message_id, to_number, image_path, caption)
+        with lock:
+            responses.append(resp)
+
+    for path in image_paths:
+        t = threading.Thread(target=send_single, args=(path,))
+        t.start()
+        threads.append(t)
+    for t in threads:
+        t.join()
+    return responses
 
 def send_audio(message_id, to_number, audio_path, retries=3):
     logging.debug("send_audio")
@@ -195,8 +193,9 @@ def handle_image_generation(message_id, chat_id, prompt):
             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.
+        # 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)
         # Cleanup generated image files.
         for path in image_paths:
@@ -349,7 +348,7 @@ async def whatsapp_webhook(request: Request):
             send_message(mid, chat, "Failed to generate trivia. Please try again.")
         return {"success": True}
 
-    # ANSWER: Accept any message starting with /answer. If additional text is provided, check it.
+    # ANSWER
     if low.startswith("/answer"):
         user_response = body[len("/answer"):].strip()
         if chat in trivia_store: