Update main.py

main.py

@@ -111,6 +111,42 @@ def create_reply_button_message(text: str, buttons: List[Dict]) -> Dict:
     }
 
 
+def handle_media_message(messenger, data, message_type, mobile):
+    """
+    Handle media messages (images, documents, audio)
+    Retrieve and process media content
+    """
+    try:
+        # Download media content based on type
+        if message_type == "image":
+            media_url = messenger.get_image_url(data)
+        elif message_type == "document":
+            media_url = messenger.get_document_url(data)
+        elif message_type == "audio":
+            media_url = messenger.get_audio_url(data)
+        else:
+            logger.error(f"Unsupported media type: {message_type}")
+            return None
+
+        # Download media content
+        response = requests.get(media_url, headers={
+            "Authorization": f"Bearer {os.getenv('whatsapp_token')}"
+        })
+        
+        if response.status_code != 200:
+            logger.error(f"Failed to download media: {response.status_code}")
+            messenger.send_message("Sorry, I couldn't process your media.", mobile)
+            return None
+
+        # Return media content for further processing
+        return response.content
+
+    except Exception as e:
+        logger.error(f"Error handling media message: {str(e)}")
+        messenger.send_message("Sorry, I couldn't process your media.", mobile)
+        return None
+
+
 @app.get("/")
 def verify_token():
     """Webhook verification endpoint"""
@@ -282,52 +318,84 @@ def hook():
         return "Internal server error", 500
 
 def handle_interactive_message(messenger, data, mobile, carts):
-    """Handle interactive message responses"""
+    """Handle interactive message responses with enhanced cart functionality"""
     try:
         response = messenger.get_interactive_response(data)
         button_id = response.get("id")
         
         if button_id == "add_to_cart":
-            # Add products to cart
+            # Add products to cart (assuming cart data is stored in the previous interaction)
             if mobile not in carts:
                 carts[mobile] = []
-            # Add current products to cart logic here
+            # Placeholder for adding last analyzed products to cart
             
             buttons = [
-                {"id": "more_shopping", "title": "Do More Shopping"},
+                {"id": "continue_shopping", "title": "Continue Shopping"},
                 {"id": "checkout", "title": "Checkout"}
             ]
-            button_message = create_reply_button_message("Products added to cart!", buttons)
+            button_message = create_reply_button_message(
+                "Products added to cart! What would you like to do next?", 
+                buttons
+            )
             messenger.send_reply_button(
                 recipient_id=mobile,
                 button=button_message
             )
 
+        elif button_id == "continue_shopping":
+            # Encourage further shopping
+            messenger.send_message(
+                "Great! You can continue shopping by sending me a text or image of what you're looking for.", 
+                mobile
+            )
+
         elif button_id == "checkout":
             if mobile not in carts or not carts[mobile]:
                 messenger.send_message("Your cart is empty!", mobile)
                 return
 
+            # Calculate total
+            total = sum(product['price'] for product in carts[mobile])
+            
+            # Generate receipt
+            receipt_text = "Your Purchase Receipt:\n\n"
+            for product in carts[mobile]:
+                receipt_text += f"- {product['product']}: ${product['price']:.2f}\n"
+            receipt_text += f"\nTotal: ${total:.2f}\n\n"
+            receipt_text += "Expected Delivery: Within 1 hour\n"
+            receipt_text += "Thank you for your purchase!"
+
             buttons = [
-                {"id": "confirm_checkout", "title": "Confirm"},
-                {"id": "cancel_checkout", "title": "Cancel"}
+                {"id": "confirm_order", "title": "Confirm Order"},
+                {"id": "modify_order", "title": "Modify Order"}
             ]
-            button_message = create_reply_button_message("Ready to complete your purchase?", buttons)
+            button_message = create_reply_button_message(receipt_text, buttons)
             messenger.send_reply_button(
                 recipient_id=mobile,
                 button=button_message
             )
 
-        elif button_id == "confirm_checkout":
-            # Process checkout
-            carts[mobile] = []  # Clear cart
-            messenger.send_message("Thank you for your purchase!", mobile)
+        elif button_id == "confirm_order":
+            # Finalize order
+            messenger.send_message(
+                "Order confirmed! Your items will be delivered within 1 hour. Thank you!", 
+                mobile
+            )
+            # Clear the cart
+            carts[mobile] = []
+
+        elif button_id == "modify_order":
+            # Allow user to modify order
+            messenger.send_message(
+                "Please tell us any changes you'd like to make to your order.", 
+                mobile
+            )
 
     except Exception as e:
         logger.error(f"Error handling interactive message: {str(e)}")
         messenger.send_message("Sorry, something went wrong.", mobile)
 
-# Keep your original media handling, DNS setup, and other utility functions
+
 
 def test_connection():
     """Test DNS resolution and connection to Graph API"""