Upload 5 files

main.py

@@ -2,6 +2,7 @@ import os, json
 from fastapi import FastAPI, Request, HTTPException, Query
 from fastapi.responses import PlainTextResponse, JSONResponse
 import httpx
+import base64
 
 app = FastAPI()
 
@@ -26,12 +27,48 @@ async def receive_update(req: Request):
     wa_id  = change["contacts"][0]["wa_id"]
     msg    = change["messages"][0]
 
-    # 1) Ignore non-text messages
+    # 1) Handle image messages by converting to base64 preview
+    if msg["type"] == "image":
+        # extract media ID
+        media_id = msg["image"]["id"]
+        # fetch media URL
+        media_url_resp = httpx.get(
+            f"https://graph.facebook.com/v23.0/{media_id}",
+            headers={"Authorization": f"Bearer {WHATSAPP_API_TOKEN}"}
+        )
+        media_url_resp.raise_for_status()
+        media_url = media_url_resp.json().get("url")
+        # download the media bytes
+        media_bytes_resp = httpx.get(media_url)
+        media_bytes_resp.raise_for_status()
+        data = media_bytes_resp.content
+        # convert to base64 and take first 10 chars
+        b64_str = base64.b64encode(data).decode('utf-8')
+        preview = b64_str[:10]
+        # reply with preview
+        reply = f"Base64: {preview}"
+        # send back via WhatsApp
+        payload = {
+            "messaging_product": "whatsapp",
+            "to": wa_id,
+            "type": "text",
+            "text": {"body": reply}
+        }
+        resp = httpx.post(
+            f"https://graph.facebook.com/v23.0/{PHONE_NUMBER_ID}/messages",
+            headers={
+                "Authorization": f"Bearer {WHATSAPP_API_TOKEN}",
+                "Content-Type": "application/json"
+            },
+            json=payload
+        )
+        resp.raise_for_status()
+        return JSONResponse({"status": "image_preview_sent", "preview": preview})
+    # 2) Existing text handler
     if msg["type"] != "text":
         print(f"Ignoring message type: {msg['type']}")
         return JSONResponse({"status": "ignored", "type": msg["type"]})
 
-    # 2) Handle text as before
     incoming = msg["text"]["body"]
     reply    = f"You said: {incoming}"