Upload main.py

main.py

@@ -61,31 +61,37 @@ async def forward_to_rumas(b64_str: str, wa_id: str, task_id: str):
             print(f"[RUMAS] ✔ {resp.status_code}")
             pretty_print_dict(body)
 
-            # ── send parsed ID back to WhatsApp ──
-            if wa_id and data.get("status") == "success" and task_id in active_tasks:
-                print(f"[TASK] Sending response for active task {task_id}")
-                det = data["results"][0]["detections"][0]
-                reply = (
-                    "✅ *ID processed*\n"
-                    f"*Name*: {det.get('name','?')} {det.get('surname','')}\n"
-                    f"*ID No.*: {det.get('id_num','?')}\n"
-                    f"*DOB*: {det.get('date_of_birth','?')}\n"
-                    f"*Gender*: {det.get('gender','?')}\n"
-                    f"*ID Type*: {det.get('id_type','?')}\n"
-                    f"*Country of Birth*: {det.get('cob','?')}\n"
-                    f"*Citizen*: {det.get('citizen','?')}\n"
-                    f"*Citizen Status*: {det.get('citizen_status','?')}\n"
-                    f"*Date Issued*: {det.get('date_issued','?')}\n"
-                    f"*Age*: {det.get('age','?')}"
-                )
-                send_whatsapp_message(wa_id, reply)
+            # ── send parsed ID back to WhatsApp or error message ──
+            if wa_id and task_id in active_tasks:
+                if data.get("status") == "success":
+                    print(f"[TASK] Sending response for active task {task_id}")
+                    det = data["results"][0]["detections"][0]
+                    reply = (
+                        "✅ *ID processed*\n"
+                        f"*Name*: {det.get('name','?')} {det.get('surname','')}\n"
+                        f"*ID No.*: {det.get('id_num','?')}\n"
+                        f"*DOB*: {det.get('date_of_birth','?')}\n"
+                        f"*Gender*: {det.get('gender','?')}\n"
+                        f"*ID Type*: {det.get('id_type','?')}\n"
+                        f"*Country of Birth*: {det.get('cob','?')}\n"
+                        f"*Citizen*: {det.get('citizen','?')}\n"
+                        f"*Citizen Status*: {det.get('citizen_status','?')}\n"
+                        f"*Date Issued*: {det.get('date_issued','?')}\n"
+                        f"*Age*: {det.get('age','?')}"
+                    )
+                    send_whatsapp_message(wa_id, reply)
+                    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}")
+                    send_whatsapp_message(wa_id, f"❌ {error_msg}")
+                else:
+                    print(f"[TASK] RUMAS 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)
-                print(f"[TASK] Completed and removed task {task_id}")
             elif task_id not in active_tasks:
                 print(f"[TASK] Ignoring response for inactive task {task_id}")
-            else:
-                print(f"[TASK] RUMAS response not successful for task {task_id}")
 
     except Exception as e:
         print(f"[RUMAS] ✘ error while sending '{short}…' →", e)