Update app.py

app.py

@@ -2780,58 +2780,136 @@ def ensure_images_dir():
 
 # New feature: Send product image with caption (product details)
 async def send_product_image_with_caption(from_number: str, product: Dict[str, Any], user_context: Dict[str, Any]):
-    logger.info(f"[Product] send_product_image_with_caption called for: {product.get('Product Name', '')}")
     """
     Send product image (if available) with product details as caption in a single WhatsApp message.
     If image is not available, send only the product details as text.
+    Now supports 'Images' column in CSV (Google Drive or direct links).
     """
     ensure_images_dir()
     product_name = product.get('Product Name', 'Unknown Product')
     details = generate_veterinary_product_response(product, user_context)
-    logger.info(f"[DEBUG] About to look up image for product: {product_name}")
-    image_path = get_product_image_path(product_name)
+    image_url = product.get('Images', '').strip() if 'Images' in product else ''
+    
+    # Force image URL for Respira Aid Plus (use cPanel public URL)
+    if product_name.lower().strip() == "respira aid plus":
+        image_url = "https://amgocus.com/uploads/images/Respira%20Aid%20Plus.jpg"
     
-    if image_path and image_path.startswith('http'):
-        media_type = get_product_image_media_type(image_path)
+    logger.info(f"[Product] Processing image for product: {product_name}")
+    logger.info(f"[Product] Image URL from CSV: {image_url}")
+    
+    try:
+        # First, check if we have an image URL from CSV
+        if image_url:
+            # Convert Google Drive link to direct download if needed
+            if 'drive.google.com' in image_url:
+                logger.info(f"[Product] Converting Google Drive link: {image_url}")
+                if '/d/' in image_url:
+                    file_id = image_url.split('/d/')[1].split('/')[0]
+                elif 'id=' in image_url:
+                    file_id = image_url.split('id=')[1].split('&')[0]
+                else:
+                    file_id = ''
+                if file_id:
+                    image_url = f"https://drive.google.com/uc?export=download&id={file_id}"
+                    logger.info(f"[Product] Converted to direct download URL: {image_url}")
+            
+            # Use the public URL directly for WhatsApp API
+            media_type = 'image/jpeg'
+            filename = f"{product_name.replace(' ', '_')}.jpg"
+            
+            # Test the image URL first
+            try:
+                logger.info(f"[Product] Testing image URL accessibility: {image_url}")
+                headers = {
+                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+                    'Accept': 'image/webp,image/apng,image/*,*/*;q=0.8',
+                    'Accept-Language': 'en-US,en;q=0.9',
+                    'Accept-Encoding': 'gzip, deflate, br',
+                    'Connection': 'keep-alive',
+                    'Upgrade-Insecure-Requests': '1'
+                }
+                test_response = requests.head(image_url, headers=headers, timeout=10, allow_redirects=True)
+                if test_response.status_code != 200:
+                    logger.warning(f"[Product] Image URL not accessible (status {test_response.status_code}): {image_url}")
+                    raise Exception(f"Image URL not accessible: {test_response.status_code}")
+                logger.info(f"[Product] Image URL is accessible")
+            except Exception as e:
+                logger.warning(f"[Product] Failed to test image URL {image_url}: {e}")
+                image_url = None
+            
+            # Send using public URL (not local file)
+            if image_url:
+                logger.info(f"[Product] Attempting to send image from CSV URL for: {product_name}")
+                success = send_whatsjet_message(
+                    from_number,
+                    details,
+                    media_type=media_type,
+                    media_path=image_url,  # Use public URL directly
+                    filename=filename
+                )
+                
+                if success:
+                    logger.info(f"[Product] Successfully sent image from CSV link with caption for product: {product_name}")
+                    return
+                else:
+                    logger.warning(f"[Product] Failed to send image from CSV link, trying fallback: {product_name}")
+        
+        # Fallback 1: Try with a known public test image
+        logger.info(f"[Product] Trying public test image for: {product_name}")
+        test_image_url = "https://www.w3schools.com/w3images/lights.jpg"
+        media_type = 'image/jpeg'
         filename = f"{product_name.replace(' ', '_')}.jpg"
-        logger.info(f"[Product] Sending public image URL with caption: {image_path}")
         
-        # Try sending with image first
         success = send_whatsjet_message(
             from_number,
             details,
             media_type=media_type,
-            media_path=image_path,
+            media_path=test_image_url,
             filename=filename
         )
         
         if success:
-            logger.info(f"[Product] Successfully sent image+caption for product: {product_name}")
+            logger.info(f"[Product] Successfully sent test image with caption for product: {product_name}")
             return
-        else:
-            logger.warning(f"[Product] Failed to send image+caption, trying alternative approaches for: {product_name}")
+        
+        # Fallback 2: Try local uploads directory (public URL)
+        logger.info(f"[Product] Trying local uploads directory for: {product_name}")
+        image_path = get_product_image_path(product_name)
+        if image_path and (image_path.startswith('http') or os.path.exists(image_path)):
+            media_type = get_product_image_media_type(image_path)
+            filename = f"{product_name.replace(' ', '_')}.jpg"
             
-            # Try sending image only (without caption) using the separate endpoint
-            try:
-                image_only_success = send_whatsjet_media_image_only(from_number, image_path, filename)
-                if image_only_success:
-                    logger.info(f"[Product] Successfully sent image only for: {product_name}")
-                    # Send caption as separate message
-                    send_whatsjet_message(from_number, details)
-                    return
-                else:
-                    logger.warning(f"[Product] Image-only approach also failed for: {product_name}")
-            except Exception as e:
-                logger.error(f"[Product] Exception in image-only approach: {e}")
+            # If it's already a public URL, use it directly
+            if image_path.startswith('http'):
+                media_path = image_path
+                logger.info(f"[Product] Using existing public URL: {media_path}")
+            else:
+                # Convert local path to public URL
+                media_path = f"https://dreamstream-1-chatbot.hf.space/uploads/{os.path.basename(image_path).replace(' ', '%20')}"
+                logger.info(f"[Product] Converted local path to public URL: {media_path}")
             
-            # Final fallback: send text only
-            logger.warning(f"[Product] All image approaches failed, sending text only: {product_name}")
+            success = send_whatsjet_message(
+                from_number,
+                details,
+                media_type=media_type,
+                media_path=media_path,  # Use public URL
+                filename=filename
+            )
+            
+            if success:
+                logger.info(f"[Product] Successfully sent image with caption for product: {product_name}")
+            else:
+                logger.warning(f"[Product] Failed to send image, sending text only: {product_name}")
+                send_whatsjet_message(from_number, details)
+        else:
+            # No image available, send text only
+            logger.info(f"[Product] No image available, sending text only for: {product_name}")
             send_whatsjet_message(from_number, details)
-            return
-    
-    # No image available, send text only
-    logger.info(f"[Product] No image available, sending text only for: {product_name}")
-    send_whatsjet_message(from_number, details)
+            
+    except Exception as e:
+        logger.error(f"[Product] Error sending product image with caption: {e}")
+        logger.info(f"[Product] Falling back to text-only message for: {product_name}")
+        send_whatsjet_message(from_number, details)
 
 # Test endpoint for product image with caption
 @app.get("/test-product-image-with-caption")