Implement WhatsApp status updater with image generation and sending functionality

app.py

@@ -26,6 +26,10 @@ BOT_STATUS_CHAT       = "[email protected]"  # Chat ID for system messages
 image_dir             = "/tmp/images"
 audio_dir             = "/tmp/audio"
 
+# Ensure necessary directories exist
+os.makedirs(image_dir, exist_ok=True)
+os.makedirs(audio_dir, exist_ok=True)
+
 if not all([GREEN_API_URL, GREEN_API_TOKEN, GREEN_API_ID_INSTANCE, WEBHOOK_AUTH_TOKEN]):
     raise ValueError("Environment variables are not set properly")
 
@@ -52,7 +56,7 @@ def inactivity_monitor():
 
 threading.Thread(target=inactivity_monitor, daemon=True).start()
 
-# --- Background Worker for normal tasks ---
+# --- Background Worker ---
 def worker():
     while True:
         task = task_queue.get()
@@ -71,60 +75,6 @@ def worker():
 
 threading.Thread(target=worker, daemon=True).start()
 
-# --- WhatsApp Status Updater ---
-def send_status_image(image_path, caption=""):
-    """
-    Send an image as WhatsApp status using the Green API.
-    The endpoint used here is assumed to be `/sendFileStatus`.
-    Adjust the URL if the docs specify a different endpoint.
-    """
-    url = f"{GREEN_API_MEDIA_URL}/waInstance{GREEN_API_ID_INSTANCE}/sendFileStatus/{GREEN_API_TOKEN}"
-    payload = {"caption": caption}
-    files = [("file", ("status.jpg", open(image_path, "rb"), "image/jpeg"))]
-    retries = 3
-    for i in range(retries):
-        try:
-            r = requests.post(url, data=payload, files=files)
-            r.raise_for_status()
-            logging.debug("Status image sent successfully.")
-            return r.json()
-        except requests.RequestException as e:
-            logging.error("send_status_image attempt %d failed: %s", i+1, e)
-            if i == retries - 1:
-                return {"error": str(e)}
-
-def status_updater():
-    """
-    Background thread that every 2 minutes:
-      1. Generates a random image prompt using the LLM.
-      2. Generates an image from that prompt.
-      3. Updates the WhatsApp status with the generated image.
-    """
-    while True:
-        try:
-            # Generate a random creative prompt for a status image.
-            prompt_llm = "Generate a unique and creative image prompt suitable for a WhatsApp status update."
-            random_prompt = generate_llm(prompt_llm).strip()
-            logging.debug("Random status prompt: %s", random_prompt)
-
-            # Generate the image from the prompt.
-            img, path, ret_prompt, url_img = generate_image(random_prompt, "status", "status", image_dir)
-            if img:
-                caption = f"Status updated! {ret_prompt}"
-                send_status_image(path, caption=caption)
-                logging.info("WhatsApp status updated with image: %s", url_img)
-                # Clean up the image file if needed.
-                if os.path.exists(path):
-                    os.remove(path)
-            else:
-                logging.error("Image generation for status failed for prompt: %s", random_prompt)
-        except Exception as e:
-            logging.error("Error in status_updater: %s", e)
-        # Wait for 2 minutes before updating status again.
-        time.sleep(120)
-
-threading.Thread(target=status_updater, daemon=True).start()
-
 # --- send helpers ---
 def send_message_to_chat(to_number, message, retries=3):
     chat_id = to_number if to_number.endswith("@g.us") else to_number
@@ -188,6 +138,30 @@ def send_audio(message_id, to_number, audio_path, retries=3):
     except Exception as e:
         return {"error": str(e)}
 
+# --- New helper: send WhatsApp status image ---
+def send_status_image(image_path, caption="Status Update", retries=3):
+    """
+    Sends an image as your WhatsApp status update using Green API's sendMediaStatus endpoint.
+    """
+    url = f"{GREEN_API_MEDIA_URL}/waInstance{GREEN_API_ID_INSTANCE}/sendMediaStatus/{GREEN_API_TOKEN}"
+    payload = {"caption": caption}
+    try:
+        with open(image_path, "rb") as f:
+            files = [("file", ("status.jpg", f, "image/jpeg"))]
+            for i in range(retries):
+                try:
+                    r = requests.post(url, data=payload, files=files)
+                    r.raise_for_status()
+                    logging.info("Status image sent successfully.")
+                    return r.json()
+                except requests.RequestException as e:
+                    if i == retries - 1:
+                        logging.error("send_status_image failed: %s", str(e))
+                        return {"error": str(e)}
+    except Exception as e:
+        logging.error("Error opening status image: %s", e)
+        return {"error": str(e)}
+
 # --- core response functions ---
 def response_text(message_id, chat_id, prompt):
     try:
@@ -214,7 +188,8 @@ def response_audio(message_id, chat_id, prompt):
 def handle_image_generation(message_id, chat_id, prompt):
     for i in range(4):
         try:
-            img, path, ret_prompt, url_img = generate_image(prompt, message_id, message_id, image_dir)
+            # Here we pass "message_id" for both the message and chat identifiers.
+            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()
@@ -223,14 +198,43 @@ def handle_image_generation(message_id, chat_id, prompt):
                     message_id,
                     chat_id,
                     path,
-                    caption=f"✨ Image ready: {url_img}\n>{chr(8203)} {formatted_ret_prompt}"
+                    caption=f"✨ Image ready: {url}\n>{chr(8203)} {formatted_ret_prompt}"
                 )
             else:
                 send_message(message_id, chat_id, "Image generation failed.")
+            break  # exit on success
         except Exception as e:
             logging.error("Error in handle_image_generation: %s", e)
             send_message(message_id, chat_id, "Error generating image.")
 
+# --- New background thread: WhatsApp Status Updater ---
+def whatsapp_status_updater():
+    """
+    Every 2 minutes, this function uses the LLM to generate a random creative image prompt,
+    generates an image from it, and sends it as a WhatsApp status update.
+    """
+    while True:
+        try:
+            # Generate a random image prompt via LLM
+            random_prompt = generate_llm("Generate a creative and random image prompt for a WhatsApp status update.")
+            logging.info("Random status prompt: %s", random_prompt)
+            # Generate image from the prompt; using "status" as a dummy message/chat id.
+            img, path, ret_prompt, url = generate_image(random_prompt, "status", "status", image_dir)
+            if img and os.path.exists(path):
+                caption = f"Status: {random_prompt}"
+                send_status_image(path, caption)
+                # Optionally remove the image file after sending
+                os.remove(path)
+            else:
+                logging.error("Image generation for status failed.")
+        except Exception as e:
+            logging.error("Error in whatsapp_status_updater: %s", e)
+        # Sleep for 2 minutes
+        time.sleep(120)
+
+# Start the WhatsApp status updater thread
+threading.Thread(target=whatsapp_status_updater, daemon=True).start()
+
 # --- Startup Message ---
 def send_startup_message():
     if BOT_STATUS_CHAT:
@@ -272,7 +276,7 @@ async def whatsapp_webhook(request: Request):
         data = await request.json()
         chat_id = data["senderData"]["chatId"]
         print(f"New message from chat ID: {chat_id}")
-    except Exception:
+    except:
         return JSONResponse({"error": "Invalid JSON"}, status_code=400)
     if data.get("typeWebhook") != "incomingMessageReceived":
         return {"success": True}
@@ -329,7 +333,7 @@ async def whatsapp_webhook(request: Request):
         try:
             joke = requests.get("https://official-joke-api.appspot.com/random_joke", timeout=5).json()
             send_message(mid, chat, f"{joke['setup']}\n\n{joke['punchline']}")
-        except Exception:
+        except:
             send_message(mid, chat, generate_llm("Tell me a short, funny joke."))
         return {"success": True}
 
@@ -338,7 +342,7 @@ async def whatsapp_webhook(request: Request):
         try:
             w = requests.get(f"http://sl.wttr.in/{loc}?format=4", timeout=5).text
             send_message(mid, chat, w)
-        except Exception:
+        except:
             send_message(mid, chat, "Could not fetch weather.")
         return {"success": True}