Upload main.py

main.py

@@ -12,9 +12,28 @@ PHONE_NUMBER_ID = os.getenv("PHONE_NUMBER_ID", "")
 WHATSAPP_API_TOKEN = os.getenv("WHATSAPP_API_TOKEN", "")
 RUMAS_API_URL = os.getenv("RUMAS_API_URL", "")
 
-processed_message_ids = set()
+# Add task tracking at the top
+active_tasks = {}
 
-async def forward_to_rumas(b64_str: str, wa_id: str):
+# Persistent deduplication using a file
+DEDUP_FILE = "processed_message_ids.txt"
+def load_processed_ids():
+    try:
+        with open(DEDUP_FILE, "r") as f:
+            return set(line.strip() for line in f if line.strip())
+    except FileNotFoundError:
+        return set()
+
+def save_processed_id(msg_id):
+    try:
+        with open(DEDUP_FILE, "a") as f:
+            f.write(msg_id + "\n")
+    except Exception as e:
+        print(f"Error saving msg_id {msg_id}: {e}")
+
+processed_message_ids = load_processed_ids()
+
+async def forward_to_rumas(b64_str: str, wa_id: str, task_id: str):
     """Fire-and-forget call to RUMAS /api/process_id."""
     payload = {
         "data": [b64_str],
@@ -43,7 +62,8 @@ async def forward_to_rumas(b64_str: str, wa_id: str):
             pretty_print_dict(body)
 
             # ── send parsed ID back to WhatsApp ──
-            if wa_id and data.get("status") == "success":
+            if wa_id and data.get("status") == "success" and task_id in active_tasks:
+                print(f"[TASK] Sending response for active task {task_id}")
                 det = data["results"][0]["detections"][0]
                 reply = (
                     "✅ *ID processed*\n"
@@ -59,9 +79,18 @@ async def forward_to_rumas(b64_str: str, wa_id: str):
                     f"*Age*: {det.get('age','?')}"
                 )
                 send_whatsapp_message(wa_id, reply)
+                # Remove task from active list
+                active_tasks.pop(task_id, None)
+                print(f"[TASK] Completed and removed task {task_id}")
+            elif task_id not in active_tasks:
+                print(f"[TASK] Ignoring response for inactive task {task_id}")
+            else:
+                print(f"[TASK] RUMAS response not successful for task {task_id}")
 
     except Exception as e:
         print(f"[RUMAS] ✘ error while sending '{short}…' →", e)
+        # Remove task from active list on error
+        active_tasks.pop(task_id, None)
 
 def pretty_print_dict(d, indent=0):
     for k, v in d.items():
@@ -160,6 +189,7 @@ async def receive_update(req: Request):
         print(f"Duplicate message {msg_id}, skipping.")
         return JSONResponse({"status": "duplicate", "id": msg_id})
     processed_message_ids.add(msg_id)
+    save_processed_id(msg_id) # Save to file
 
     # TEXT: send welcome message
     if msg.get("type") == "text":
@@ -190,7 +220,10 @@ async def receive_update(req: Request):
         preview  = full_b64[:10]
 
         # 4) forward full string to RUMAS (non-blocking)
-        asyncio.create_task(forward_to_rumas(full_b64, wa_id))
+        task_id = str(msg_id) # Use msg_id as task_id
+        active_tasks[task_id] = True # Add task to active list
+        print(f"[TASK] Created task {task_id} for message {msg_id}")
+        asyncio.create_task(forward_to_rumas(full_b64, wa_id, task_id))
 
         # 5) echo preview to the user
         return JSONResponse({"status": "image_preview_sent", "preview": preview})