Upload 5 files

Dockerfile

@@ -9,4 +9,4 @@ COPY --chown=user . .
 
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] 
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"] 

main.py

@@ -0,0 +1,84 @@
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse, PlainTextResponse
+import os
+import httpx
+import base64
+
+app = FastAPI()
+
+@app.get("/")
+def root():
+    return {"message": "Hello from WhatsApp ChatBot"}
+
+@app.get("/ping")
+def ping():
+    return {"pong": True}
+
+@app.post("/test")
+def test():
+    return {"status": "ok"}
+
+@app.get("/webhook")
+async def verify_webhook(request: Request):
+    params = request.query_params
+    mode = params.get("hub.mode")
+    token = params.get("hub.verify_token")
+    challenge = params.get("hub.challenge")
+    verify_token = os.getenv("WHATSAPP_TOKEN", "")
+    
+    if mode == "subscribe" and token == verify_token and challenge:
+        return PlainTextResponse(content=challenge, status_code=200)
+    return JSONResponse(content={"error": "Invalid token"}, status_code=403)
+
+@app.post("/webhook")
+async def receive_webhook(request: Request):
+    data = await request.json()
+    try:
+        msg = data["entry"][0]["changes"][0]["value"]["messages"][0]
+        wa_id = msg["from"]
+        msg_type = msg["type"]
+
+        if msg_type == "image":
+            media_id = msg["image"]["id"]
+            image_bytes = await download_whatsapp_media(media_id)
+            if not image_bytes:
+                return {"status": "error", "message": "Failed to process image"}
+            b64_str = base64.b64encode(image_bytes).decode('utf-8')
+            reply = f"Base64: {b64_str[:10]}"
+            send_whatsapp_message(wa_id, reply)
+            return {"status": "success", "base64_first_10": b64_str[:10]}
+        else:
+            send_whatsapp_message(wa_id, "Please send an image.")
+            return {"status": "replied", "message": "Please send an image"}
+    except Exception as e:
+        print("Webhook error:", e)
+        return JSONResponse(content={"error": str(e)}, status_code=400)
+
+async def download_whatsapp_media(media_id):
+    try:
+        with open("test.jpg", "rb") as f:
+            return f.read()
+    except FileNotFoundError:
+        print("test.jpg not found")
+        return None
+
+def send_whatsapp_message(wa_id, text):
+    url = f"https://graph.facebook.com/v19.0/{os.getenv('PHONE_NUMBER_ID')}/messages"
+    headers = {
+        "Authorization": f"Bearer {os.getenv('WHATSAPP_API_TOKEN')}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "messaging_product": "whatsapp",
+        "to": wa_id,
+        "type": "text",
+        "text": {"body": text}
+    }
+    httpx.post(url, headers=headers, json=payload)
+
+@app.post("/send")
+async def check_token():
+    token = os.getenv("WHATSAPP_TOKEN", "")
+    if token:
+        return {"token_loaded": True}
+    return JSONResponse(content={"error": "Token not found in environment"}, status_code=400)