uptade reply handeling

app.py

@@ -299,6 +299,8 @@ class WhatsAppBot:
         self.conv_manager = ConversationManager(config.MAX_HISTORY_SIZE)
         self.intent_router = IntentRouter(self.conv_manager, self.logger)
         self.task_queue = queue.Queue()
+        # Cache to store recent image message IDs and their download URLs
+        self.image_cache = defaultdict(lambda: deque(maxlen=50))
         self.fastapi_app = FastAPI(title="WhatsApp Eve Bot", version="2.0.0")
         self._setup_routes()
         self._start_workers()
@@ -360,28 +362,42 @@ class WhatsAppBot:
             message_id = payload["idMessage"]
             
             # ** CHAT RESTRICTION LOGIC **
-            # Check if the sender is in the allowed list
             if chat_id not in self.config.ALLOWED_CHATS:
                 self.logger.warning(f"Ignoring message from unauthorized chat ID: {chat_id}")
-                return  # Stop processing immediately
+                return
 
-            # Set thread context for logging
             ThreadContext.set_context(chat_id, message_id)
 
             message_data = payload.get("messageData", {})
             type_message = message_data.get("typeMessage")
             
+            # --- Improved message processing logic ---
+            
+            # 1. Handle image caching
+            if type_message == "imageMessage":
+                download_url = message_data.get("fileMessageData", {}).get("downloadUrl")
+                if download_url:
+                    self.logger.debug(f"Caching image message {message_id} from {chat_id} with URL: {download_url}")
+                    self.image_cache[chat_id].append((message_id, download_url))
+
+            # 2. Extract text from various message types
             text = ""
             if type_message == "textMessage":
-                text = message_data["textMessageData"]["textMessage"]
+                text = message_data.get("textMessageData", {}).get("textMessage", "")
             elif type_message == "extendedTextMessage":
-                text = message_data["extendedTextMessageData"]["text"]
+                text = message_data.get("extendedTextMessageData", {}).get("text", "")
+            elif type_message == "quotedMessage":
+                text = message_data.get("extendedTextMessageData", {}).get("text", "")
+            elif type_message == "imageMessage":
+                text = message_data.get("fileMessageData", {}).get("caption", "")
 
             text = text.strip()
             if not text:
-                self.logger.debug(f"Received empty message from {chat_id}. Ignoring.")
+                self.logger.debug(f"Received message type '{type_message}' without actionable text from {chat_id}. No action taken.")
                 return
             
+            # --- End of improved logic ---
+            
             self.logger.info(f"Processing message from {chat_id}: '{text}'")
             self.conv_manager.add_user_message(chat_id, text)
 
@@ -450,12 +466,40 @@ class WhatsAppBot:
             
     def _dispatch_edit_image(self, chat_id, message_id, prompt, payload):
         """Checks for a replied-to image and dispatches the edit task."""
-        quoted_message = payload.get("messageData", {}).get("extendedTextMessageData", {}).get("quotedMessage")
-        if not quoted_message or quoted_message.get("typeMessage") != "imageMessage":
-            self.api_client.send_message(chat_id, "To edit an image, please reply to it with your instructions.", message_id)
+        message_data = payload.get("messageData", {})
+        
+        # Ensure we are dealing with a reply
+        if message_data.get("typeMessage") != "quotedMessage":
+            self.api_client.send_message(chat_id, "To edit an image, please reply to an image with your instructions.", message_id)
+            return
+
+        quoted_info = message_data.get("quotedMessage")
+        if not quoted_info or quoted_info.get("typeMessage") != "imageMessage":
+            self.api_client.send_message(chat_id, "You must reply to an image to edit it.", message_id)
             return
             
-        download_url = quoted_message["imageMessage"]["downloadUrl"]
+        # Get the ID of the message being replied to
+        quoted_message_id = quoted_info.get("stanzaId")
+        if not quoted_message_id:
+            self.logger.error(f"Could not find stanzaId in quoted message: {quoted_info}")
+            self.api_client.send_message(chat_id, "Sorry, I couldn't identify which image to edit.", message_id)
+            return
+
+        # Find the downloadUrl from our cache
+        download_url = None
+        if chat_id in self.image_cache:
+            # Search the cache for the matching message ID
+            for msg_id, url in self.image_cache[chat_id]:
+                if msg_id == quoted_message_id:
+                    download_url = url
+                    break
+        
+        if not download_url:
+            self.logger.warning(f"Could not find download URL for message ID {quoted_message_id} in cache for chat {chat_id}.")
+            self.api_client.send_message(chat_id, "I couldn't find the original image. It might be too old. Please send it again before editing.", message_id)
+            return
+
+        self.logger.info(f"Found cached image URL for message {quoted_message_id}. Dispatching edit task.")
         self.task_queue.put({
             "type": "edit_image",
             "chat_id": chat_id,