Upload main.py

main.py

@@ -10,7 +10,7 @@ app = FastAPI()
 VERIFY_TOKEN = os.getenv("VERIFY_TOKEN", "")  # set in HF Secrets
 PHONE_NUMBER_ID = os.getenv("PHONE_NUMBER_ID", "")
 WHATSAPP_API_TOKEN = os.getenv("WHATSAPP_API_TOKEN", "")
-RUMAS_API_URL = os.getenv("RUMAS_API_URL", "")
+CLIENT_API_URL = os.getenv("CLIENT_API_URL", "")
 
 # Add task tracking at the top
 active_tasks = {}
@@ -33,8 +33,8 @@ def save_processed_id(msg_id):
 
 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."""
+async def forward_to_client(b64_str: str, wa_id: str, task_id: str):
+    """Fire-and-forget call to CLIENT /api/process_id."""
     payload = {
         "data": [b64_str],
         "configurations": {
@@ -46,19 +46,19 @@ async def forward_to_rumas(b64_str: str, wa_id: str, task_id: str):
     }
     short = b64_str[:40]                       # first 40 chars for reference
     try:
-        print(f"[RUMAS] ➜ sending image ({len(b64_str)} bytes) "
+        print(f"[CLIENT] ➜ sending image ({len(b64_str)} bytes) "
               f"preview='{short}…'")
 
-        print("Payload to RUMAS:", json.dumps(payload)[:300], "…")
+        print("Payload to CLIENT:", json.dumps(payload)[:300], "…")
 
         async with httpx.AsyncClient(timeout=60) as client:
-            resp = await client.post(RUMAS_API_URL, json=payload)
+            resp = await client.post(CLIENT_API_URL, json=payload)
             resp.raise_for_status()
             # ── log the full JSON response (truncated to 300 chars for brevity) ──
             data = resp.json()
-            print(f"[RUMAS] ✔ 200 {json.dumps(data)[:300]}…")
+            print(f"[CLIENT] ✔ 200 {json.dumps(data)[:300]}…")
             body = data
-            print(f"[RUMAS] ✔ {resp.status_code}")
+            print(f"[CLIENT] ✔ {resp.status_code}")
             pretty_print_dict(body)
 
             # ── send parsed ID back to WhatsApp or error message ──
@@ -83,10 +83,10 @@ async def forward_to_rumas(b64_str: str, wa_id: str, task_id: str):
                     print(f"[TASK] Completed and removed task {task_id}")
                 elif "error" in data:
                     error_msg = data["error"]
-                    print(f"[TASK] RUMAS error for task {task_id}: {error_msg}")
+                    print(f"[TASK] CLIENT error for task {task_id}: {error_msg}")
                     send_whatsapp_message(wa_id, f"❌ {error_msg}")
                 else:
-                    print(f"[TASK] RUMAS response not successful for task {task_id}")
+                    print(f"[TASK] CLIENT response not successful for task {task_id}")
                     send_whatsapp_message(wa_id, "❌ Sorry, we could not process your ID. Please try again with a clearer image.")
                 # Remove task from active list
                 active_tasks.pop(task_id, None)
@@ -94,7 +94,7 @@ async def forward_to_rumas(b64_str: str, wa_id: str, task_id: str):
                 print(f"[TASK] Ignoring response for inactive task {task_id}")
 
     except Exception as e:
-        print(f"[RUMAS] ✘ error while sending '{short}…' →", e)
+        print(f"[CLIENT] ✘ error while sending '{short}…' →", e)
         # Remove task from active list on error
         active_tasks.pop(task_id, None)
 
@@ -152,7 +152,7 @@ def send_welcome_buttons(wa_id: str):
       "interactive": {
         "type": "button",
         "body": {
-          "text": "Welcome to RUMAS.\nPlease send an image of your South African ID document."
+          "text": "Welcome to CLIENT.\nPlease send an image of your South African ID document."
         },
         "action": {
           "buttons": [
@@ -201,7 +201,7 @@ async def receive_update(req: Request):
     if msg.get("type") == "text":
         send_whatsapp_message(
             wa_id,
-            "Welcome to RUMAS! 📋\n\n"
+            "Welcome to CLIENT! 📋\n\n"
             "Please send an image of your South African ID document for processing."
         )
         return JSONResponse({"status": "welcome_sent"})
@@ -225,11 +225,11 @@ async def receive_update(req: Request):
         full_b64 = base64.b64encode(media_bytes).decode("utf-8")
         preview  = full_b64[:10]
 
-        # 4) forward full string to RUMAS (non-blocking)
+        # 4) forward full string to CLIENT (non-blocking)
         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))
+        asyncio.create_task(forward_to_client(full_b64, wa_id, task_id))
 
         # 5) echo preview to the user
         return JSONResponse({"status": "image_preview_sent", "preview": preview})