Update app.py

app.py

@@ -888,33 +888,47 @@ def split_message_for_whatsapp(message: str, max_length: int = 1000) -> list:
     return [message[i:i+max_length] for i in range(0, len(message), max_length)]
 
 def send_whatsjet_message(phone_number: str, message: str, media_type: str = None, media_path: str = None, filename: str = None) -> bool:
-    """Send a message using WhatsJet API with optional media attachment"""
+    """Send a message using WhatsJet API with optional media attachment (now supports header_image for images)"""
     if not all([WHATSJET_API_URL, WHATSJET_VENDOR_UID, WHATSJET_API_TOKEN]):
         logger.error("[WhatsJet] Missing environment variables.")
         return False
 
     url = f"{WHATSJET_API_URL}/{WHATSJET_VENDOR_UID}/contact/send-message?token={WHATSJET_API_TOKEN}"
 
-    # Handle media messages (including images with captions)
+    # Use WhatsJet's header_image for images with captions
+    if media_type in ["image/jpeg", "image/png"] and media_path and (media_path.startswith("http://") or media_path.startswith("https://")):
+        payload = {
+            "phone_number": phone_number,
+            "header_image": media_path,
+            "message_body": message
+        }
+        try:
+            response = httpx.post(
+                url,
+                json=payload,
+                timeout=15
+            )
+            response.raise_for_status()
+            logger.info(f"[WhatsJet] header_image message sent successfully to {phone_number}")
+            return True
+        except Exception as e:
+            logger.error(f"[WhatsJet] Exception sending header_image message: {e}")
+            return False
+
+    # Handle other media messages (existing logic)
     if media_type and media_path:
         try:
             # For public URLs, download the content first
             if isinstance(media_path, str) and (media_path.startswith("http://") or media_path.startswith("https://")):
-                # Download from URL
                 response = requests.get(media_path, stream=True, timeout=15)
                 response.raise_for_status()
                 media_content = response.content
-                # Log the first 20 bytes and content type
                 logger.info(f"[WhatsJet][DEBUG] Downloaded image content-type: {response.headers.get('Content-Type')}")
                 logger.info(f"[WhatsJet][DEBUG] First 20 bytes: {media_content[:20]}")
             else:
-                # Read from local file
                 with open(media_path, 'rb') as f:
                     media_content = f.read()
-            
-            # Encode to base64
             media_b64 = base64.b64encode(media_content).decode('utf-8')
-            
             payload = {
                 "phone_number": phone_number, 
                 "message_body": message,
@@ -922,7 +936,6 @@ def send_whatsjet_message(phone_number: str, message: str, media_type: str = Non
                 'media_content': media_b64,
                 'media_filename': filename or os.path.basename(media_path) if not media_path.startswith('http') else filename or 'image.jpg'
             }
-            
             try:
                 response = httpx.post(
                     url,