Update app.py

app.py

@@ -3609,8 +3609,9 @@ async def send_product_image_with_caption(from_number: str, product: Dict[str, A
     product_name = product.get('Product Name', 'Unknown Product')
     details = generate_veterinary_product_response(product, user_context)
     image_url = product.get('Images', '').strip() if 'Images' in product else ''
-    temp_file_path = None
+    
     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:
@@ -3622,58 +3623,60 @@ async def send_product_image_with_caption(from_number: str, product: Dict[str, A
                     file_id = ''
                 if file_id:
                     image_url = f"https://drive.google.com/uc?export=download&id={file_id}"
-            # Download the image to a temp file
-            response = requests.get(image_url, stream=True, timeout=10)
-            if response.status_code == 200:
-                with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as tmp:
-                    for chunk in response.iter_content(1024):
-                        tmp.write(chunk)
-                    temp_file_path = tmp.name
-                media_type = 'image/jpeg'
-                filename = f"{product_name.replace(' ', '_')}.jpg"
-                success = send_whatsjet_message(
-                    from_number,
-                    details,
-                    media_type=media_type,
-                    media_path=temp_file_path,
-                    filename=filename
-                )
-                if success:
-                    logger.info(f"[Product] Sent image from CSV link with caption for product: {product_name}")
-                    return
-                else:
-                    logger.warning(f"[Product] Failed to send image from CSV link, sending text only: {product_name}")
+            
+            # Use the public URL directly for WhatsApp API
+            media_type = 'image/jpeg'
+            filename = f"{product_name.replace(' ', '_')}.jpg"
+            
+            # Send using public URL (not local file)
+            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] Sent image from CSV link with caption for product: {product_name}")
+                return
             else:
-                logger.warning(f"[Product] Could not download image from CSV link: {image_url}")
-        # Fallback to local static/images
+                logger.warning(f"[Product] Failed to send image from CSV link, trying fallback: {product_name}")
+        
+        # Fallback to local uploads directory (public URL)
         image_path = get_product_image_path(product_name)
-        if image_path and os.path.exists(image_path):
+        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"
+            
+            # If it's already a public URL, use it directly
+            if image_path.startswith('http'):
+                media_path = image_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')}"
+            
             success = send_whatsjet_message(
                 from_number,
                 details,
                 media_type=media_type,
-                media_path=image_path,
+                media_path=media_path,  # Use public URL
                 filename=filename
             )
+            
             if success:
                 logger.info(f"[Product] 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
             send_whatsjet_message(from_number, details)
             logger.info(f"[Product] Sent product info without image: {product_name}")
+            
     except Exception as e:
         logger.error(f"[Product] Error sending product image with caption: {e}")
         send_whatsjet_message(from_number, details)
-    finally:
-        if temp_file_path and os.path.exists(temp_file_path):
-            try:
-                os.remove(temp_file_path)
-            except Exception:
-                pass
 
 @app.get("/test-product-image-with-caption")
 async def test_product_image_with_caption(phone: str):
@@ -3696,4 +3699,4 @@ async def test_product_image_with_caption(phone: str):
 if __name__ == "__main__":
     # Launch FastAPI app
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)