Upload main.py

main.py

@@ -107,10 +107,11 @@ def send_whatsapp_message(wa_id, reply):
 async def receive_update(req: Request):
     data = await req.json()
     try:
-        payload  = data["entry"][0]["changes"][0]["value"]
-        contacts = payload.get("contacts")
-        messages = payload.get("messages", [])
+        change   = data["entry"][0]["changes"][0]["value"]
+        contacts = change.get("contacts")
+        messages = change.get("messages")
         if not contacts or not messages:
+            # Missing expected payload—ignore
             print("Ignoring update, no contacts or messages field")
             return JSONResponse({"status": "ignored", "reason": "no contacts/messages"})
 
@@ -120,43 +121,46 @@ async def receive_update(req: Request):
         print("Malformed webhook payload:", e)
         return JSONResponse({"status": "ignored", "reason": "malformed payload"})
 
-    # 1) ignore our own messages to prevent loops
-    if msg.get("from") == PHONE_NUMBER_ID:
-        return JSONResponse({"status":"ignored","reason":"bot message"})
-
-    # 2) if user sends any text, prompt for ID image
-    if msg.get("type") == "text":
-        send_whatsapp_message(wa_id, "Please send an image of your ID")
-        return JSONResponse({"status":"prompted_for_image"})
-
-    # 3) if user sends an image, process it once
-    if msg.get("type") == "image":
-        msg_id = msg.get("id")
-        if not msg_id:
-            return JSONResponse({"status": "ignored", "reason": "no message id"})
-        if msg_id in processed_message_ids:
-            print(f"Duplicate image message {msg_id}, skipping.")
-            return JSONResponse({"status": "duplicate", "id": msg_id})
-        processed_message_ids.add(msg_id)
-        # (Optional: persist msg_id if you want across restarts)
+    msg_id = msg.get("id")
+    if msg_id in processed_message_ids:
+        print(f"Duplicate message {msg_id}, skipping.")
+        return JSONResponse({"status": "duplicate", "id": msg_id})
+    processed_message_ids.add(msg_id)
+
+    # IMAGE: if user sent an image, convert to Base64 preview
+    if msg["type"] == "image":
         media_id = msg["image"]["id"]
-        # download, convert, forward to RUMAS, and reply with parsed fields
+        # 1) get media URL
         media_resp = httpx.get(
             f"https://graph.facebook.com/v23.0/{media_id}",
             headers={"Authorization": f"Bearer {WHATSAPP_API_TOKEN}"}
         )
         media_resp.raise_for_status()
         media_url = media_resp.json()["url"]
+        # 2) download bytes with auth
         media_bytes = httpx.get(
             media_url,
             headers={"Authorization": f"Bearer {WHATSAPP_API_TOKEN}"}
         ).content
+        # 3) encode full & preview
         full_b64 = base64.b64encode(media_bytes).decode("utf-8")
+        preview  = full_b64[:10]
+
+        # 4) forward full string to RUMAS (non-blocking)
         asyncio.create_task(forward_to_rumas(full_b64, wa_id))
-        return JSONResponse({"status":"processed_id"})
 
-    # 4) ignore all other message types
-    return JSONResponse({"status":"ignored","reason":"unsupported_message_type"})
+        # 5) echo preview to the user
+        return JSONResponse({"status": "image_preview_sent", "preview": preview})
+
+    # TEXT: simple echo
+    if msg["type"] == "text":
+        incoming = msg["text"]["body"]
+        reply    = f"You said: {incoming}"
+        send_whatsapp_message(wa_id, reply)
+        return JSONResponse({"status": "replied"})
+
+    # OTHER: ignore
+    return JSONResponse({"status": "ignored", "type": msg["type"]})
 
 # health-check endpoint for token
 @app.post("/send")