Update app.py

app.py

@@ -43,6 +43,7 @@ import httpx
 import langdetect
 from langdetect import detect
 import threading
+import shutil
 
 # Configure advanced logging
 logging.basicConfig(
@@ -2018,7 +2019,7 @@ async def handle_all_products_selection(from_number: str, selection: str, user_c
         send_helpful_guidance(from_number, 'all_products_menu')
 
 async def handle_intelligent_product_inquiry(from_number: str, query: str, user_context: dict, reply_language: str = 'en'):
-    """Handle product inquiry with OpenAI intelligence"""
+    """Handle product inquiry with OpenAI intelligence and media support"""
     try:
         # First try direct product search
         products = get_veterinary_product_matches(query)
@@ -2141,7 +2142,7 @@ Format your response professionally with emojis and clear structure. Keep it con
                 context_manager.add_to_history(from_number, query, message)
                 
             else:
-                # Single product found - show detailed information as before
+                # Single product found - show detailed information with media support
                 selected_product = products[0]
                 context_manager.update_context(
                     from_number, 
@@ -2153,21 +2154,15 @@ Format your response professionally with emojis and clear structure. Keep it con
                 
                 # Get updated context with last message
                 context = context_manager.get_context(from_number)
-                response = generate_veterinary_product_response(selected_product, context)
                 
-                # Translate response if needed
-                if reply_language == 'ur':
-                    try:
-                        translated_response = GoogleTranslator(source='auto', target='ur').translate(response)
-                        send_whatsjet_message(from_number, translated_response)
-                    except Exception as e:
-                        logger.error(f"[AI] Translation error: {e}")
-                        send_whatsjet_message(from_number, response)
-                else:
-                    send_whatsjet_message(from_number, response)
+                # Use enhanced response with media support
+                response_with_media = generate_veterinary_product_response_with_media(selected_product, context)
+                
+                # Send the response
+                send_whatsjet_message(from_number, response_with_media['text'])
                 
                 # Add to conversation history
-                context_manager.add_to_history(from_number, query, response)
+                context_manager.add_to_history(from_number, query, response_with_media['text'])
             
         else:
             # Enhanced "not found" response with veterinary suggestions
@@ -3564,6 +3559,107 @@ def generate_veterinary_product_response_with_media(product_info: Dict[str, Any]
         'product_name': product_name
     }
 
+# Ensure static/images directory exists for product images
+def ensure_images_dir():
+    images_dir = os.path.join('static', 'images')
+    os.makedirs(images_dir, exist_ok=True)
+    return 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]):
+    """
+    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)
+    image_url = product.get('Images', '').strip() if 'Images' in product else ''
+    temp_file_path = None
+    try:
+        if image_url:
+            # Convert Google Drive link to direct download if needed
+            if 'drive.google.com' in 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}"
+            # 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}")
+            else:
+                logger.warning(f"[Product] Could not download image from CSV link: {image_url}")
+        # Fallback to local static/images
+        image_path = get_product_image_path(product_name)
+        if image_path and os.path.exists(image_path):
+            media_type = get_product_image_media_type(image_path)
+            filename = f"{product_name.replace(' ', '_')}.jpg"
+            success = send_whatsjet_message(
+                from_number,
+                details,
+                media_type=media_type,
+                media_path=image_path,
+                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:
+            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):
+    """
+    Test endpoint to send Heposel product image and details as caption to the given phone number.
+    """
+    # Load Heposel product from CSV
+    try:
+        df = pd.read_csv('Veterinary.csv')
+        row = df[df['Product Name'].str.lower() == 'heposel']
+        if row.empty:
+            return {"error": "Heposel not found in CSV"}
+        product = row.iloc[0].to_dict()
+        user_context = context_manager.get_context(phone)
+        await send_product_image_with_caption(phone, product, user_context)
+        return {"status": "sent", "phone": phone}
+    except Exception as e:
+        return {"error": str(e)}
+
 if __name__ == "__main__":
     # Launch FastAPI app
     import uvicorn