Spaces:
Sleeping
Sleeping
| import os, json, requests, time, re, base64, tempfile, pathlib | |
| from html import escape | |
| from fastapi import FastAPI, Request, Response | |
| app = FastAPI(title="Zalo Proxy Space") | |
| BOT_TOKEN = "__TOKEN__" | |
| TARGET_API = "https://bot-api.zaloplatforms.com" | |
| PROXY_NAME = "__SENDER_DISPLAY__" | |
| HF_TOKEN = os.getenv("HF_TOKEN", "") | |
| DATASET_ID = "__DATASET_ID__" | |
| MAIN_SPACE_URL = "__MAIN_SPACE_URL__" | |
| _logs = [] | |
| def _send(cid, text): | |
| headers = {"Content-Type": "application/json"} | |
| url = f"{TARGET_API}/bot{BOT_TOKEN}/sendMessage" | |
| return requests.post(url, json={"chat_id": cid, "text": text}, headers=headers) | |
| def _safe_name(name): | |
| return re.sub(r'[^a-zA-Z0-9]', '_', str(name))[:30] | |
| def _save_to_dataset(image_url, image_data_b64, description, price, category, sender_id, sender_name): | |
| """Save product data to HF Dataset.""" | |
| if not HF_TOKEN or not DATASET_ID: | |
| _log("dataset_skip", sender_id, "N/A", "HF_TOKEN or DATASET_ID missing") | |
| return None | |
| try: | |
| from huggingface_hub import HfApi | |
| api = HfApi(token=HF_TOKEN) | |
| ts = time.strftime("%Y%m%d_%H%M%S") | |
| safe_sender = _safe_name(sender_id) or "unknown" | |
| img_filename = f"images/{ts}_{safe_sender}.jpg" | |
| meta_filename = f"data/{ts}_{safe_sender}.json" | |
| img_bytes = None | |
| if image_data_b64: | |
| try: | |
| img_bytes = base64.b64decode(image_data_b64) | |
| except Exception: | |
| img_bytes = None | |
| elif image_url: | |
| try: | |
| r = requests.get(image_url, timeout=15) | |
| img_bytes = r.content | |
| except Exception as e: | |
| _log("image_download_fail", sender_id, "N/A", str(e)) | |
| img_bytes = None | |
| if img_bytes: | |
| with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp: | |
| tmp.write(img_bytes) | |
| tmp_path = tmp.name | |
| try: | |
| api.upload_file( | |
| path_or_fileobj=tmp_path, | |
| path_in_repo=img_filename, | |
| repo_id=DATASET_ID, | |
| repo_type="dataset", | |
| token=HF_TOKEN, | |
| commit_message=f"Add product image from {sender_name}", | |
| ) | |
| except Exception as e: | |
| _log("image_upload_fail", sender_id, "N/A", str(e)) | |
| finally: | |
| pathlib.Path(tmp_path).unlink(missing_ok=True) | |
| record = { | |
| "image": img_filename if img_bytes else None, | |
| "description": str(description)[:500] if description else "", | |
| "price": str(price) if price else "", | |
| "category": str(category) if category else "", | |
| "sender_id": str(sender_id), | |
| "sender_name": str(sender_name), | |
| "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"), | |
| } | |
| with tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False) as tmp: | |
| json.dump(record, tmp, indent=2, ensure_ascii=False) | |
| tmp_path = tmp.name | |
| try: | |
| api.upload_file( | |
| path_or_fileobj=tmp_path, | |
| path_in_repo=meta_filename, | |
| repo_id=DATASET_ID, | |
| repo_type="dataset", | |
| token=HF_TOKEN, | |
| commit_message=f"Add product metadata from {sender_name}", | |
| ) | |
| finally: | |
| pathlib.Path(tmp_path).unlink(missing_ok=True) | |
| _log("dataset_saved", sender_id, "N/A", f"Saved to {DATASET_ID}") | |
| return DATASET_ID | |
| except Exception as e: | |
| _log("dataset_error", sender_id, "N/A", str(e)) | |
| return None | |
| async def root(): | |
| return {"status": "ok"} | |
| async def health(): | |
| return {"status": "ok", "proxy_target": "Zalo Bot API", "bot_name": PROXY_NAME, "dataset": DATASET_ID} | |
| async def webhooks_get(): | |
| return Response(content=json.dumps({"message": "Success"}), media_type="application/json", status_code=200) | |
| async def webhooks(request: Request): | |
| body = await request.body() | |
| body_str = body.decode("utf-8") if body else "" | |
| try: | |
| data = json.loads(body_str) | |
| except Exception: | |
| _log("parse_error", "N/A", "N/A", "Bad JSON") | |
| return Response(content=json.dumps({"message": "Bad JSON"}), media_type="application/json", status_code=400) | |
| result = data.get("result", data) | |
| event = result.get("event_name", "unknown") | |
| msg = result.get("message", {}) | |
| sender = msg.get("from", {}) | |
| chat = msg.get("chat", {}) | |
| text = msg.get("text", "") | |
| sender_id = str(sender.get("id", "")) | |
| sender_name = sender.get("display_name") or sender.get("name") or sender_id | |
| chat_id = str(chat.get("id", "")) | |
| chat_type = str(chat.get("chat_type", "")) | |
| attachments = msg.get("attachment", {}) | |
| image_url = "" | |
| image_data_b64 = "" | |
| if attachments: | |
| payload = attachments.get("payload", {}) | |
| if isinstance(payload, str): | |
| try: | |
| payload = json.loads(payload) | |
| except Exception: | |
| payload = {} | |
| image_url = payload.get("url", "") | |
| image_data_b64 = payload.get("data", "") or msg.get("image", "") | |
| _log(event, sender_id, chat_id, text, sender_name, chat_type) | |
| if event == "message.text.received" and chat_id: | |
| description = "" | |
| price = "" | |
| category = "" | |
| desc_match = re.search(r"(?:mo ta|description|desc)[:\s]*([^|]+)", text, re.IGNORECASE) | |
| price_match = re.search(r"(?:gia|price)[:\s]*([\d,.]+)", text, re.IGNORECASE) | |
| cat_match = re.search(r"(?:chuyen muc|category|danh muc)[:\s]*([^|]+?)(?:$|\n)", text, re.IGNORECASE) | |
| if desc_match: | |
| description = desc_match.group(1).strip() | |
| if price_match: | |
| price = price_match.group(1).strip() | |
| if cat_match: | |
| category = cat_match.group(1).strip() | |
| if image_url or image_data_b64 or description or price or category: | |
| dataset_id = _save_to_dataset( | |
| image_url=image_url, | |
| image_data_b64=image_data_b64, | |
| description=description or text[:200], | |
| price=price, | |
| category=category, | |
| sender_id=sender_id, | |
| sender_name=sender_name, | |
| ) | |
| if dataset_id: | |
| reply = ( | |
| " Bot Zalo của bạn đã được AUTOMATION SALE thiết lập thành công!\n\n" | |
| " Mọi cấu hình đã tự động hoàn tất.\n\n" | |
| " Bạn có thể vào https://zalo.me/3019064351949097305 để quản lý và cấu hình bot của mình.\n\n" | |
| f" Dữ liệu sản phẩm đã được lưu vào dataset:\n" | |
| f"https://huggingface.co/datasets/{DATASET_ID}\n\n" | |
| f" Webhook URL: https://{MAIN_SPACE_URL}/webhooks\n" | |
| f" Logs: https://{MAIN_SPACE_URL}/logs\n" | |
| f" Quản lý proxy: https://{MAIN_SPACE_URL}/proxy-spaces\n\n" | |
| " Bot của bạn sẽ tự động trả lời khi có người nhắn tin." | |
| ) | |
| else: | |
| reply = ( | |
| " Bot Zalo của bạn đã được AUTOMATION SALE thiết lập thành công!\n\n" | |
| " Mọi cấu hình đã tự động hoàn tất.\n\n" | |
| " Bạn có thể vào https://zalo.me/3019064351949097305 để quản lý và cấu hình bot của mình." | |
| ) | |
| else: | |
| reply = ( | |
| " Bot Zalo của bạn đã được AUTOMATION SALE thiết lập thành công!\n\n" | |
| " Mọi cấu hình đã tự động hoàn tất.\n\n" | |
| " Bạn có thể vào https://zalo.me/3019064351949097305 để quản lý và cấu hình bot của mình.\n\n" | |
| f" Webhook URL: https://{MAIN_SPACE_URL}/webhooks\n" | |
| f" Logs: https://{MAIN_SPACE_URL}/logs\n" | |
| f" Quản lý proxy: https://{MAIN_SPACE_URL}/proxy-spaces\n\n" | |
| " Bot của bạn sẽ tự động trả lời khi có người nhắn tin." | |
| ) | |
| try: | |
| _send(chat_id, reply) | |
| except Exception as e: | |
| pass | |
| return Response(content=json.dumps({"message": "Success"}), media_type="application/json", status_code=200) | |
| async def proxy_spaces_get(): | |
| """Trang quản lý của proxy space này.""" | |
| html = [ | |
| '<!DOCTYPE html><html><head><title>Quản lý Proxy</title>', | |
| '<meta http-equiv="refresh" content="5">', | |
| '<style>body{font-family:Arial,sans-serif;max-width:1000px;margin:0 auto;padding:16px;background:#fafafa;}h1{color:#1a73e8;}</style></head><body>', | |
| '<h1> Quản lý Proxy</h1>', | |
| f'<p>Webhook URL: <code>https://{MAIN_SPACE_URL}/webhooks</code></p>', | |
| f'<p>Logs: <a href="/logs">https://{MAIN_SPACE_URL}/logs</a></p>', | |
| ] | |
| if DATASET_ID: | |
| html.append(f'<p>Dataset: <a href="https://huggingface.co/datasets/{DATASET_ID}" target="_blank">{DATASET_ID}</a></p>') | |
| html.append('<p>💡 Gửi ảnh kèm mô tả/giá/chuyên mục để lưu sản phẩm vào dataset.</p>') | |
| html.append('</body></html>') | |
| return Response(content="".join(html), media_type="text/html") | |
| async def proxy_logs(): | |
| rows = "" | |
| for log in reversed(_logs[-50:]): | |
| rows += ( | |
| f"<div style='margin:6px 0;padding:8px;background:#f5f5f5;border-radius:4px'>" | |
| f"<b>[{log['event']}]</b> 👤{escape(log['sender_name'])} " | |
| f"🆔<code>{escape(log['sender_id'])}</code> " | |
| f"💬<code>{escape(log['chat_id'])}</code> " | |
| f"[{escape(log['chat_type'])}]<br>" | |
| f"<span style='font-family:monospace;font-size:12px;color:#333'>" | |
| f"{escape(log['text'][:200])}</span><br>" | |
| f"<small style='color:#999'>⏰ {log['time']}</small></div>" | |
| ) | |
| empty_msg = "<p style='color:#999'>Chưa có sự kiện</p>" | |
| content = rows if rows else empty_msg | |
| return Response(content=f"<!DOCTYPE html><html><head><title>Proxy Logs</title><meta http-equiv='refresh' content='5'><style>body{{font-family:Arial,sans-serif;max-width:1000px;margin:0 auto;padding:16px;}}h1{{color:#1a73e8;}} .log-c{{max-height:600px;overflow-y:auto;background:#fff;border-radius:8px;padding:8px;}}</style></head><body><h1> Proxy Logs — {escape(PROXY_NAME)}</h1><p>Webhook proxy cho: <b>{escape(PROXY_NAME)}</b></p><div class='log-c'>{content}</div></body></html>", media_type="text/html") | |
| def _log(event, sender_id, chat_id, text, sender_name="", chat_type=""): | |
| _logs.append({ | |
| "event": str(event), | |
| "sender_id": str(sender_id), | |
| "sender_name": str(sender_name), | |
| "chat_id": str(chat_id), | |
| "chat_type": str(chat_type), | |
| "text": str(text)[:200], | |
| "time": time.strftime("%Y-%m-%d %H:%M:%S"), | |
| }) | |
| if len(_logs) > 100: | |
| del _logs[:50] | |