Upload 5 files

Dockerfile

@@ -1,12 +1,7 @@
-FROM python:3.9
-
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-
-WORKDIR /app
-COPY --chown=user . .
-
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
-
+FROM python:3.11-slim
+WORKDIR /code
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+COPY . .
+EXPOSE 7860
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"] 

main.py

@@ -1,101 +1,27 @@
-from fastapi import FastAPI, Request, Query, HTTPException
-from fastapi.responses import JSONResponse, PlainTextResponse
-from typing import Optional
-import os
-import httpx
-import base64
+import os, json
+from fastapi import FastAPI, Request, HTTPException, Query
+from fastapi.responses import PlainTextResponse, JSONResponse
 
 app = FastAPI()
-VERIFY_TOKEN = os.getenv('VERIFY_TOKEN')
-PHONE_NUMBER_ID = os.getenv('PHONE_NUMBER_ID')
-WHATSAPP_API_TOKEN = os.getenv('WHATSAPP_API_TOKEN')
 
-if not all([VERIFY_TOKEN, WHATSAPP_API_TOKEN, PHONE_NUMBER_ID]):
-    raise ValueError("Missing required environment variables")
+VERIFY_TOKEN = os.getenv("VERIFY_TOKEN", "")  # set in HF Secrets
 
-@app.get("/")
-def root():
-    return {"message": "Hello from WhatsApp ChatBot"}
-
-@app.get("/ping")
-def ping():
-    return {"pong": True}
-
-@app.get("/test")
-async def test_get():
-    return {"detail": "GET method working"}
-
-@app.post("/test")
-async def test_post():
-    return {"detail": "POST method working"}
-
-@app.get("/webhook")
+@app.get("/webhook")  # verification
 def verify(
-    hub_mode:        Optional[str] = Query(None, alias="hub.mode"),
-    hub_challenge:   Optional[str] = Query(None, alias="hub.challenge"),
-    hub_verify_token: Optional[str] = Query(None, alias="hub.verify_token"),
+    hub_mode: str = Query(None, alias="hub.mode"),
+    hub_token: str = Query(None, alias="hub.verify_token"),
+    hub_challenge: str = Query(None, alias="hub.challenge")
 ):
-    # 1) Initial probe
-    if hub_mode is None and hub_challenge is None and hub_verify_token is None:
-        return PlainTextResponse(status_code=200, content="")
-
-    # 2) Real handshake: echo back challenge (temporarily ignore token)
-    if hub_mode == "subscribe":
-        return PlainTextResponse(content=hub_challenge or "", media_type="text/plain")
+    if hub_mode == "subscribe" and hub_token == VERIFY_TOKEN:
+        return PlainTextResponse(hub_challenge)
+    raise HTTPException(status_code=403, detail="Invalid token")
 
-    # 3) Everything else → forbidden
-    if hub_verify_token != VERIFY_TOKEN:
-        raise HTTPException(status_code=403, detail="Invalid verify token")
+@app.post("/webhook")  # incoming messages
+async def receive_update(req: Request):
+    data = await req.json()
+    print("WA update:", json.dumps(data)[:300])
+    return JSONResponse({"status": "received"})
 
-@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/{PHONE_NUMBER_ID}/messages"
-    headers = {
-        "Authorization": f"Bearer {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 = WHATSAPP_API_TOKEN
-    if token:
-        return {"token_loaded": True}
-    return JSONResponse(content={"error": "Token not found in environment"}, status_code=400) 
+@app.get("/ping")      # health check
+def ping():
+    return {"pong": True} 

requirements.txt

@@ -1,3 +1,2 @@
 fastapi
-uvicorn
-httpx 
+uvicorn[standard]